mirrors/elicznik
1
0
Fork 0
mirror of https://github.com/mlesniew/elicznik.git synced 2025-12-21 13:23:06 +01:00
Code Issues Projects Releases Wiki Activity

Tauron API update

Browse Source
This commit is contained in:
Michał Leśniewski
2023-01-10 21:19:37 +01:00
parent 19dd0e52d9
commit 2e851b1955
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/elicznik/elicznik.py
View File

@@ -7,7 +7,7 @@ from .session import Session
class ELicznik:
LOGIN_URL = "https://logowanie.tauron-dystrybucja.pl/login"
CHART_URL = "https://elicznik.tauron-dystrybucja.pl/index/charts"
CHART_URL = "https://elicznik.tauron-dystrybucja.pl/energia/api"
def __init__(self, username, password):
self.username = username
@@ -32,23 +32,19 @@ class ELicznik:
def __exit__(self, exc_type, exc_val, exc_tb):
pass
def get_raw_readings(self, start_date, end_date=None):
def _get_raw_readings(self, type_, start_date, end_date=None):
end_date = end_date or start_date
return self.session.post(
data = self.session.post(
self.CHART_URL,
data={
# "dane[smartNr]": "?"
# "dane[chartDay]": date.strftime("%d.%m.%Y"),
"dane[paramType]": "csv",
"dane[trybCSV]": "godzin",
"dane[startDay]": start_date.strftime("%d.%m.%Y"),
"dane[endDay]": end_date.strftime("%d.%m.%Y"),
"dane[checkOZE]": "on",
"type": type_,
"from": start_date.strftime("%d.%m.%Y"),
"to": end_date.strftime("%d.%m.%Y"),
"profile": "full time",
},
).json()
@staticmethod
def _extract_values_with_timestamps(data):
data = data.get("data", {}).get("allData", {})
for element in data:
date = element.get("Date")
hour = int(element.get("Hour"))
@@ -59,10 +55,15 @@ class ELicznik:
value = element.get("EC")
yield timestamp, value
def get_readings_production(self, start_date, end_date=None):
return dict(self._get_raw_readings("oze", start_date, end_date))
def get_readings_consumption(self, start_date, end_date=None):
return dict(self._get_raw_readings("consum", start_date, end_date))
def get_readings(self, start_date, end_date=None):
data = self.get_raw_readings(start_date, end_date).get("dane", {})
consumed = dict(self._extract_values_with_timestamps(data.get("chart", [])))
produced = dict(self._extract_values_with_timestamps(data.get("OZE", [])))
consumed = self.get_readings_production(start_date, end_date)
produced = self.get_readings_consumption(start_date, end_date)
return sorted(
(timestamp, float(consumed.get(timestamp)), float(produced.get(timestamp)))
for timestamp in set(consumed) | set(produced)