aioamazondevices 1.1.0__tar.gz → 1.3.0__tar.gz

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: aioamazondevices
-Version: 1.1.0
+Version: 1.3.0
 Summary: Python library to control Amazon devices
 License: Apache-2.0
 Author: Simone Chemelli
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries
+Requires-Dist: babel
 Requires-Dist: beautifulsoup4
 Requires-Dist: colorlog
 Requires-Dist: httpx
@@ -69,12 +70,23 @@ Install this via pip (or your favourite package manager):
 
 `pip install aioamazondevices`
 
-## Usage
+## Test
 
-Start by importing it:
+Test the library with:
 
-```python
-import aioamazondevices
+`python library_test.py`
+
+The script accept command line arguments or a library_test.json config file:
+
+```json
+{
+  "country": "IT",
+  "email": "<my_address@gmail.com>",
+  "password": "<my_password>",
+  "device_name": "Echo Dot Livingroom",
+  "login_data_file": "out/login_data.json",
+  "save_raw_data": "True"
+}
 ```
 
 ## Contributors ✨

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/README.md

@@ -41,12 +41,23 @@ Install this via pip (or your favourite package manager):
 
 `pip install aioamazondevices`
 
-## Usage
+## Test
 
-Start by importing it:
+Test the library with:
 
-```python
-import aioamazondevices
+`python library_test.py`
+
+The script accept command line arguments or a library_test.json config file:
+
+```json
+{
+  "country": "IT",
+  "email": "<my_address@gmail.com>",
+  "password": "<my_password>",
+  "device_name": "Echo Dot Livingroom",
+  "login_data_file": "out/login_data.json",
+  "save_raw_data": "True"
+}
 ```
 
 ## Contributors ✨

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "aioamazondevices"
-version = "1.1.0"
+version = "1.3.0"
 description = "Python library to control Amazon devices"
 authors = ["Simone Chemelli <simone.chemelli@gmail.com>"]
 license = "Apache-2.0"
@@ -23,6 +23,7 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.12"
+babel = "*"
 beautifulsoup4 = "*"
 colorlog = "*"
 httpx = "*"

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/src/aioamazondevices/__init__.py

@@ -1,6 +1,6 @@
 """aioamazondevices library."""
 
-__version__ = "1.1.0"
+__version__ = "1.3.0"
 
 
 from .api import AmazonDevice, AmazonEchoApi

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/src/aioamazondevices/api.py

@@ -13,6 +13,7 @@ from typing import Any, cast
 from urllib.parse import parse_qs, urlencode
 
 import orjson
+from babel import Locale
 from bs4 import BeautifulSoup, Tag
 from httpx import URL, AsyncClient, Response
 
@@ -25,6 +26,8 @@ from .const import (
     AMAZON_CLIENT_OS,
     AMAZON_DEVICE_SOFTWARE_VERSION,
     AMAZON_DEVICE_TYPE,
+    BIN_EXTENSION,
+    CSRF_COOKIE,
     DEFAULT_ASSOC_HANDLE,
     DEFAULT_HEADERS,
     DOMAIN_BY_ISO3166_COUNTRY,
@@ -48,6 +51,7 @@ class AmazonDevice:
     capabilities: list[str]
     device_family: str
     device_type: str
+    device_owner_customer_id: str
     online: bool
     serial_number: str
     software_version: str
@@ -82,13 +86,14 @@ class AmazonEchoApi:
 
         self._login_email = login_email
         self._login_password = login_password
+        self._login_country_code = country_code
         self._domain = domain
         self._cookies = self._build_init_cookies()
+        self._csrf_cookie: str | None = None
         self._headers = DEFAULT_HEADERS
         self._save_raw_data = save_raw_data
         self._login_stored_data = login_data
         self._serial = self._serial_number()
-        self._website_cookies: dict[str, Any] = self._load_website_cookies()
 
         self.session: AsyncClient
 
@@ -232,12 +237,24 @@ class AmazonEchoApi:
             input_data,
             json_data,
         )
