pymammotion 0.4.35__py3-none-any.whl → 0.4.37__py3-none-any.whl

pymammotion/http/http.py

@@ -3,9 +3,9 @@ from typing import cast
 
 from aiohttp import ClientSession
 
-from pymammotion.aliyun.model.stream_subscription_response import StreamSubscriptionResponse
 from pymammotion.const import MAMMOTION_API_DOMAIN, MAMMOTION_CLIENT_ID, MAMMOTION_CLIENT_SECRET, MAMMOTION_DOMAIN
 from pymammotion.http.encryption import EncryptionUtils
+from pymammotion.http.model.camera_stream import StreamSubscriptionResponse, VideoResourceResponse
 from pymammotion.http.model.http import ErrorInfo, LoginResponseData, Response
 
 
@@ -108,6 +108,51 @@ class MammotionHTTP:
                 # Assuming the data format matches the expected structure
                 return Response[StreamSubscriptionResponse].from_dict(data)
 
+    async def get_stream_subscription_mini_or_x_series(
+        self, iot_id: str, is_yuka: bool
+    ) -> Response[StreamSubscriptionResponse]:
+        """Get agora.io data for view camera stream (New models 2025)"""
+
+        # Prepare the payload with cameraStates based on is_yuka flag
+        payload = {"deviceId": iot_id, "mode": 0, "cameraStates": []}
+
+        # Add appropriate cameraStates based on the is_yuka flag
+        if is_yuka:
+            payload["cameraStates"] = [{"cameraState": 1}, {"cameraState": 0}, {"cameraState": 1}]
+        else:
+            payload["cameraStates"] = [{"cameraState": 1}, {"cameraState": 0}, {"cameraState": 0}]
+
+        async with ClientSession(MAMMOTION_API_DOMAIN) as session:
+            async with session.post(
+                "/device-server/v1/stream/token",
+                json=payload,
+                headers={
+                    "Authorization": f"{self._headers.get('Authorization', "")}",
+                    "Content-Type": "application/json",
+                    "User-Agent": "okhttp/3.14.9",
+                },
+            ) as resp:
+                data = await resp.json()
+                # TODO catch errors from mismatch like token expire etc
+                # Assuming the data format matches the expected structure
+                return Response[StreamSubscriptionResponse].from_dict(data)
+
+    async def get_video_resource(self, iot_id: str) -> Response[VideoResourceResponse]:
+        """Get video resource for new models (2025 series)"""
+        async with ClientSession(MAMMOTION_API_DOMAIN) as session:
+            async with session.get(
+                f"/device-server/v1/video-resource/{iot_id}",
+                headers={
+                    "Authorization": f"{self._headers.get('Authorization', '')}",
+                    "Content-Type": "application/json",
+                    "User-Agent": "okhttp/3.14.9",
+                },
+            ) as resp:
+                data = await resp.json()
+                # TODO catch errors from mismatch like token expire etc
+                # Assuming the data format matches the expected structure
+                return Response[VideoResourceResponse].from_dict(data)
+
     async def refresh_login(self, account: str, password: str | None = None) -> Response[LoginResponseData]:
         if self._password is None and password is not None:
             self._password = password

pymammotion/{aliyun/model/stream_subscription_response.py → http/model/camera_stream.py}

@@ -16,3 +16,16 @@ class StreamSubscriptionResponse(DataClassORJSONMixin):
     channelName: str
     token: str
     uid: int
+    license: str | None = None
+    availableTime: int | None = None
+
+
+@dataclass
+class VideoResourceResponse(DataClassORJSONMixin):
+    id: str
+    deviceId: str
+    deviceName: str
+    cycleType: int
+    usageYearMonth: str
+    totalTime: int
+    availableTime: int

pymammotion/mammotion/devices/mammotion.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import asyncio
-from enum import Enum
 import logging
 from typing import Any
 
@@ -15,41 +14,36 @@ from pymammotion.data.model.device import MowingDevice
 from pymammotion.data.model.enums import ConnectionPreference
 from pymammotion.data.state_manager import StateManager
 from pymammotion.http.http import MammotionHTTP
+from pymammotion.http.model.camera_stream import StreamSubscriptionResponse, VideoResourceResponse
+from pymammotion.http.model.http import Response
 from pymammotion.mammotion.devices.mammotion_bluetooth import MammotionBaseBLEDevice
 from pymammotion.mammotion.devices.mammotion_cloud import MammotionBaseCloudDevice, MammotionCloud
 from pymammotion.mqtt import MammotionMQTT
+from pymammotion.utility.device_type import DeviceType
 
 TIMEOUT_CLOUD_RESPONSE = 10
 
 _LOGGER = logging.getLogger(__name__)
 
 
-class ConnectionPreference(Enum):
-    """Enum for connection preference."""
-
-    EITHER = 0
-    WIFI = 1
-    BLUETOOTH = 2
-
-
 class MammotionMixedDeviceManager:
