#!/usr/bin/python3
""" get grid tariffs"""
import sys
from datetime import datetime, timedelta
import common
import requests
from tzlocal import get_localzone
# API documentation: https://elvia.portal.azure-api.net/docs/services/gridtariffapi/operations/post-digin-api-v-tariffquery-meteringpointsgridtariffs?
apiKey = common.env("el_elvia_grid_api_key")
apiUrl = "https://elvia.azure-api.net/grid-tariff/digin/api/1/tariffquery/meteringpointsgridtariffs"
meteringPointId = common.env("el_meteringPointId")
startTime = datetime.now(get_localzone()) - timedelta(days=2)
startTime = startTime.strftime("%Y-%m-%d")
# startTime = '2023-05-23' # <- edit for manual starttime. Like when filling in missing info.
endTime = datetime.now(get_localzone()) + timedelta(days=2)
endTime = endTime.strftime("%Y-%m-%d")
### Get the data
try:
url = apiUrl
# Request headers
hdr = {
"Cache-Control": "no-cache",
"X-API-Key": apiKey,
"Content-Type": "application/json",
}
# Request body
body = {
"starttime": startTime,
"endtime": endTime,
"meteringPointIds": [meteringPointId],
}
response = requests.post(url, headers=hdr, json=body, timeout=10)
response.raise_for_status()
except requests.exceptions.RequestException as e:
sys.exit(e)
data = response.json()
num_items = len(data["gridTariffCollections"][0]["gridTariff"]["tariffPrice"]["hours"])
print(f"Got {num_items} items from between {startTime} and {endTime}")
### insert data into database
values = []
for item in data["gridTariffCollections"][0]["gridTariff"]["tariffPrice"]["hours"]:
values.append(
(
meteringPointId,
item["startTime"],
item["expiredAt"],
item["shortName"],
item["isPublicHoliday"],
item["energyPrice"]["total"],
item["energyPrice"]["totalExVat"],
)
)
# SQL
sql = """INSERT INTO elvia_gridtariff
VALUES(%s, %s, %s, %s, %s, %s, %s)
ON CONFLICT (meteringPointId,startTime,endTime) DO NOTHING"""
common.dbi(sql, values, verbose=True)
