aioamazondevices 3.1.23__tar.gz → 3.2.1__tar.gz

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: aioamazondevices
-Version: 3.1.23
+Version: 3.2.1
 Summary: Python library to control Amazon devices
 License: Apache-2.0
 Author: Simone Chemelli

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aioamazondevices"
-version = "3.1.23"
+version = "3.2.1"
 requires-python = ">=3.12"
 description = "Python library to control Amazon devices"
 authors = [

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/src/aioamazondevices/__init__.py

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

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/src/aioamazondevices/api.py

@@ -32,6 +32,7 @@ from .const import (
     AMAZON_DEVICE_TYPE,
     BIN_EXTENSION,
     CSRF_COOKIE,
+    DEFAULT_AGENT,
     DEFAULT_ASSOC_HANDLE,
     DEFAULT_HEADERS,
     DEVICE_TO_IGNORE,
@@ -56,6 +57,7 @@ from .exceptions import (
     CannotConnect,
     CannotRegisterDevice,
     CannotRetrieveData,
+    WrongCountry,
     WrongMethod,
 )
 from .utils import obfuscate_email, scrub_fields
@@ -160,7 +162,17 @@ class AmazonEchoApi:
         if not self._login_stored_data:
             return {}
 
-        return cast("dict", self._login_stored_data["website_cookies"])
+        website_cookies: dict[str, Any] = self._login_stored_data["website_cookies"]
+        website_cookies.update(
+            {
+                "session-token": self._login_stored_data["store_authentication_cookie"][
+                    "cookie"
+                ]
+            }
+        )
+        website_cookies.update({"lc-acbit": self._language})
+
+        return website_cookies
 
     def _serial_number(self) -> str:
         """Get or calculate device serial number."""
@@ -324,6 +336,7 @@ class AmazonEchoApi:
         url: str,
         input_data: dict[str, Any] | None = None,
         json_data: bool = False,
+        amazon_user_agent: bool = True,
     ) -> tuple[BeautifulSoup, ClientResponse]:
         """Return request response context data."""
         _LOGGER.debug(
@@ -336,6 +349,9 @@ class AmazonEchoApi:
 
         headers = DEFAULT_HEADERS
         headers.update({"Accept-Language": self._language})
+        if not amazon_user_agent:
+            _LOGGER.debug("Changing User-Agent to %s", DEFAULT_AGENT)
+            headers.update({"User-Agent": DEFAULT_AGENT})
         if self._csrf_cookie and CSRF_COOKIE not in headers:
             csrf = {CSRF_COOKIE: self._csrf_cookie}
             _LOGGER.debug("Adding <%s> to headers", csrf)
@@ -525,11 +541,37 @@ class AmazonEchoApi:
         await self._save_to_file(login_data, "login_data", JSON_EXTENSION)
         return login_data
 
+    async def _check_country(self) -> None:
+        """Check if user selected country matches Amazon account country."""
+        url = f"https://alexa.amazon.{self._domain}/api/users/me"
+        _, resp_me = await self._session_request(HTTPMethod.GET, url)
+
+        if resp_me.status != HTTPStatus.OK:
+            raise CannotAuthenticate
+
+        resp_me_json = await resp_me.json()
+        market = resp_me_json["marketPlaceDomainName"]
+        language = resp_me_json["marketPlaceLocale"]
+
+        _domain = f"https://www.amazon.{self._domain}"
+
+        if market != _domain or language != self._language:
+            _LOGGER.debug(
+                "Selected country <%s> doesn't matches Amazon account:\n%s\n vs \n%s",
+                self._login_country_code.upper(),
+                {"site  ": _domain, "locale": self._language},
+                {"market": market, "locale": language},
+            )
+            raise WrongCountry
+
+        _LOGGER.debug("User selected country matches Amazon account one")
+
     async def _get_devices_ids(self) -> list[dict[str, str]]:
         """Retrieve devices entityId and applianceId."""
         _, raw_resp = await self._session_request(
             "GET",
             url=f"https://alexa.amazon.{self._domain}{URI_IDS}",
+            amazon_user_agent=False,
         )
         json_data = await raw_resp.json()
 
@@ -698,6 +740,9 @@ class AmazonEchoApi:
         self._login_stored_data = register_device
 
         _LOGGER.info("Register device: %s", scrub_fields(register_device))
+
+        await self._check_country()
+
         return register_device
 
     async def login_mode_stored_data(self) -> dict[str, Any]:
@@ -716,6 +761,8 @@ class AmazonEchoApi:
 
         self._client_session()
 
+        await self._check_country()
+
         return self._login_stored_data
 
     async def close(self) -> None:

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/src/aioamazondevices/const.py

@@ -86,6 +86,7 @@ DEFAULT_HEADERS = {
     "Accept-Encoding": "gzip",
     "Connection": "keep-alive",
 }
+DEFAULT_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36 Edg/138.0.0.0"  # noqa: E501
 CSRF_COOKIE = "csrf"
 
 NODE_DEVICES = "devices"

{aioamazondevices-3.1.23 → aioamazondevices-3.2.1}/src/aioamazondevices/exceptions.py

@@ -25,3 +25,7 @@ class CannotRegisterDevice(AmazonError):
 
 class WrongMethod(AmazonError):
     """Exception raised when the wrong login metho is used."""
+
+
+class WrongCountry(AmazonError):
+    """Exceptio nraised when Amazon country."""