Mercurial > ~darius > hgwebdir.cgi > vanlogger
annotate graph.py @ 6:10ae66476521
Ignore file
author | Daniel O'Connor <darius@dons.net.au> |
---|---|
date | Fri, 22 Dec 2017 13:04:21 +0100 |
parents | a7e9775b33f6 |
children | 30e7adf283ca |
rev | line source |
---|---|
4
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
1 #!/usr/bin/env python |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
2 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
3 import argparse |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
4 import datetime |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
5 import dateutil |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
6 import exceptions |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
7 import matplotlib |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
8 import matplotlib.dates |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
9 import numpy |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
10 import os |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
11 import requests |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
12 import sqlite3 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
13 import tzlocal |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
14 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
15 class Column(object): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
16 def __init__(self, rowname, title, table, units, limits = (None, None), conv = None): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
17 self.rowname = rowname |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
18 self.title = title |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
19 self.table = table |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
20 self.units = units |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
21 self.limits = limits |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
22 self.conv = None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
23 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
24 columns = [ |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
25 Column('main_voltage', 'Battery Voltage', 'eprolog', 'Vdc'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
26 Column('aux_voltage', 'Aux Voltage', 'eprolog', 'Vdc'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
27 Column('battery_curr', 'Battery Current', 'eprolog', 'A'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
28 Column('amp_hours', 'Battery Amp Hours', 'eprolog', 'Ah'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
29 Column('state_of_charge', 'State of Charge', 'eprolog', '%', (0, 100)), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
30 Column('time_remaining', 'Time Remaining', 'eprolog', 'min'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
31 Column('battery_temp', 'Battery Temperature', 'eprolog', 'C'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
32 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
33 Column('ac_act_power', 'Active Power', 'giantlog', 'W'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
34 Column('ac_app_power', 'Apparent Power', 'giantlog', 'W'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
35 Column('ac_frequency', 'AC Frequency', 'giantlog', 'Hz'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
36 Column('ac_volts', 'AC Voltage', 'giantlog', 'Vac'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
37 Column('batt_chr_curr', 'Discharge Current', 'giantlog', 'A'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
38 Column('batt_dis_curr', 'Charge Current', 'giantlog', 'A'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
39 Column('battery_cap', 'Battery Capacity', 'giantlog', '%', (0, 100)), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
40 Column('battery_volts', 'Battery Voltage', 'giantlog', 'Vdc'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
41 Column('grid_frequency', 'Grid Frequency', 'giantlog', 'Hz'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
42 Column('grid_volts', 'Grid Voltage', 'giantlog', 'Vac'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
43 Column('hs_temperature', 'HS Temperature', 'giantlog', 'C'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
44 Column('load_pct', 'Load', 'giantlog', '%', (0, 100)), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
45 ] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
46 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
47 def valid_date(s): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
48 try: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
49 return datetime.datetime.strptime(s, "%Y-%m-%d") |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
50 except ValueError: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
51 raise argparse.ArgumentTypeError("Not a valid date: '{0}'.".format(s)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
52 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
53 def main(): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
54 parser = argparse.ArgumentParser() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
55 parser.add_argument('-f', '--filename', help = 'Path to database', type = str, required = True) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
56 parser.add_argument('-g', '--graphfn', help = 'File to write graph to', type = str) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
57 parser.add_argument('-d', '--days', help = 'Days ago to graph', type = int) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
58 parser.add_argument('-s', '--start', help = 'Start date for graph (YYYY-MM-DD)', type = valid_date) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
59 parser.add_argument('-e', '--end', help = 'End date for graph (YYYY-MM-DD)', type = valid_date) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
60 parser.add_argument('-c', '--column', help = 'Column to plot (can be specified multiple times)', type = str, action = 'append') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
61 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
62 args = parser.parse_args() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
63 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
64 if args.days is not None and args.days < 0: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
65 parser.error('days must be non-negative') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
66 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
67 # Can specify.. |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
68 # Start and end |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
69 # Start and days |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
70 # End and days |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
71 # Nothing |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
72 # Want to end up with a start & end |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
73 if args.start is not None and args.end is not None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
74 pass |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
75 elif args.start is not None and args.days is not None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
76 args.end = args.start + datetime.timedelta(days = args.days) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
77 elif args.end is not None and args.days is not None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
78 args.start = args.end - datetime.timedelta(days = args.days) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
79 elif args.start is None and args.end is None and args.days is None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
80 end = datetime.date.today() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
81 end = datetime.datetime(start.year, start.month, start.day) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
82 args.start = args.end - datetime.timedelta(days = args.days) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
83 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
84 parser.error('can\'t specify days, start and end simultaneously') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
85 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
86 if args.start >= args.end: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
87 parser.error('Start must be before end') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
88 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
89 cols = args.column |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
90 if cols == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
91 cols = ['main_voltage', 'aux_voltage', 'ac_app_power'] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
92 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
93 dbh = sqlite3.connect(args.filename, detect_types = sqlite3.PARSE_DECLTYPES) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
94 cur = dbh.cursor() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
95 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
96 # Get local timezone name and convert start/end to it |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
97 # Why is this so hard... |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
98 ltname = tzlocal.get_localzone().zone |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
99 ltname = 'Australia/Adelaide' |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
100 lt = dateutil.tz.gettz(ltname) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
101 utc = dateutil.tz.gettz('UTC') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
102 matplotlib.rcParams['timezone'] = ltname |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
103 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
104 if args.start.tzinfo == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
105 args.start = args.start.replace(tzinfo = lt) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
106 if args.end.tzinfo == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
107 args.end = args.end.replace(tzinfo = lt) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
108 startlt = args.start |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
109 endlt = args.end |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
110 args.start = args.start.astimezone(utc) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
111 args.end = args.end.astimezone(utc) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
112 graph(args.graphfn, cur, cols, int(args.start.strftime('%s')), int(args.end.strftime('%s')), lt, utc) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
113 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
114 def graph(fname, cur, _cols, start, end, lt, utc): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
115 import numpy |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
116 import matplotlib |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
117 import matplotlib.dates |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
118 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
119 startdt = datetime.datetime.fromtimestamp(start).replace(tzinfo = utc).astimezone(lt) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
120 enddt = datetime.datetime.fromtimestamp(end).replace(tzinfo = utc).astimezone(lt) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
121 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
122 colourlist = ['b','g','r','c','m','y','k'] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
123 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
124 cols = [] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
125 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
126 yaxisunits1 = None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
127 yaxisunits2 = None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
128 ax1lines = [] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
129 ax2lines = [] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
130 colouridx = 0 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
131 for col in _cols: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
132 # Check the column exists |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
133 for c in columns: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
134 if col == c.rowname: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
135 cols.append(c) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
136 break |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
137 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
138 raise exceptions.Exception('Unknown column name ' + c) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
139 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
140 # Work out what axes we are using |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
141 if yaxisunits1 == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
142 yaxisunits1 = c.units |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
143 if yaxisunits2 == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
144 if c.units != yaxisunits1: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
145 yaxisunits2 = c.units |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
146 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
147 if c.units != yaxisunits1 and c.units != yaxisunits2: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
148 raise exceptions.Exception('Asked to graph >2 different units') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
149 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
150 for c in cols: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
151 # Get the data |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
152 cur.execute('SELECT tstamp, ' + c.rowname + ' FROM ' + c.table + ' WHERE tstamp > ? AND tstamp < ? ORDER BY tstamp', |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
153 (start, end)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
154 ary = numpy.array(cur.fetchall()) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
155 if ary.shape[0] == 0: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
156 print('No data for ' + c.rowname) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
157 return |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
158 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
159 # Create TZ naive from POSIX stamp, then convert to TZ aware UTC then adjust to local time |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
160 c.xdata = map(lambda f: datetime.datetime.fromtimestamp(f).replace(tzinfo = utc).astimezone(lt), ary[:,0]) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
161 c.ydata = ary[:,1] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
162 if c.conv != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
163 c.ydata = map(c.conv, c.ydata) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
164 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
165 scale_min, scale_max = c.limits |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
166 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
167 # DoD? |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
168 c.annotation = None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
169 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
170 # Work out which axis to plot on |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
171 if c.units == yaxisunits1: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
172 ax = ax1lines |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
173 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
174 ax = ax2lines |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
175 c.colour = colourlist[colouridx] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
176 colouridx += 1 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
177 ax.append(c) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
178 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
179 # Load the right backend for display or save |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
180 if fname == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
181 import matplotlib.pylab |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
182 fig = matplotlib.pylab.figure() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
183 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
184 import matplotlib.backends.backend_agg |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
185 fig = matplotlib.figure.Figure(figsize = (12, 6), dpi = 75) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
186 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
187 # Do the plot |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
188 ax1 = fig.add_subplot(111) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
189 ax1.set_ylabel(yaxisunits1) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
190 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
191 annotations = [] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
192 for line in ax1lines: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
193 ax1.plot(line.xdata, line.ydata, label = line.title, color = line.colour) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
194 if line.limits[0] != None or line.limits[1] != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
195 ax1.set_ylim(line.limits[0], line.limits[1]) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
196 if line.annotation != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
197 annotations.append(line.annotation) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
198 ax1.legend(loc = 'upper left') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
199 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
200 if len(ax2lines) > 0: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
201 ax2 = ax1.twinx() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
202 ax2.set_ylabel(yaxisunits2) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
203 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
204 for line in ax2lines: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
205 ax2.plot(line.xdata, line.ydata, label = line.title, color = line.colour) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
206 if line.limits[0] != None or line.limits[1] != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
207 ax2.set_ylim(line.limits[0], line.limits[1]) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
208 if line.annotation != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
209 annotations.append(line.annotation) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
210 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
211 ax2.legend(loc = 'upper right') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
212 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
213 if len(annotations) > 0: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
214 ax1.text(0.02, 0.9, reduce(lambda a, b: a + '\n' + b, annotations), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
215 transform = ax1.transAxes, bbox = dict(facecolor = 'red', alpha = 0.5), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
216 ha = 'left', va = 'top') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
217 ndays = int(max(1, round((end - start) / 86400))) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
218 for ax in fig.get_axes(): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
219 if (enddt - startdt).total_seconds() > 86400: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
220 ax.set_title('%s to %s' % (startdt.strftime('%Y-%m-%d'), enddt.strftime('%Y-%m-%d'))) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
221 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
222 ax.set_title('%s' % (startdt.strftime('%Y-%m-%d'))) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
223 ax.set_xlim([startdt, enddt]) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
224 ax.format_xdata = lambda d: matplotlib.dates.num2date(d).strftime('%d %b %H:%M') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
225 ax.xaxis.grid(True) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
226 ax.xaxis.set_major_formatter(matplotlib.dates.DateFormatter('%d %b\n%H:%M')) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
227 ax.xaxis.set_major_locator(matplotlib.dates.HourLocator(interval = 2 * ndays)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
228 ax.xaxis.set_minor_locator(matplotlib.dates.MinuteLocator(interval = 5 * ndays)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
229 for label in ax.get_xticklabels(): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
230 label.set_ha('center') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
231 label.set_rotation(90) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
232 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
233 # Fudge margins to give more graph and less space |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
234 fig.subplots_adjust(left = 0.10, right = 0.88, top = 0.95, bottom = 0.15) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
235 if fname == None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
236 matplotlib.pyplot.show() |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
237 else: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
238 canvas = matplotlib.backends.backend_agg.FigureCanvasAgg(fig) # Sets canvas in fig too |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
239 fig.savefig(startlt.strftime(fname)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
240 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
241 def updatedb(cur, data): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
242 mkdb(cur) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
243 for d in data['reads']['data']: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
244 ts = datetime.datetime.strptime(d['t_stamp'], '%Y-%m-%dT%H:%M:%SZ') |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
245 # Note we rename *energy* to *power* here to match what it actually means |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
246 vals = [ts, d['battery_charge'], d['battery_energy'], d['energy_consumed'], d['energy_expected'], d['energy_exported'], d['energy_generated'], |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
247 d['energy_imported'], d['estimated_savings'], d['pv_forecast'], d['pv_generation']['battery_energy'], |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
248 d['pv_generation']['grid_energy'], d['pv_generation']['site_energy'], d['site_consumption']['battery_energy'], |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
249 d['site_consumption']['grid_energy'], d['site_consumption']['pv_energy']] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
250 skip = True |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
251 for v in vals[1:]: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
252 if v != None: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
253 skip = False |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
254 break |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
255 if skip: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
256 print('Skipping empty record at ' + str(ts)) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
257 continue |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
258 cur.execute('INSERT OR IGNORE INTO agl(t_stamp, battery_charge, battery_power, power_consumed, power_expected, power_exported, power_generated, power_imported, estimated_savings, pv_forecast, pv_gen_battery, pv_gen_grid, pv_gen_site, site_cons_battery, site_cons_grid, site_cons_pv) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)', vals) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
259 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
260 def gettoken(username, password): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
261 authblob = json.encoder.JSONEncoder().encode({'email' : username, 'password' : password}) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
262 reply = requests.request('POST', loginurl, data = authblob, headers = {'Content-Type' : 'application/json'}) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
263 if reply.status_code != 200: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
264 return None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
265 return json.decoder.JSONDecoder().decode(reply.content)['access_token'] |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
266 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
267 def getdata(token, startdate, enddate): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
268 #print('getting ' + startdate.strftime('%Y-%m-%d')) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
269 reply = requests.request('GET', dataurl, params = { |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
270 'startDate' : startdate.strftime('%Y-%m-%d'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
271 'endDate' : enddate.strftime('%Y-%m-%d'), |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
272 'granularity' : 'Minute', |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
273 'metrics' : 'read', |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
274 'units' : 'W', |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
275 }, headers = { 'Authorization' : 'Bearer ' + token}) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
276 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
277 if reply.status_code != 200: |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
278 return None |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
279 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
280 return json.decoder.JSONDecoder().decode(reply.content) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
281 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
282 def logout(token): |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
283 reply = requests.request('GET', logouturl, headers = { 'Authorization' : 'Bearer ' + token}) |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
284 return reply.status_code == 200 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
285 |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
286 if __name__ == '__main__': |
a7e9775b33f6
Add graph script based on AGL code (but better)
Daniel O'Connor <darius@dons.net.au>
parents:
diff
changeset
|
287 main() |