-    preference: ConnectionPreference
-    _ble_device: MammotionBaseBLEDevice | None = None
-    _cloud_device: MammotionBaseCloudDevice | None = None
-
     def __init__(
         self,
         name: str,
+        mammotion_http: MammotionHTTP,
         cloud_device: Device | None = None,
         ble_device: BLEDevice | None = None,
         mqtt: MammotionCloud | None = None,
         preference: ConnectionPreference = ConnectionPreference.BLUETOOTH,
     ) -> None:
+        self._ble_device: MammotionBaseBLEDevice | None = None
+        self._cloud_device: MammotionBaseCloudDevice | None = None
         self.name = name
         self._state_manager = StateManager(MowingDevice())
         self._state_manager.get_device().name = name
         self.add_ble(cloud_device, ble_device)
         self.add_cloud(cloud_device, mqtt)
+        self.mammotion_http = mammotion_http
         self.preference = preference
         self._state_manager.preference = preference
 
@@ -169,19 +163,6 @@ class Mammotion:
         """Initialize MammotionDevice."""
         self._login_lock = asyncio.Lock()
 
-    def add_ble_device(
-        self,
-        cloud_device: Device,
-        ble_device: BLEDevice,
-        preference: ConnectionPreference = ConnectionPreference.BLUETOOTH,
-    ) -> None:
-        if ble_device:
-            self.device_manager.add_device(
-                MammotionMixedDeviceManager(
-                    name=ble_device.name, cloud_device=cloud_device, ble_device=ble_device, preference=preference
-                )
-            )
-
     async def login_and_initiate_cloud(self, account, password, force: bool = False) -> None:
         async with self._login_lock:
             exists: MammotionCloud | None = self.mqtt_list.get(account)
