python-aidot 0.2.4__tar.gz → 0.2.6__tar.gz

{python_aidot-0.2.4 → python_aidot-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: python-aidot
-Version: 0.2.4
+Version: 0.2.6
 Summary: aidot control wifi lights
 Home-page: https://github.com/Aidot-Development-Team/python-aidot
 Author: aidotdev2024

python_aidot-0.2.6/aidot/client.py

@@ -0,0 +1,210 @@
+"""The aidot integration."""
+
+import logging
+from typing import Any, Optional
+from aiohttp import ClientResponseError, ClientSession
+import base64
+import aiohttp
+from cryptography.hazmat.backends import default_backend
+from cryptography.hazmat.primitives import serialization
+from cryptography.hazmat.primitives.asymmetric import padding
+from .login_const import APP_ID, PUBLIC_KEY_PEM, BASE_URL
+from .const import (
+    SUPPORTED_COUNTRYS,
+    DEFAULT_COUNTRY_NAME,
+    CONF_PRODUCT_ID,
+    CONF_ID,
+    CONF_PRODUCT,
+    CONF_ACCESS_TOKEN,
+    CONF_REFRESH_TOKEN,
+    CONF_TERMINAL,
+    CONF_APP_ID,
+    CONF_REGION,
+    CONF_COUNTRY,
+    CONF_USERNAME,
+    CONF_PASSWORD,
+    CONF_CODE,
+    CONF_TOKEN,
+    ServerErrorCode
+)
+from .exceptions import AidotAuthFailed,AidotUserOrPassIncorrect
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def rsa_password_encrypt(message: str):
+    """Get password rsa encrypt."""
+    public_key = serialization.load_pem_public_key(
+        PUBLIC_KEY_PEM, backend=default_backend()
+    )
+
+    encrypted = public_key.encrypt(
+        message.encode("utf-8"),
+        padding.PKCS1v15(),
+    )
+
+    encrypted_base64 = base64.b64encode(encrypted).decode("utf-8")
+    return encrypted_base64
+
+
+class AidotClient:
+    _base_url: str = BASE_URL
+    _region: str = "us"
+    session: Optional[ClientSession] = None
+    username: str = ""
+    password: str = ""
+    country_name: str = DEFAULT_COUNTRY_NAME
+    login_info: dict[str, Any] = {}
+
+    def __init__(
+        self,
+        session: Optional[ClientSession],
+        country_name: str | None = None,
+        username: str | None = None,
+        password: str | None = None,
+        token: dict | None = None,
+    ) -> None:
+        self.session = session
+        self.country_name = country_name
+        self.username = username
+        self.password = password
+        self.login_info = token
+        for item in SUPPORTED_COUNTRYS:
+            if item["name"] == self.country_name:
+                self._region = item["region"].lower()
+                self._base_url = f"https://prod-{self._region}-api.arnoo.com/v17"
+                break
+        if token is not None:
+            self.username = token[CONF_USERNAME]
+            self.password = token[CONF_PASSWORD]
+            self._region = token[CONF_REGION]
+            self.country_name = token[CONF_COUNTRY]
+
+    def set_token_fresh_cb(self, callback):
+        self._token_fresh_cb = callback
+
+    def get_identifier(self) -> str:
+        return f"{self._region}-{self.username}"
+    
+    def update_password(self, password: str):
+        self.password = password
+
+    async def async_post_login(self):
+        """Login the user input allows us to connect."""
+        url = f"{self._base_url}/users/loginWithFreeVerification"
+        headers = {CONF_APP_ID: APP_ID, CONF_TERMINAL: "app"}
+        data = {
+            "countryKey": "region:UnitedStates",
+            "username": self.username,
+            "password": rsa_password_encrypt(self.password),
+            "terminalId": "gvz3gjae10l4zii00t7y0",
+            "webVersion": "0.5.0",
+            "area": "Asia/Shanghai",
+            "UTC": "UTC+8",
+        }
+
+        try:
+            response = await self.session.post(url, headers=headers, json=data)
+            response_data = await response.json()
+            response.raise_for_status()
+            self.login_info = response_data
+            self.login_info[CONF_PASSWORD] = self.password
+            self.login_info[CONF_REGION] = self._region
+            self.login_info[CONF_COUNTRY] = self.country_name
+            return self.login_info
+        except aiohttp.ClientError as e:
+            _LOGGER.info(f"async_post_login ClientError {e}")
+            if response_data[CONF_CODE] == ServerErrorCode.USER_PWD_INCORRECT:
+                raise AidotUserOrPassIncorrect
+            return None
+
+    async def async_refresh_token(self):
+        url = f"{self._base_url}/users/refreshToken"
+        headers = {CONF_APP_ID: APP_ID, CONF_TERMINAL: "app"}
+        data = {
+            CONF_REFRESH_TOKEN: self.login_info[CONF_REFRESH_TOKEN],
+        }
+
+        try:
+            response = await self.session.post(url, headers=headers, json=data)
+            response_data = await response.json()
+            response.raise_for_status()
+            self.login_info[CONF_ACCESS_TOKEN] = response_data[CONF_ACCESS_TOKEN]
+            if response_data[CONF_REFRESH_TOKEN] is not None:
+                self.login_info[CONF_REFRESH_TOKEN] = response_data[CONF_REFRESH_TOKEN]
+            _LOGGER.info(f"refresh token {response_data}")
+            if self._token_fresh_cb:
+                self._token_fresh_cb()
+            return response_data
+        except aiohttp.ClientError as e:
+            _LOGGER.info(f"async_refresh_token ClientError {e}")
+            if response_data[CONF_CODE] == ServerErrorCode.LOGIN_INVALID:
+                raise AidotAuthFailed
+            return None
+
+    async def async_session_get(self, params: str, headers: str | None = None):
+        url = f"{self._base_url}{params}"
+        token = self.login_info[CONF_ACCESS_TOKEN]
+        if token is None:
+            raise AidotAuthFailed()
+        if headers is None:
+            headers = {
+                CONF_TERMINAL: "app",
+                CONF_TOKEN: token,
+                CONF_APP_ID: APP_ID,
+            }
+        try:
+            response = await self.session.get(url, headers=headers)
+            response_data = await response.json()
+            response.raise_for_status()
+            return response_data
+        except aiohttp.ClientError as e:
+            _LOGGER.info(f"async_get ClientError {e}")
+            code = response_data[CONF_CODE]
+            if code == ServerErrorCode.TOKEN_EXPIRED:
+                try:
+                    await self.async_refresh_token()
+                    return await self.async_session_get(params)
+                except AidotAuthFailed:
+                    raise AidotAuthFailed
+            elif code == ServerErrorCode.LOGIN_INVALID or code == 21027 or code == 21041:
+                self.login_info[CONF_ACCESS_TOKEN] = None
+                raise AidotAuthFailed
+            return None
+
+    async def async_get_products(self, product_ids: str):
+        """Get device list."""
+        params = f"/products/{product_ids}"
+        return await self.async_session_get(params)
+
+    async def async_get_devices(self, house_id: str):
+        """Get device list."""
+        params = f"/devices?houseId={house_id}"
+        return await self.async_session_get(params)
+
+    async def async_get_houses(self):
+        """Get house list."""
+        params = "/houses"
+        return await self.async_session_get(params)
+
+    async def async_get_all_device(self):
+        final_device_list: list[dict[str, Any]] = []
+        try:
+            houses = await self.async_get_houses()
+            for house in houses:
+                # get device_list
+                device_list = await self.async_get_devices(house[CONF_ID])
+                if device_list:
+                    final_device_list.extend(device_list)
+
+            # get product_list
+            productIds = ",".join([item[CONF_PRODUCT_ID] for item in final_device_list])
+            product_list = await self.async_get_products(productIds)
+
+            for product in product_list:
+                for device in final_device_list:
+                    if device[CONF_PRODUCT_ID] == product[CONF_ID]:
+                        device[CONF_PRODUCT] = product
+        except Exception as e:
+            raise e
+        return final_device_list

{python_aidot-0.2.4 → python_aidot-0.2.6}/aidot/const.py

@@ -1,4 +1,4 @@
-from enum import StrEnum
+from enum import StrEnum,IntEnum
 SUPPORTED_COUNTRYS = [
     {"_id": "1-0", "id": "AL", "name": "Albania", "ext": "", "region": "EU"},
     {
@@ -163,11 +163,20 @@ SUPPORTED_COUNTRYS = [
 
 SUPPORTED_COUNTRY_NAMES = [item["name"] for item in SUPPORTED_COUNTRYS]
 DEFAULT_COUNTRY_NAME = "United States"
+CONF_APP_ID = "Appid"
+CONF_TERMINAL = "Terminal"
 CONF_LOGIN_RESPONSE = "login_response"
+CONF_LOGIN_INFO = "login_info"
 CONF_SELECTED_HOUSE = "selected_house"
 CONF_DEVICE_LIST = "device_list"
 CONF_PRODUCT_LIST = "product_list"
 CONF_ACCESS_TOKEN = "accessToken"
+CONF_REFRESH_TOKEN = "refreshToken"
+CONF_TOKEN = "Token"
+CONF_USERNAME = "username"
+CONF_PASSWORD = "password"
+CONF_REGION = "region"
+CONF_COUNTRY = "country"
 CONF_ID = "id"
 CONF_NAME = "name"
 CONF_PRODUCT_ID = "productId"
@@ -185,7 +194,7 @@ CONF_PROPERTIES = "properties"
 CONF_MINVALUE = "minValue"
 CONF_MAXVALUE = "maxValue"
 CONF_IPADDRESS = "ipAddress"
-
+CONF_CODE = "code"
 class Identity(StrEnum):
     """Available entity identity."""
     RGBW = "control.light.rgbw"
@@ -194,4 +203,9 @@ class Identity(StrEnum):
 class Attribute(StrEnum):
     """Available entity attributes."""
     RGBW = "rgbw"
-    CCT = "cct"
+    CCT = "cct"
+
+class ServerErrorCode(IntEnum):
+    TOKEN_EXPIRED = 21026
+    LOGIN_INVALID = 21025
+    USER_PWD_INCORRECT = 560080

python_aidot-0.2.6/aidot/exceptions.py

@@ -0,0 +1,22 @@
+"""aidot Exceptions."""
+class AidotError(Exception):
+    """Aidot api exception."""
+
+class InvalidURL(AidotError):
+    """Invalid url exception."""
+
+class HTTPError(AidotError):
+    """Invalid host exception."""
+
+class InvalidHost(AidotError):
+    """Invalid host exception."""
+
+class AidotAuthTokenExpired(AidotError):
+    """Authentication failed because token is invalid or expired."""
+
+class AidotAuthFailed(AidotError):
+    """Authentication failed because MFA verification code is required."""
+
+class AidotUserOrPassIncorrect(AidotError):
+    """The password or email address is incorrect."""
+    

{python_aidot-0.2.4 → python_aidot-0.2.6}/python_aidot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: python-aidot
-Version: 0.2.4
+Version: 0.2.6
 Summary: aidot control wifi lights
 Home-page: https://github.com/Aidot-Development-Team/python-aidot
 Author: aidotdev2024

{python_aidot-0.2.4 → python_aidot-0.2.6}/python_aidot.egg-info/SOURCES.txt

@@ -4,8 +4,10 @@ setup.cfg
 setup.py
 aidot/__init__.py
 aidot/aes_utils.py
+aidot/client.py
 aidot/const.py
 aidot/discover.py
+aidot/exceptions.py
 aidot/lan.py
 aidot/login_const.py
 aidot/login_control.py

{python_aidot-0.2.4 → python_aidot-0.2.6}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="python-aidot",
-    version="0.2.4",
+    version="0.2.6",
     author="aidotdev2024",
     url='https://github.com/Aidot-Development-Team/python-aidot',
     description="aidot control wifi lights",