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 with HonAPI() as hon:
async with HonAPI(anonymous=True) as hon:
keys = await hon.translation_keys(language)
if json_output:
print(json.dumps(keys, indent=4))

View file

@ -10,24 +10,29 @@ _LOGGER = logging.getLogger()
class HonAPI:
def __init__(self, email="", password="", anonymous=False) -> None:
def __init__(self, email="", password="", anonymous=False, session=None) -> None:
super().__init__()
self._email = email
self._password = password
self._anonymous = anonymous
self._hon = None
self._hon_anonymous = None
self._session = session
async def __aenter__(self):
return await self.create()
async def __aexit__(self, exc_type, exc_val, exc_tb):
await self._hon.close()
await self.close()
async def create(self):
self._hon_anonymous = HonAnonymousConnectionHandler()
self._hon_anonymous = await HonAnonymousConnectionHandler(
self._session
).create()
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
async def load_appliances(self):
@ -145,4 +150,7 @@ class HonAPI:
return {}
async def close(self):
await self._hon.close()
if self._hon:
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:
_HEADERS = {"user-agent": const.USER_AGENT, "Content-Type": "application/json"}
def __init__(self):
self._session = None
def __init__(self, session=None):
self._session = session
self._auth = None
async def __aenter__(self):
@ -26,20 +26,26 @@ class HonBaseConnectionHandler:
return self
@asynccontextmanager
async def get(self, *args, **kwargs):
async def _intercept(self, method, *args, loop=0, **kwargs):
raise NotImplementedError
@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):
raise NotImplementedError
async with self._intercept(self._session.post, *args, **kwargs) as response:
yield response
async def close(self):
await self._session.close()
class HonConnectionHandler(HonBaseConnectionHandler):
def __init__(self, email, password):
super().__init__()
def __init__(self, email, password, session=None):
super().__init__(session=session)
self._device = HonDevice()
self._email = email
self._password = password
@ -108,26 +114,14 @@ class HonConnectionHandler(HonBaseConnectionHandler):
)
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):
_HEADERS = HonBaseConnectionHandler._HEADERS | {"x-api-key": const.API_KEY}
@asynccontextmanager
async def get(self, *args, **kwargs):
async with self._session.post(*args, **kwargs) as response:
yield response
@asynccontextmanager
async def post(self, *args, **kwargs):
async with self._session.post(*args, **kwargs) as response:
async def _intercept(self, method, *args, loop=0, **kwargs):
kwargs["headers"] = kwargs.pop("headers", {}) | self._HEADERS
async with method(*args, **kwargs) as response:
if response.status == 403:
print("Can't authorize")
yield response

View file

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