@@ -231,6 +212,7 @@ class Mammotion:
             if device.deviceName.startswith(("Luba-", "Yuka-")) and mower_device is None:
                 mixed_device = MammotionMixedDeviceManager(
                     name=device.deviceName,
+                    mammotion_http=mqtt_client.cloud_client.mammotion_http,
                     cloud_device=device,
                     mqtt=mqtt_client,
                     preference=ConnectionPreference.WIFI,
@@ -315,15 +297,27 @@ class Mammotion:
                 return await device.cloud().start_map_sync()
             # TODO work with both with EITHER
 
-    async def get_stream_subscription(self, name: str):
+    async def get_stream_subscription(self, name: str, iot_id: str) -> Response[StreamSubscriptionResponse] | Any:
         device = self.get_device_by_name(name)
-        if device.preference is ConnectionPreference.WIFI:
-            if device.has_cloud():
-                _stream_response = await device.cloud().mqtt.cloud_client.mammotion_http.get_stream_subscription(
-                    device.cloud().iot_id
-                )
-                _LOGGER.debug(_stream_response)
-                return _stream_response
+        if DeviceType.is_mini_or_x_series(name):
+            _stream_response = await device.mammotion_http.get_stream_subscription_mini_or_x_series(
+                iot_id, DeviceType.is_yuka(name)
+            )
+            _LOGGER.debug(_stream_response)
+            return _stream_response
+        else:
+            _stream_response = await device.mammotion_http.get_stream_subscription(iot_id)
+            _LOGGER.debug(_stream_response)
+            return _stream_response
+
+    async def get_video_resource(self, name: str, iot_id: str) -> Response[VideoResourceResponse] | None:
+        device = self.get_device_by_name(name)
+
+        if DeviceType.is_mini_or_x_series(name):
+            _video_resource_response = await device.mammotion_http.get_video_resource(iot_id)
+            _LOGGER.debug(_video_resource_response)
+            return _video_resource_response
+        return None
 
     def mower(self, name: str) -> MowingDevice | None:
         device = self.get_device_by_name(name)

{pymammotion-0.4.35.dist-info → pymammotion-0.4.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pymammotion
-Version: 0.4.35
+Version: 0.4.37
 Summary: 
 License: GPL-3.0
 Author: Michael Arthur

{pymammotion-0.4.35.dist-info → pymammotion-0.4.37.dist-info}/RECORD

@@ -8,7 +8,6 @@ pymammotion/aliyun/model/dev_by_account_response.py,sha256=P9yYy4Z2tLkJSqXA_5XGa
 pymammotion/aliyun/model/login_by_oauth_response.py,sha256=g7JnvEjoa3SplHd-UqCuK6x0qtODpHlDyJCHRz7tfDI,1228
 pymammotion/aliyun/model/regions_response.py,sha256=HSnpPcgpjr6VNXBQHw__gn-xWCkQ-MZ-Tmus9_va9mI,635
 pymammotion/aliyun/model/session_by_authcode_response.py,sha256=0owdNcGFIP7rsVqLIf9rT-iOtvWmKCt2AW0cUUXwFiQ,427
-pymammotion/aliyun/model/stream_subscription_response.py,sha256=7d0i8cNMAMX7aQjd2yebMDdqQkugcCzk-SD6oVtf18M,333
 pymammotion/aliyun/regions.py,sha256=ctlRGrmdE4-xgItl9slCANYOV502qVN5lkAU4lj92sk,2518
 pymammotion/aliyun/tea/core.py,sha256=4SjhRkbPMbw-uI0lQnCN0SBNAHAgVFrpHeaauuu6nZY,10200
 pymammotion/aliyun/tmp_constant.py,sha256=M4Hq_lrGB3LZdX6R2XohRPFoK1NDnNV-pTJwJcJ9838,6650
@@ -49,9 +48,12 @@ pymammotion/data/mqtt/status.py,sha256=SgdrpE1Uldb01hybO6hYhgU1Sp1eILghC0UhMZMHr
 pymammotion/data/state_manager.py,sha256=ClWwxwO88BLRkox0ljoZNzwdTWiPIBMyLBj-15V0GfI,9828
 pymammotion/event/__init__.py,sha256=mgATR6vPHACNQ-0zH5fi7NdzeTCDV1CZyaWPmtUusi8,115
 pymammotion/event/event.py,sha256=bj2RirSIRyBs0QvkcrOtwZWUX_8F3m1sySuHVyKmZLs,2143
+pymammotion/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/http/_init_.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/http/encryption.py,sha256=lzXu3WwBdQlzjXxWnlJuRgkCrKdPbxx5drhMitVKIEk,8287
-pymammotion/http/http.py,sha256=r8jvdh01OAQEBNYW8QHZ2v5MjzCFkv6FibE3kWRmdSg,7359
+pymammotion/http/http.py,sha256=rVTPyc5PQAsW6dILSfpp4hnnl6k3Tuy0bqC4lvnT0j8,9558
+pymammotion/http/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pymammotion/http/model/camera_stream.py,sha256=ilxQNny_w9Frwt-m8kbHinvyjDv4Bx8C2swfZ2lTEDE,600
 pymammotion/http/model/http.py,sha256=tM5ikwVkWhRdXc2xi1NOLmWPH2mQEQelpaVgMlAEmlI,2333
 pymammotion/mammotion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/mammotion/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -69,7 +71,7 @@ pymammotion/mammotion/control/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 pymammotion/mammotion/control/joystick.py,sha256=QfBVxM_gxpWsZAGO90whtgxCI2tIZ3TTad9wHIPsU9s,5640
 pymammotion/mammotion/devices/__init__.py,sha256=f2qQFPgLGmV85W2hSlMUh5BYuht9o_Ar_JEAAMD4fsE,102
 pymammotion/mammotion/devices/base.py,sha256=bL6aPx5o8sv3BrdgSIkpqaBYtW6yQPJ4rKd9JEr-u6c,11815
-pymammotion/mammotion/devices/mammotion.py,sha256=W8LtaZyYa6WhtSPpC7KHBPL3B9OHLlvZTg-Dr1oEFpo,13989
+pymammotion/mammotion/devices/mammotion.py,sha256=ugzmER_EvSzdZf_1OTYnNy3ynZo4NGX0eqN7SAR1hvM,14332
 pymammotion/mammotion/devices/mammotion_bluetooth.py,sha256=r-IoqiBsgEUOxcvU0Ryz8RkW1oUJ6LcXLQa710cYmZI,18862
 pymammotion/mammotion/devices/mammotion_cloud.py,sha256=sVw-McsTJ_8wlkToL5rkjAlN2d3ituvqLona5w5bd1o,14160
 pymammotion/mqtt/__init__.py,sha256=Ocs5e-HLJvTuDpVXyECEsWIvwsUaxzj7lZ9mSYutNDY,105
@@ -121,7 +123,7 @@ pymammotion/utility/movement.py,sha256=N75oAoAgFydqoaOedYIxGUHmuTCtPzAOtb-d_29tp
 pymammotion/utility/mur_mur_hash.py,sha256=xEfOZVbqRawJj66eLgtnZ85OauDR47oIPr29OHelzPI,4468
 pymammotion/utility/periodic.py,sha256=MbeSb9cfhxzYmdT_RiE0dZe3H9IfbQW_zSqhmSX2RUc,3321
 pymammotion/utility/rocker_util.py,sha256=6tX7sS87qoQC_tsxbx3NLL-HgS08wtzXiZkhDiz7uo0,7179
-pymammotion-0.4.35.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pymammotion-0.4.35.dist-info/METADATA,sha256=UrRNXv4V4GLVp1LAxOXfjRCaCX_-BDWc_3ZPtXTbaQ4,3878
-pymammotion-0.4.35.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-pymammotion-0.4.35.dist-info/RECORD,,
+pymammotion-0.4.37.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pymammotion-0.4.37.dist-info/METADATA,sha256=8MqFqYhVRZVIlQg_NR7-yriZo9_1DoN1_zL2CHAsN0o,3878
+pymammotion-0.4.37.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+pymammotion-0.4.37.dist-info/RECORD,,