# HG changeset patch # User Daniel O'Connor # Date 1519361055 -37800 # Node ID 57dece5674e3a4713fb2287eedabf6d4c06a0ce5 # Parent 29a61ec4755b6ac2fbc8b1afdf97ae6ba4e7e732 Be more clever about what 'today' is diff -r 29a61ec4755b -r 57dece5674e3 graph.py --- a/graph.py Fri Feb 23 15:04:59 2018 +1030 +++ b/graph.py Fri Feb 23 15:14:15 2018 +1030 @@ -66,6 +66,11 @@ if args.days is not None and args.days < 0: parser.error('days must be non-negative') + # If it's before 6am then plot yesterday + today = datetime.date.today() + if datetime.datetime.now().hour < 6: + today -= datetime.timedelta(days = 1) + selector = [args.start is not None, args.end is not None, args.days is not None] if selector == [True, True, False]: # Start and end pass @@ -78,11 +83,11 @@ args.days = 1 args.start = args.end - datetime.timedelta(days = args.days) elif selector == [False, False, True]: # Days - args.end = datetime.date.today() + datetime.timedelta(days = 1) + args.end = today + datetime.timedelta(days = 1) args.end = datetime.datetime(args.end.year, args.end.month, args.end.day) args.start = args.end - datetime.timedelta(days = args.days) elif selector == [False, False, False]: # Nothing - args.start = datetime.date.today() + args.start = today args.start = datetime.datetime(args.start.year, args.start.month, args.start.day) args.end = args.start + datetime.timedelta(days = 1) else: