# HG changeset patch
# User Daniel O'Connor <darius@dons.net.au>
# Date 1505129274 -34200
# Node ID 988e511a5f29ba582a800d346017c4eb97550762
# Parent  b610ce4ec081d249908581fc8a10bcb1455da90f
Run strftime on filename.
Use localtime for display / filename.

diff -r b610ce4ec081 -r 988e511a5f29 agl.py
--- a/agl.py	Mon Sep 11 20:57:10 2017 +0930
+++ b/agl.py	Mon Sep 11 20:57:54 2017 +0930
@@ -27,7 +27,7 @@
     parser = argparse.ArgumentParser()
     parser.add_argument('-u', '--update', help = 'Update data', action="store_true")
     parser.add_argument('-g', '--graph', help = 'Produce graph', action="store_true")
-    parser.add_argument('-f', '--filename', help = 'Filename to save graph as', type = str)
+    parser.add_argument('-f', '--filename', help = 'Filename to save graph as (uses strftime on start)', type = str)
     parser.add_argument('-s', '--start', help = 'Start date for graph (YYYY-MM-DD)', type = valid_date)
     parser.add_argument('-e', '--end', help = 'End date for graph (YYYY-MM-DD)', type = valid_date)
 
@@ -55,7 +55,6 @@
     if end is None:
         end = start + datetime.timedelta(days = 1)
         end = datetime.datetime(end.year, end.month, end.day)
-
     dbh = sqlite3.connect(dbfn, detect_types = sqlite3.PARSE_DECLTYPES)
     cur = dbh.cursor()
     if args.update:
@@ -80,7 +79,9 @@
             date += datetime.timedelta(days = 1)
 
     if args.graph:
-        graph(args.filename, cur, ['battery_charge', 'battery_power', 'power_imported', 'power_exported', 'power_consumed', 'power_generated'], start, end)
+        graph(args.filename, cur,
+                  ['battery_charge', 'battery_power', 'power_imported', 'power_exported', 'power_consumed', 'power_generated'],
+                  start, end)
 
 def mkdb(cur):
     cur.execute('''
@@ -187,7 +188,8 @@
         start = start.replace(tzinfo = lt)
     if end.tzinfo == None:
         end = end.replace(tzinfo = lt)
-
+    startlt = start
+    endlt = end
     start = start.astimezone(utc)
     end = end.astimezone(utc)
 
@@ -264,7 +266,7 @@
                  ha = 'left', va = 'top')
     ndays = int(max(1, round(((end - start).total_seconds()) / 86400)))
     for ax in fig.get_axes():
-        ax.set_title('%s to %s' % (start.strftime('%Y-%m-%d'), end.strftime('%Y-%m-%d')))
+        ax.set_title('%s to %s' % (startlt.strftime('%Y-%m-%d'), endlt.strftime('%Y-%m-%d')))
         ax.set_xlim([start, end])
         ax.format_xdata = lambda d: matplotlib.dates.num2date(d).strftime('%d %b %H:%M')
         ax.xaxis.grid(True)
@@ -281,7 +283,7 @@
         matplotlib.pyplot.show()
     else:
         canvas = matplotlib.backends.backend_agg.FigureCanvasAgg(fig) # Sets canvas in fig too
-        fig.savefig(fname)
+        fig.savefig(startlt.strftime(fname))
 
 def updatedb(cur, data):
     mkdb(cur)