Fix translation command

This commit is contained in:
Andre Basche 2023-04-10 06:34:19 +02:00
parent e8531f3faf
commit 3e3fc7ad66
4 changed files with 45 additions and 34 deletions

View file

@ -90,7 +90,7 @@ def create_command(commands, concat=False):
async def translate(language, json_output=False): async def translate(language, json_output=False):
async with HonAPI() as hon: async with HonAPI(anonymous=True) as hon:
keys = await hon.translation_keys(language) keys = await hon.translation_keys(language)
if json_output: if json_output:
print(json.dumps(keys, indent=4)) print(json.dumps(keys, indent=4))

View file

@ -10,24 +10,29 @@ _LOGGER = logging.getLogger()
class HonAPI: class HonAPI:
def __init__(self, email="", password="", anonymous=False) -> None: def __init__(self, email="", password="", anonymous=False, session=None) -> None:
super().__init__() super().__init__()
self._email = email self._email = email
self._password = password self._password = password
self._anonymous = anonymous self._anonymous = anonymous
self._hon = None self._hon = None
self._hon_anonymous = None self._hon_anonymous = None
self._session = session
async def __aenter__(self): async def __aenter__(self):
return await self.create() return await self.create()
async def __aexit__(self, exc_type, exc_val, exc_tb): async def __aexit__(self, exc_type, exc_val, exc_tb):
await self._hon.close() await self.close()
async def create(self): async def create(self):
self._hon_anonymous = HonAnonymousConnectionHandler() self._hon_anonymous = await HonAnonymousConnectionHandler(
self._session
).create()
if not self._anonymous: if not self._anonymous:
self._hon = await HonConnectionHandler(self._email, self._password).create() self._hon = await HonConnectionHandler(
self._email, self._password, self._session
).create()
return self return self
async def load_appliances(self): async def load_appliances(self):
@ -145,4 +150,7 @@ class HonAPI:
return {} return {}
async def close(self): async def close(self):
if self._hon:
await self._hon.close() await self._hon.close()
if self._hon_anonymous:
await self._hon_anonymous.close()

View file

@ -11,8 +11,8 @@ from pyhon.connection.device import HonDevice
class HonBaseConnectionHandler: class HonBaseConnectionHandler:
_HEADERS = {"user-agent": const.USER_AGENT, "Content-Type": "application/json"} _HEADERS = {"user-agent": const.USER_AGENT, "Content-Type": "application/json"}
def __init__(self): def __init__(self, session=None):
self._session = None self._session = session
self._auth = None self._auth = None
async def __aenter__(self): async def __aenter__(self):
@ -26,20 +26,26 @@ class HonBaseConnectionHandler:
return self return self
@asynccontextmanager @asynccontextmanager
async def get(self, *args, **kwargs): async def _intercept(self, method, *args, loop=0, **kwargs):
raise NotImplementedError raise NotImplementedError
@asynccontextmanager
async def get(self, *args, **kwargs):
async with self._intercept(self._session.get, *args, **kwargs) as response:
yield response
@asynccontextmanager @asynccontextmanager
async def post(self, *args, **kwargs): async def post(self, *args, **kwargs):
raise NotImplementedError async with self._intercept(self._session.post, *args, **kwargs) as response:
yield response
async def close(self): async def close(self):
await self._session.close() await self._session.close()
class HonConnectionHandler(HonBaseConnectionHandler): class HonConnectionHandler(HonBaseConnectionHandler):
def __init__(self, email, password): def __init__(self, email, password, session=None):
super().__init__() super().__init__(session=session)
self._device = HonDevice() self._device = HonDevice()
self._email = email self._email = email
self._password = password self._password = password
@ -108,26 +114,14 @@ class HonConnectionHandler(HonBaseConnectionHandler):
) )
yield {} yield {}
@asynccontextmanager
async def get(self, *args, **kwargs):
async with self._intercept(self._session.get, *args, **kwargs) as response:
yield response
@asynccontextmanager
async def post(self, *args, **kwargs):
async with self._intercept(self._session.post, *args, **kwargs) as response:
yield response
class HonAnonymousConnectionHandler(HonBaseConnectionHandler): class HonAnonymousConnectionHandler(HonBaseConnectionHandler):
_HEADERS = HonBaseConnectionHandler._HEADERS | {"x-api-key": const.API_KEY} _HEADERS = HonBaseConnectionHandler._HEADERS | {"x-api-key": const.API_KEY}
@asynccontextmanager @asynccontextmanager
async def get(self, *args, **kwargs): async def _intercept(self, method, *args, loop=0, **kwargs):
async with self._session.post(*args, **kwargs) as response: kwargs["headers"] = kwargs.pop("headers", {}) | self._HEADERS
yield response async with method(*args, **kwargs) as response:
if response.status == 403:
@asynccontextmanager print("Can't authorize")
async def post(self, *args, **kwargs):
async with self._session.post(*args, **kwargs) as response:
yield response yield response

View file

@ -6,19 +6,25 @@ from pyhon.appliance import HonAppliance
class Hon: class Hon:
def __init__(self, email, password): def __init__(self, email, password, session=None):
self._email = email self._email = email
self._password = password self._password = password
self._session = session
self._appliances = [] self._appliances = []
self._api = None self._api = None
async def __aenter__(self): async def __aenter__(self):
self._api = await HonAPI(self._email, self._password).create() return await self.create()
await self.setup()
return self
async def __aexit__(self, exc_type, exc_val, exc_tb): async def __aexit__(self, exc_type, exc_val, exc_tb):
await self._api.close() await self.close()
async def create(self):
self._api = await HonAPI(
self._email, self._password, session=self._session
).create()
await self.setup()
return self
@property @property
def appliances(self) -> List[HonAppliance]: def appliances(self) -> List[HonAppliance]:
@ -37,3 +43,6 @@ class Hon:
] ]
) )
self._appliances.append(appliance) self._appliances.append(appliance)
async def close(self):
await self._api.close()