diff --git a/get_user_token.py b/Inverter.py similarity index 50% rename from get_user_token.py rename to Inverter.py index 1ae027d..30c4130 100644 --- a/get_user_token.py +++ b/Inverter.py @@ -1,32 +1,32 @@ +import asyncio import logging -import os.path + import requests from requests import Response - -def generate_user_token(username: str, password: str) -> str: - url = "https://app.api.apsystemsema.com:9223/api/token/generateToken/user/login" - - data = { - "password": password, - "app_id": "4029817264d4821d0164d4821dd80015", - "app_secret": "EZAd2023", - "username": username - } - post = requests.post(url, data=data) - return post.json()["data"]['access_token'] +from InverterData import InverterData -def cache_user_token_on_disk(user_token) -> None: - with open("user_token", "w") as f: - f.write(user_token) +class Inverter: + def __init__(self, queue: asyncio.Queue, inverter_id, user_token): + self._queue = queue + self.inverter_id = inverter_id + self.user_token = user_token + self.gather_data = False + async def start(self): + self.gather_data = True + await self.gather() -def get_cached_user_token() -> str: - if os.path.exists("user_token"): - with open("user_token", "r") as f: - return f.read() - return "" + async def gather(self): + while self.gather_data: + realtime_data: dict = get_realtime_data(self.inverter_id, self.user_token) + data = InverterData( + self.inverter_id, + realtime_data + ) + await self._queue.put(data) + await asyncio.sleep(60) def get_realtime_data(inverter_id: str, user_token: str) -> dict: @@ -49,14 +49,3 @@ def handle_api_return_codes(result: Response) -> [dict, None]: exit(1) else: logging.error(f"Something went wrong in the request. Status code: {result.status_code}") - - -token = get_cached_user_token() -if not token: - token = generate_user_token("stefan@regnery.biz", "muffbaer21") - cache_user_token_on_disk(token) - - -inverter = "E07000000405" - -print(get_realtime_data(inverter, token)) diff --git a/InverterData.py b/InverterData.py new file mode 100644 index 0000000..b49e374 --- /dev/null +++ b/InverterData.py @@ -0,0 +1,7 @@ +from dataclasses import dataclass + + +@dataclass +class InverterData: + name: str + data: dict diff --git a/Usertoken.py b/Usertoken.py new file mode 100644 index 0000000..3468289 --- /dev/null +++ b/Usertoken.py @@ -0,0 +1,27 @@ +import os +import requests + + +def generate_user_token(username: str, password: str) -> str: + url = "https://app.api.apsystemsema.com:9223/api/token/generateToken/user/login" + + data = { + "password": password, + "app_id": "4029817264d4821d0164d4821dd80015", + "app_secret": "EZAd2023", + "username": username + } + post = requests.post(url, data=data) + return post.json()["data"]['access_token'] + + +def cache_user_token_on_disk(user_token) -> None: + with open("user_token", "w") as f: + f.write(user_token) + + +def get_cached_user_token() -> str: + if os.path.exists("user_token"): + with open("user_token", "r") as f: + return f.read() + return "" diff --git a/config.toml-default b/config.toml-default new file mode 100644 index 0000000..5403e1f --- /dev/null +++ b/config.toml-default @@ -0,0 +1,2 @@ +username = "" +password = "" \ No newline at end of file diff --git a/run.py b/run.py new file mode 100644 index 0000000..cb29f68 --- /dev/null +++ b/run.py @@ -0,0 +1,34 @@ +import asyncio +import os +import toml + +from Inverter import Inverter +from Usertoken import generate_user_token, cache_user_token_on_disk, get_cached_user_token + + +async def main(): + os.path.dirname(__file__) + with open("config.toml", "r") as f: + config = toml.load(f) + + token = get_cached_user_token() + + if not token: + token = generate_user_token(config["username"], config["password"]) + cache_user_token_on_disk(token) + + inverter_stefan = "E07000000405" + inverter_christian = "E07000000083" + + queue = asyncio.Queue() + inverters = [Inverter(queue, inverter_christian, token)] + for inverter in inverters: + asyncio.create_task(inverter.start()) + + while True: + item = await queue.get() + print(item) + + +if __name__ == '__main__': + asyncio.run(main())