Grab a month at a time (Still need to merge!)

py/downkwh.py

@@ -1,13 +1,28 @@
 from util import getQuery, pickleQuery, getkwh
 import pandas as p
-# import matplotlib
-# matplotlib.use('agg')
-# import matplotlib.pyplot as plt
-# import seaborn as sns
+import gc
 
-kwhdata = getkwh('2017-01-01', '2017-02-01', '2017-01-01 00:30:00', '2017-02-01 00:00:00', '%%1')
+months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
+mstarts = list(range(1, 13))
+mends = mstarts[1:13]
+mends.append(1)
 
-print(kwhdata.info())
+print(months)
+print(mstarts)
+print(mends)
+
+for i, m in enumerate(months):
+    print(i)
+    print(m)
+    kwhdata = getkwh('2017-{:02d}-01'.format(mstarts[i]), 
+                     '2017-{:02d}-01'.format(mends[i]), 
+                     '2017-{:02d}-01 00:30:00'.format(mstarts[i]), 
+                     '2017-{:02d}-01 00:00:00'.format(mends[i]), 
+                     '%%1')
+    print("Pickling")
+    kwhdata.to_pickle('../data/2017-{}-all.pkl'.format(m))
+    del kwhdata
+    gc.collect()
+
+print('Done')
 
-print("Pickling")
-kwhdata.to_pickle("../data/2017-Jan-all.pkl")