diff options
Diffstat (limited to 'scripts/elvia.py')
-rw-r--r-- | scripts/elvia.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/scripts/elvia.py b/scripts/elvia.py new file mode 100644 index 0000000..be5eaae --- /dev/null +++ b/scripts/elvia.py @@ -0,0 +1,57 @@ +#!/usr/bin/env python3 +''' elvia2pgsql ''' + +import os +import sys +import requests + +from datetime import datetime +from datetime import timedelta +from tzlocal import get_localzone + +from common import dbi + +apiKey = os.environ['el_elvia_token'] +apiUrl = "https://elvia.azure-api.net/customer/metervalues/api/v1/metervalues" + +pg_table = "elvia" + +startTime = datetime.now(get_localzone()) - timedelta(days = 2) +startTime = startTime.isoformat('T') +endTime = datetime.now(get_localzone()).isoformat('T') + + +### Get the data +try: + url = apiUrl + "?startTime=" + startTime + "&endTime=" + endTime + + # Request headers + hdr = { + 'Cache-Control': 'no-cache', + 'Authorization': "Bearer " + apiKey, + } + + response = requests.get(url, headers=hdr) + if response.status_code != 200: + print(response.status_code) + print("Oh shit") + response.raise_for_status() + +except Exception as e: + print("oh lol") + sys.exit(e) + +data = response.json() +print("Got "+ str(len(data['meteringpoints'][0]['metervalue']['timeSeries'])) +" items from between "+ startTime + " and " + endTime) + +### insert data into database + +values = [] +for item in data['meteringpoints'][0]['metervalue']['timeSeries']: + # Only deal with verified items. + if item['verified']: + values.append((data['meteringpoints'][0]['meteringPointId'], item['startTime'], item['endTime'], item['value'], item['uom'], item['production'])) + + +# Count new rows and insert +dbi("INSERT INTO " + pg_table + " VALUES(%s,%s,%s,%s,%s,%s) ON CONFLICT (startTime,endTime) DO NOTHING", values, verbose=True) |