+
+        headers = DEFAULT_HEADERS
+        if self._csrf_cookie and CSRF_COOKIE not in headers:
+            csrf = {CSRF_COOKIE: self._csrf_cookie}
+            _LOGGER.debug("Adding <%s> to headers", csrf)
+            headers.update(csrf)
+
+        if json_data:
+            json_header = {"Content-Type": "application/json"}
+            _LOGGER.debug("Adding %s to headers", json_header)
+            headers.update(json_header)
+
         resp = await self.session.request(
             method,
             url,
             data=input_data if not json_data else orjson.dumps(input_data),
-            cookies=self._website_cookies,
-            headers={"Content-Type": "application/json"} if json_data else None,
+            cookies=self._load_website_cookies(),
+            headers=headers,
         )
         content_type: str = resp.headers.get("Content-Type", "")
         _LOGGER.debug(
@@ -278,7 +295,7 @@ class AmazonEchoApi:
 
         if type(raw_data) is dict:
             data = orjson.dumps(raw_data, option=orjson.OPT_INDENT_2).decode("utf-8")
-        elif extension == HTML_EXTENSION:
+        elif extension in [HTML_EXTENSION, BIN_EXTENSION]:
             data = raw_data
         else:
             data = orjson.dumps(
@@ -340,8 +357,8 @@ class AmazonEchoApi:
 
         register_url = f"https://api.amazon.{self._domain}/auth/register"
         _, resp = await self._session_request(
-            "POST",
-            register_url,
+            method="POST",
+            url=register_url,
             input_data=body,
             json_data=True,
         )
@@ -404,7 +421,7 @@ class AmazonEchoApi:
         _LOGGER.debug("Build oauth URL")
         login_url = self._build_oauth_url(code_verifier, client_id)
 
-        login_soup, _ = await self._session_request("GET", login_url)
+        login_soup, _ = await self._session_request(method="GET", url=login_url)
         login_method, login_url = self._get_request_from_soup(login_soup)
         login_inputs = self._get_inputs_from_soup(login_soup)
         login_inputs["email"] = self._login_email
@@ -412,9 +429,9 @@ class AmazonEchoApi:
 
         _LOGGER.debug("Register at %s", login_url)
         login_soup, _ = await self._session_request(
-            login_method,
-            login_url,
-            login_inputs,
+            method=login_method,
+            url=login_url,
+            input_data=login_inputs,
         )
 
         if not login_soup.find("input", id="auth-mfa-otpcode"):
@@ -431,9 +448,9 @@ class AmazonEchoApi:
         login_inputs["rememberDevice"] = "false"
 
         login_soup, login_resp = await self._session_request(
-            login_method,
-            login_url,
-            login_inputs,
+            method=login_method,
+            url=login_url,
+            input_data=login_inputs,
         )
 
         authcode_url = None
@@ -456,6 +473,7 @@ class AmazonEchoApi:
         }
 
         register_device = await self._register_device(device_login_data)
+        self._login_stored_data = register_device
 
         _LOGGER.info("Register device: %s", register_device)
         return register_device
@@ -465,7 +483,7 @@ class AmazonEchoApi:
         if not self._login_stored_data:
             _LOGGER.debug(
                 "Cannot find previous login data,\
-                    use login_interactive() method instead",
+                    use login_mode_interactive() method instead",
             )
             raise WrongMethod
 
@@ -491,15 +509,19 @@ class AmazonEchoApi:
         devices: dict[str, Any] = {}
         for key in URI_QUERIES:
             _, raw_resp = await self._session_request(
-                "GET",
-                f"https://alexa.amazon.{self._domain}{URI_QUERIES[key]}",
+                method="GET",
+                url=f"https://alexa.amazon.{self._domain}{URI_QUERIES[key]}",
             )
             _LOGGER.debug("Response URL: %s", raw_resp.url)
             response_code = raw_resp.status_code
-            _LOGGER.debug("Response code: %s", response_code)
+            _LOGGER.debug("Response code: |%s|", response_code)
 
             response_data = raw_resp.text
             _LOGGER.debug("Response data: |%s|", response_data)
+
+            if not self._csrf_cookie:
+                self._csrf_cookie = raw_resp.cookies.get(CSRF_COOKIE)
+
             json_data = {} if len(response_data) == 0 else raw_resp.json()
 
             _LOGGER.debug("JSON data: |%s|", json_data)
@@ -527,6 +549,7 @@ class AmazonEchoApi:
                 capabilities=device[NODE_DEVICES]["capabilities"],
                 device_family=device[NODE_DEVICES]["deviceFamily"],
                 device_type=device[NODE_DEVICES]["deviceType"],
+                device_owner_customer_id=device[NODE_DEVICES]["deviceOwnerCustomerId"],
                 online=device[NODE_DEVICES]["online"],
                 serial_number=serial_number,
                 software_version=device[NODE_DEVICES]["softwareVersion"],
@@ -536,3 +559,83 @@ class AmazonEchoApi:
             )
 
         return final_devices_list
