aboutsummaryrefslogtreecommitdiffstats
path: root/scripts/elvia_gridtariff.py
blob: 3e8ed1e2e08bb3338fb2c3bdf14632295dcd4950 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#!/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)