# HG changeset patch
# User Daniel O'Connor <darius@dons.net.au>
# Date 1513945183 -3600
# Node ID e1bec6dfd524b4c41bfbdf374bdaa62f753073ee
# Parent  3f5b617b471593d4d6d11995ccfc8506cc79f511
Improve start/end/days handling

diff -r 3f5b617b4715 -r e1bec6dfd524 graph.py
--- a/graph.py	Fri Dec 22 13:14:23 2017 +0100
+++ b/graph.py	Fri Dec 22 13:19:43 2017 +0100
@@ -67,17 +67,22 @@
 
     # Can specify..
     # Start and end
-    # Start and days
-    # End and days
+    # Start and days or Start
+    # End and days or End
     # Nothing
     # Want to end up with a start & end
-    if args.start is not None and args.end is not None:
+    selector = [args.start is not None, args.end is not None, args.days is not None]
+    if selector == [True, True, False]:
         pass
-    elif args.start is not None and args.days is not None:
+    elif selector == [True, False, True] or selector == [True, False, False]:
+        if args.days == None:
+            args.days = 1
         args.end = args.start + datetime.timedelta(days = args.days)
-    elif args.end is not None and args.days is not None:
+    elif selector == [False, True, True] or selector == [False, True, False]:
+        if args.days == None:
+            args.days = 1
         args.start = args.end - datetime.timedelta(days = args.days)
-    elif args.start is None and args.end is None and args.days is None:
+    elif selector == [False, False, True]:
         end = datetime.date.today()
         end = datetime.datetime(start.year, start.month, start.day)
         args.start = args.end - datetime.timedelta(days = args.days)