From 8e65e2d43cd7af44761a24d7232d34dff0c6123b Mon Sep 17 00:00:00 2001
From: giuliof <giuliofieramosca@gmail.com>
Date: Sat, 24 Nov 2018 18:48:48 +0100
Subject: [PATCH] [B.END] code for testing purposes

---
 getInfo.py | 38 ++++++++++++++++++++++----------------
 1 file changed, 22 insertions(+), 16 deletions(-)

diff --git a/getInfo.py b/getInfo.py
index cd650f5..d734298 100755
--- a/getInfo.py
+++ b/getInfo.py
@@ -137,22 +137,28 @@ def getEvents(baseDay, interval):
   # Return events sorted by date, AllDay first
   return sorted(events, key=lambda k: "%s %d" % (k['DATETIME'],k['ALLDAY'] == 0))
 
-# Check if this file is executed as main file or included
-######## SHOW DATA #######
+# Only for test purposes
 if __name__ == '__main__':
-    print("ZERo Optimized CALdav CAlendar Reader Engine")
-    print("--------------------------------------------")
+  print("ZERo Optimized CALdav CAlendar Reader Engine")
+  print("--------------------------------------------")
 
-    events = getEvents(dt.datetime.today(),'week')
 
-    for event in events:
-        try:
-          print ("Event Name: %s" % event['NAME'])
-          print ("Event Date: %s" % event['DATETIME'].date())
-          if event['ALLDAY']:
-              print("All Day")
-          else:
-              print ("Event Time: %s" % event['DATETIME'].time())
-        except :
-          print("Malformed event")
-        print("----------------")
+  baseDay = dt.datetime.today()
+  interval = 'week'
+  if len(sys.argv) == 3:
+    baseDay += dt.timedelta(days=int(sys.argv[1]))
+    interval = sys.argv[2]
+
+  events = getEvents(baseDay, interval)
+
+  for event in events:
+      try:
+        print ("Event Name: %s" % event['NAME'])
+        print ("Event Date: %s" % event['DATETIME'].date())
+        if event['ALLDAY']:
+            print("All Day")
+        else:
+            print ("Event Time: %s" % event['DATETIME'].time())
+      except :
+        print("Malformed event")
+      print("----------------")