+
+    async def auth_check_status(self) -> bool:
+        """Check AUTH status."""
+        _, raw_resp = await self._session_request(
+            method="GET",
+            url=f"https://alexa.amazon.{self._domain}/api/bootstrap?version=0",
+        )
+        if raw_resp.status_code != HTTPStatus.OK:
+            _LOGGER.debug(
+                "Session not authenticated: reply error %s",
+                raw_resp.status_code,
+            )
+            return False
+
+        resp_json = raw_resp.json()
+        if not (authentication := resp_json.get("authentication")):
+            _LOGGER.debug('Session not authenticated: reply missing "authentication"')
+            return False
+
+        authenticated = authentication.get("authenticated")
+        _LOGGER.debug("Session authenticated: %s", authenticated)
+        return bool(authenticated)
+
+    async def call_alexa_speak(
+        self,
+        device: AmazonDevice,
+        message_body: str,
+    ) -> dict[str, Any]:
+        """Call Alexa.Speak to send a message."""
+        locale_data = Locale.parse(f"und_{self._login_country_code}")
+        locale = f"{locale_data.language}-{locale_data.language}"
+
+        if not self._login_stored_data:
+            _LOGGER.warning("Trying to send message before login")
+            return {}
+
+        sequence = {
+            "@type": "com.amazon.alexa.behaviors.model.Sequence",
+            "startNode": {
+                "@type": "com.amazon.alexa.behaviors.model.SerialNode",
+                "nodesToExecute": [
+                    {
+                        "@type": "com.amazon.alexa.behaviors.model.OpaquePayloadOperationNode",  # noqa: E501
+                        "type": "Alexa.Speak",
+                        "operationPayload": {
+                            "deviceType": device.device_type,
+                            "deviceSerialNumber": device.serial_number,
+                            "locale": locale,
+                            "customerId": device.device_owner_customer_id,
+                            "textToSpeak": message_body,
+                            "target": {
+                                "customerId": device.device_owner_customer_id,
+                                "devices": [
+                                    {
+                                        "deviceSerialNumber": device.serial_number,
+                                        "deviceTypeId": device.device_type,
+                                    },
+                                ],
+                            },
+                            "skillId": "amzn1.ask.1p.saysomething",
+                        },
+                    },
+                ],
+            },
+        }
+        node_data = {
+            "behaviorId": "PREVIEW",
+            "sequenceJson": orjson.dumps(sequence).decode("utf-8"),
+            "status": "ENABLED",
+        }
+
+        _LOGGER.debug("Preview data payload: %s", node_data)
+        await self._session_request(
+            method="POST",
+            url=f"https://alexa.amazon.{self._domain}/api/behaviors/preview",
+            input_data=node_data,
+            json_data=True,
+        )
+
+        return node_data

{aioamazondevices-1.1.0 → aioamazondevices-1.3.0}/src/aioamazondevices/const.py

@@ -43,6 +43,7 @@ DEFAULT_HEADERS = {
     "Accept-Language": "en-US",
     "Accept-Encoding": "gzip",
 }
+CSRF_COOKIE = "csrf"
 
 NODE_DEVICES = "devices"
 NODE_DO_NOT_DISTURB = "doNotDisturbDeviceStatusList"
@@ -60,6 +61,7 @@ URI_QUERIES = {
 SAVE_PATH = "out"
 HTML_EXTENSION = ".html"
 JSON_EXTENSION = ".json"
+BIN_EXTENSION = ".bin"
 
 DEVICE_TYPE_TO_MODEL = {
     "A1RABVCI4QCIKC": "Echo Dot (Gen3)",