pymammotion 0.5.34__py3-none-any.whl → 0.5.53__py3-none-any.whl

pymammotion/__init__.py

@@ -15,12 +15,12 @@ from pymammotion.bluetooth.ble import MammotionBLE
 from pymammotion.http.http import MammotionHTTP
 
 # TODO make a working device that will work outside HA too.
-from pymammotion.mqtt import MammotionMQTT
+from pymammotion.mqtt import AliyunMQTT, MammotionMQTT
 
 logger = logging.getLogger(__name__)
 
 
-__all__ = ["MammotionBLE", "MammotionHTTP", "MammotionMQTT", "logger"]
+__all__ = ["MammotionBLE", "MammotionHTTP", "AliyunMQTT", "MammotionMQTT", "logger"]
 
 
 # TODO provide interface to pick between mqtt/cloud/bluetooth
@@ -37,7 +37,7 @@ if __name__ == "__main__":
     REGION = os.environ.get("REGION")
     mammotion_http = MammotionHTTP()
     cloud_client = CloudIOTGateway(mammotion_http)
-    luba = MammotionMQTT(
+    luba = AliyunMQTT(
         iot_token=IOT_TOKEN or "",
         region_id=REGION or "",
         product_key=PRODUCT_KEY or "",

pymammotion/aliyun/cloud_gateway.py

@@ -22,7 +22,7 @@ from Tea.exceptions import UnretryableException
 from pymammotion.aliyun.client import Client
 from pymammotion.aliyun.model.aep_response import AepResponse
 from pymammotion.aliyun.model.connect_response import ConnectResponse
-from pymammotion.aliyun.model.dev_by_account_response import ListingDevByAccountResponse
+from pymammotion.aliyun.model.dev_by_account_response import ListingDevAccountResponse
 from pymammotion.aliyun.model.login_by_oauth_response import LoginByOAuthResponse
 from pymammotion.aliyun.model.regions_response import RegionResponse
 from pymammotion.aliyun.model.session_by_authcode_response import SessionByAuthCodeResponse
@@ -103,6 +103,9 @@ class CheckSessionException(Exception):
     """Raise exception when checking session results in a failure."""
 
 
+EXPIRED_CREDENTIAL_EXCEPTIONS = (CheckSessionException, SetupException)
+
+
 class CloudIOTGateway:
     """Class for interacting with Aliyun Cloud IoT Gateway."""
 
@@ -120,7 +123,7 @@ class CloudIOTGateway:
         aep_response: AepResponse | None = None,
         session_by_authcode_response: SessionByAuthCodeResponse | None = None,
         region_response: RegionResponse | None = None,
-        dev_by_account: ListingDevByAccountResponse | None = None,
+        dev_by_account: ListingDevAccountResponse | None = None,
     ) -> None:
         """Initialize the CloudIOTGateway."""
         self.mammotion_http: MammotionHTTP = mammotion_http
@@ -146,7 +149,7 @@ class CloudIOTGateway:
             )
 
     @staticmethod
-    def generate_random_string(length: int):
+    def generate_random_string(length: int) -> str:
         """Generate a random string of specified length."""
         characters = string.ascii_letters + string.digits
         return "".join(random.choice(characters) for _ in range(length))
@@ -165,7 +168,7 @@ class CloudIOTGateway:
             logger.error("Couldn't decode message %s", response_body_str)
             return {"code": 22000}
 
-    def sign(self, data):
+    def sign(self, data: dict) -> str:
         """Generate signature for the given data."""
         keys = ["appKey", "clientId", "deviceSn", "timestamp"]
         concatenated_str = ""
@@ -180,7 +183,7 @@ class CloudIOTGateway:
             hashlib.sha1,
         ).hexdigest()
 
-    async def get_region(self, country_code: str):
+    async def get_region(self, country_code: str) -> RegionResponse:
         """Get the region based on country code and auth code."""
         auth_code = self.mammotion_http.login_info.authorization_code
 
@@ -244,7 +247,7 @@ class CloudIOTGateway:
 
         return response.body
 
-    async def aep_handle(self):
+    async def aep_handle(self) -> AepResponse:
         """Handle AEP authentication."""
         aep_domain = self.domain
 
@@ -300,9 +303,9 @@ class CloudIOTGateway:
 
         logger.debug(response_body_dict)
 
-        return response.body
+        return self._aep_response
 
-    async def connect(self):
+    async def connect(self) -> ConnectResponse:
         """Connect to the Aliyun Cloud IoT Gateway."""
         region_url = "sdk.openaccount.aliyun.com"
         time_now = time.time()
@@ -449,7 +452,7 @@ class CloudIOTGateway:
                     return self._login_by_oauth_response
                 raise LoginException(data)
 
-    async def session_by_auth_code(self):
+    async def session_by_auth_code(self) -> SessionByAuthCodeResponse:
         """Create a session by auth code."""
         config = Config(
             app_key=self._app_key,
@@ -608,7 +611,8 @@ class CloudIOTGateway:
         session_data = session.data
 
         if (
-            session_data.identityId is None
+            session_data is None
+            or session_data.identityId is None
             or session_data.refreshTokenExpire is None
             or session_data.iotToken is None
             or session_data.iotTokenExpire is None
@@ -619,7 +623,7 @@ class CloudIOTGateway:
         self._session_by_authcode_response = session
         self._iot_token_issued_at = int(time.time())
 
-    async def list_binding_by_account(self) -> ListingDevByAccountResponse:
+    async def list_binding_by_account(self) -> ListingDevAccountResponse:
         """List bindings by account."""
         config = Config(
             app_key=self._app_key,
@@ -658,9 +662,9 @@ class CloudIOTGateway:
         response_body_dict = self.parse_json_response(response_body_str)
 
         if int(response_body_dict.get("code")) != 200:
-            raise Exception("Error in creating session: " + response_body_dict["msg"])
+            raise Exception("Error in creating session: " + response_body_dict["message"])
 
-        self._devices_by_account_response = ListingDevByAccountResponse.from_dict(response_body_dict)
+        self._devices_by_account_response = ListingDevAccountResponse.from_dict(response_body_dict)
         return self._devices_by_account_response
 
     async def list_binding_by_dev(self, iot_id: str):
@@ -698,10 +702,94 @@ class CloudIOTGateway:
         # Load the JSON string into a dictionary
         response_body_dict = self.parse_json_response(response_body_str)
 
+        if int(response_body_dict.get("code")) != 200:
+            raise Exception("Error in getting shared device list: " + response_body_dict["msg"])
+
+        self._devices_by_account_response = ListingDevAccountResponse.from_dict(response_body_dict)
+        return self._devices_by_account_response
+
+    async def confirm_share(self, record_list: list[str]) -> bool:
+        config = Config(
+            app_key=self._app_key,
+            app_secret=self._app_secret,
+            domain=self._region_response.data.apiGatewayEndpoint,
+        )
+
+        client = Client(config)
+
+        # build request
+        request = CommonParams(
+            api_ver="1.0.7",
+            language="en-US",
+            iot_token=self._session_by_authcode_response.data.iotToken,
+        )
+        body = IoTApiRequest(
+            id=str(uuid.uuid4()),
+            params={"agree": 1, "recordIdList": record_list},
+            request=request,
+            version="1.0",
+        )
+
+        # send request
+        response = await client.async_do_request("/uc/confirmShare", "https", "POST", None, body, RuntimeOptions())
+        logger.debug(response.status_message)
+        logger.debug(response.headers)
+        logger.debug(response.status_code)
+        logger.debug(response.body)
+
+        # Decode the response body
+        response_body_str = response.body.decode("utf-8")
+
+        # Load the JSON string into a dictionary
+        response_body_dict = self.parse_json_response(response_body_str)
+
+        if int(response_body_dict.get("code")) != 200:
+            raise Exception("Error in accepting share: " + response_body_dict["msg"])
+
+        return True
+
+    async def get_shared_notice_list(self):
+        ### status 0 accepted status -1 ready to be accepted 3 expired
+        config = Config(
+            app_key=self._app_key,
+            app_secret=self._app_secret,
+            domain=self._region_response.data.apiGatewayEndpoint,
+        )
+
+        client = Client(config)
+
+        # build request
+        request = CommonParams(
+            api_ver="1.0.9",
+            language="en-US",
+            iot_token=self._session_by_authcode_response.data.iotToken,
+        )
+        body = IoTApiRequest(
+            id=str(uuid.uuid4()),
+            params={"pageSize": 100, "pageNo": 1},
+            request=request,
+            version="1.0",
+        )
+
+        # send request
+        response = await client.async_do_request(
+            "/uc/getShareNoticeList", "https", "POST", None, body, RuntimeOptions()
+        )
+        logger.debug(response.status_message)
+        logger.debug(response.headers)
+        logger.debug(response.status_code)
+        logger.debug(response.body)
+
+        # Decode the response body
+        response_body_str = response.body.decode("utf-8")
+
+        # Load the JSON string into a dictionary
+        response_body_dict = self.parse_json_response(response_body_str)
+
         if int(response_body_dict.get("code")) != 200:
             raise Exception("Error in creating session: " + response_body_dict["msg"])
 
-        self._devices_by_account_response = ListingDevByAccountResponse.from_dict(response_body_dict)
+        self._devices_by_account_response = ListingDevAccountResponse.from_dict(response_body_dict)
         return self._devices_by_account_response
 
     async def send_cloud_command(self, iot_id: str, command: bytes) -> str:
@@ -867,11 +955,11 @@ class CloudIOTGateway:
         self.mammotion_http = mammotion_http
 
     @property
-    def region_response(self) -> RegionResponse:
+    def region_response(self) -> RegionResponse | None:
         return self._region_response
 
     @property
-    def aep_response(self) -> AepResponse:
+    def aep_response(self) -> AepResponse | None:
         return self._aep_response
 
     @property
@@ -883,9 +971,9 @@ class CloudIOTGateway:
         return self._client_id
 
     @property
-    def login_by_oauth_response(self) -> LoginByOAuthResponse:
+    def login_by_oauth_response(self) -> LoginByOAuthResponse | None:
         return self._login_by_oauth_response
 
     @property
-    def connect_response(self) -> ConnectResponse:
+    def connect_response(self) -> ConnectResponse | None:
         return self._connect_response

pymammotion/aliyun/model/dev_by_account_response.py

@@ -1,35 +1,183 @@
 from dataclasses import dataclass
+from typing import Annotated, Optional
 
 from mashumaro.config import BaseConfig
 from mashumaro.mixins.orjson import DataClassORJSONMixin
+from mashumaro.types import Alias
 
 
 @dataclass
 class Device(DataClassORJSONMixin):
-    gmtModified: int
-    netType: str
-    categoryKey: str
-    productKey: str
-    nodeType: str
-    isEdgeGateway: bool
-    deviceName: str
-    categoryName: str
-    identityAlias: str
-    productName: str
-    iotId: str
-    bindTime: int
-    owned: int
-    identityId: str
-    thingType: str
+    """Unified device model supporting both Device and ShareNotification data"""
+
+    # Core device fields (from Device model)
+    gmt_modified: Annotated[int, Alias("gmtModified")]
+    node_type: Annotated[str, Alias("nodeType")]
+    device_name: Annotated[str, Alias("deviceName")]
+    product_name: Annotated[str, Alias("productName")]
     status: int
-    nickName: str | None = None
-    description: str | None = None
-    productImage: str | None = None
-    categoryImage: str | None = None
-    productModel: str | None = None
+    identity_id: Annotated[str, Alias("identityId")]
+
+    # Required fields from original Device model
+    net_type: Annotated[str, Alias("netType")]
+    category_key: Annotated[str, Alias("categoryKey")]
+    product_key: Annotated[str, Alias("productKey")]
+    is_edge_gateway: Annotated[bool, Alias("isEdgeGateway")]
+    category_name: Annotated[str, Alias("categoryName")]
+    identity_alias: Annotated[str, Alias("identityAlias")]
+    iot_id: Annotated[str, Alias("iotId")]
+    bind_time: Annotated[int, Alias("bindTime")]
+    owned: int
+    thing_type: Annotated[str, Alias("thingType")]
+
+    # Optional fields (common to both or nullable)
+    nick_name: Annotated[Optional[str], Alias("nickName")] = None
+    description: Optional[str] = None
+    product_image: Annotated[Optional[str], Alias("productImage")] = None
+    category_image: Annotated[Optional[str], Alias("categoryImage")] = None
+    product_model: Annotated[Optional[str], Alias("productModel")] = None
+
+    # Optional fields from ShareNotification only
+    target_id: Annotated[Optional[str], Alias("targetId")] = None
+    receiver_identity_id: Annotated[Optional[str], Alias("receiverIdentityId")] = None
+    target_type: Annotated[Optional[str], Alias("targetType")] = None
+    gmt_create: Annotated[Optional[int], Alias("gmtCreate")] = None
+    batch_id: Annotated[Optional[str], Alias("batchId")] = None
+    record_id: Annotated[Optional[str], Alias("recordId")] = None
+    initiator_identity_id: Annotated[Optional[str], Alias("initiatorIdentityId")] = None
+    is_receiver: Annotated[Optional[int], Alias("isReceiver")] = None
+    initiator_alias: Annotated[Optional[str], Alias("initiatorAlias")] = None
+    receiver_alias: Annotated[Optional[str], Alias("receiverAlias")] = None
 
     class Config(BaseConfig):
         omit_default = True
+        allow_deserialization_not_by_alias = True
+
+
+# # Alternative: Keep them separate but with a common base class
+# @dataclass
+# class BaseDevice(DataClassORJSONMixin):
+#     """Base device model with common fields"""
+#
+#     gmt_modified: int
+#     node_type: str
+#     device_name: str
+#     product_name: str
+#     status: int
+#     product_image: Optional[str] = None
+#     category_image: Optional[str] = None
+#     description: Optional[str] = None
+#
+#     class Config(BaseConfig):
+#         omit_default = True
+#         serialize_by_alias = True
+#         aliases = {
+#             "gmt_modified": "gmtModified",
+#             "node_type": "nodeType",
+#             "device_name": "deviceName",
+#             "product_name": "productName",
+#             "product_image": "productImage",
+#             "category_image": "categoryImage",
+#         }
+#
+#
+# @dataclass
+# class Device(BaseDevice):
+#     """Full device model"""
+#
+#     net_type: str
+#     category_key: str
+#     product_key: str
+#     is_edge_gateway: bool
+#     category_name: str
+#     identity_alias: str
+#     iot_id: str
+#     bind_time: int
+#     owned: int
+#     identity_id: str
+#     thing_type: str
+#     nick_name: Optional[str] = None
+#     product_model: Optional[str] = None
+#
+#     class Config(BaseConfig):
+#         omit_default = True
+#         serialize_by_alias = True
+#         aliases = {
+#             **BaseDevice.Config.aliases,
+#             "net_type": "netType",
+#             "category_key": "categoryKey",
+#             "product_key": "productKey",
+#             "is_edge_gateway": "isEdgeGateway",
+#             "category_name": "categoryName",
+#             "identity_alias": "identityAlias",
+#             "iot_id": "iotId",
+#             "bind_time": "bindTime",
+#             "identity_id": "identityId",
+#             "thing_type": "thingType",
+#             "nick_name": "nickName",
+#             "product_model": "productModel",
+#         }
+#
+#
+# @dataclass
+# class ShareNotification(BaseDevice):
+#     """Share notification model extending base device"""
+#
+#     target_id: str
+#     receiver_identity_id: str
+#     target_type: str
+#     gmt_create: int
+#     batch_id: str
+#     record_id: str
+#     initiator_identity_id: str
+#     is_receiver: int
+#     initiator_alias: str
+#     receiver_alias: str
+#
+#     # Optional fields that Device has but ShareNotification might not
+#     net_type: Optional[str] = None
+#     category_key: Optional[str] = None
+#     product_key: Optional[str] = None
+#     is_edge_gateway: Optional[bool] = None
+#     category_name: Optional[str] = None
+#     identity_alias: Optional[str] = None
+#     iot_id: Optional[str] = None
+#     bind_time: Optional[int] = None
+#     owned: Optional[int] = None
+#     identity_id: Optional[str] = None
+#     thing_type: Optional[str] = None
+#     nick_name: Optional[str] = None
+#     product_model: Optional[str] = None
+#
+#     class Config(BaseConfig):
+#         omit_default = True
+#         serialize_by_alias = True
+#         aliases = {
+#             **BaseDevice.Config.aliases,
+#             "target_id": "targetId",
+#             "receiver_identity_id": "receiverIdentityId",
+#             "target_type": "targetType",
+#             "gmt_create": "gmtCreate",
+#             "batch_id": "batchId",
+#             "record_id": "recordId",
+#             "initiator_identity_id": "initiatorIdentityId",
+#             "is_receiver": "isReceiver",
+#             "initiator_alias": "initiatorAlias",
+#             "receiver_alias": "receiverAlias",
+#             # Device fields that might be present
+#             "net_type": "netType",
+#             "category_key": "categoryKey",
+#             "product_key": "productKey",
+#             "is_edge_gateway": "isEdgeGateway",
+#             "category_name": "categoryName",
+#             "identity_alias": "identityAlias",
+#             "iot_id": "iotId",
+#             "bind_time": "bindTime",
+#             "identity_id": "identityId",
+#             "thing_type": "thingType",
+#             "nick_name": "nickName",
+#             "product_model": "productModel",
+#         }
 
 
 @dataclass
@@ -41,7 +189,7 @@ class Data(DataClassORJSONMixin):
 
 
 @dataclass
-class ListingDevByAccountResponse(DataClassORJSONMixin):
+class ListingDevAccountResponse(DataClassORJSONMixin):
     code: int
     data: Data | None
     id: str | None = None

pymammotion/const.py

@@ -8,3 +8,6 @@ MAMMOTION_DOMAIN = "https://id.mammotion.com"
 MAMMOTION_API_DOMAIN = "https://domestic.mammotion.com"
 MAMMOTION_CLIENT_ID = "MADKALUBAS"
 MAMMOTION_CLIENT_SECRET = "GshzGRZJjuMUgd2sYHM7"
+
+MAMMOTION_OUATH2_CLIENT_ID = "GxebgSt8si6pKqR"
+MAMMOTION_OUATH2_CLIENT_SECRET = "JP0508SRJFa0A90ADpzLINDBxMa4Vj"

pymammotion/data/model/device.py

@@ -8,6 +8,7 @@ from mashumaro.mixins.orjson import DataClassORJSONMixin
 from pymammotion.data.model import HashList, RapidState
 from pymammotion.data.model.device_info import DeviceFirmwares, DeviceNonWorkingHours, MowerInfo
 from pymammotion.data.model.errors import DeviceErrors
+from pymammotion.data.model.events import Events
 from pymammotion.data.model.location import Location
 from pymammotion.data.model.report_info import ReportData
 from pymammotion.data.model.work import CurrentTaskSettings
@@ -41,6 +42,7 @@ class MowingDevice(DataClassORJSONMixin):
     device_firmwares: DeviceFirmwares = field(default_factory=DeviceFirmwares)
     errors: DeviceErrors = field(default_factory=DeviceErrors)
     non_work_hours: DeviceNonWorkingHours = field(default_factory=DeviceNonWorkingHours)
+    events: Events = field(default_factory=Events)
 
     def buffer(self, buffer_list: SystemUpdateBufMsg) -> None:
         """Update the device based on which buffer we are reading from."""
@@ -51,9 +53,11 @@ class MowingDevice(DataClassORJSONMixin):
                     self.location.RTK.latitude = parse_double(buffer_list.update_buf_data[5], 8.0)
                     self.location.RTK.longitude = parse_double(buffer_list.update_buf_data[6], 8.0)
                 if buffer_list.update_buf_data[7] != 0:
-                    self.location.dock.latitude = parse_double(buffer_list.update_buf_data[7], 4.0)
-                    self.location.dock.longitude = parse_double(buffer_list.update_buf_data[8], 4.0)
-                    self.location.dock.rotation = buffer_list.update_buf_data[3] + 180
+                    # latitude Y longitude X
+                    self.location.dock.longitude = parse_double(buffer_list.update_buf_data[7], 4.0)
+                    self.location.dock.latitude = parse_double(buffer_list.update_buf_data[8], 4.0)
+                    self.location.dock.rotation = buffer_list.update_buf_data[3]
+
             case 2:
                 self.errors.err_code_list.clear()
                 self.errors.err_code_list_time.clear()
@@ -85,6 +89,20 @@ class MowingDevice(DataClassORJSONMixin):
                         buffer_list.update_buf_data[22],
                     ]
                 )
+            case 3:
+                # task state event
+                task_area_map: dict[int, int] = {}
+                task_area_ids = []
+
+                for i in range(3, len(buffer_list.update_buf_data), 2):
+                    area_id = buffer_list.update_buf_data[i]
+
+                    if area_id != 0:
+                        area_value = int(buffer_list.update_buf_data[i + 1])
+                        task_area_map[area_id] = area_value
+                        task_area_ids.append(area_id)
+                self.events.work_tasks_event.hash_list = task_area_map
+                self.events.work_tasks_event.ids = task_area_ids
 
     def update_report_data(self, toapp_report_data: ReportInfoData) -> None:
         """Set report data for the mower."""
@@ -96,16 +114,11 @@ class MowingDevice(DataClassORJSONMixin):
                 self.location.device = coordinate_converter.enu_to_lla(
                     parse_double(location.real_pos_y, 4.0), parse_double(location.real_pos_x, 4.0)
                 )
+                self.map.invalidate_maps(location.bol_hash)
                 if location.zone_hash:
                     self.location.work_zone = (
                         location.zone_hash if self.report_data.dev.sys_status == WorkMode.MODE_WORKING else 0
                     )
-                # if location.bol_hash:
-                #     for loc in toapp_report_data.locations:
-                #         if loc.bol_hash:
-                #             if loc.bol_hash != location.bol_hash:
-                #                 self.map = HashList()
-                # MurMurHashUtil.hash_unsigned_list(list(self.map.area.keys()))
 
         if toapp_report_data.fw_info:
             self.update_device_firmwares(toapp_report_data.fw_info)

pymammotion/data/model/device_config.py

@@ -30,7 +30,7 @@ class OperationSettings(DataClassORJSONMixin):
     obstacle_laps: int = 1
     mowing_laps: int = 1  # border laps
     start_progress: int = 0
-    areas: list[int] = field(default_factory=list)
+    areas: set[int] = field(default_factory=set)
 
 
 def create_path_order(operation_mode: OperationSettings, device_name: str) -> str:

pymammotion/data/model/device_info.py

@@ -16,6 +16,7 @@ class SideLight(DataClassORJSONMixin):
 
 @dataclass
 class DeviceNonWorkingHours(DataClassORJSONMixin):
+    sub_cmd: int = 0
     start_time: str = ""
     end_time: str = ""
 

pymammotion/data/model/enums.py

@@ -4,9 +4,11 @@ from enum import Enum
 class ConnectionPreference(Enum):
     """Enum for connection preference."""
 
-    EITHER = 0
+    ANY = 0
     WIFI = 1
     BLUETOOTH = 2
+    PREFER_WIFI = 3
+    PREFER_BLUETOOTH = 4
 
 
 class PositionMode(Enum):
@@ -17,7 +19,7 @@ class PositionMode(Enum):
     UNKNOWN = 4
 
     @staticmethod
-    def from_value(value: int):
+    def from_value(value: int) -> "PositionMode":
         if value == 0:
             return PositionMode.FIX
         elif value == 1:
@@ -50,7 +52,7 @@ class RTKStatus(Enum):
     UNKNOWN = 6
 
     @staticmethod
-    def from_value(value: int):
+    def from_value(value: int) -> "RTKStatus":
         if value == 0:
             return RTKStatus.NONE
         elif value == 1 or value == 2:

pymammotion/data/model/events.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass, field
+
+from mashumaro.mixins.orjson import DataClassORJSONMixin
+
+
+@dataclass
+class WorkTaskEvent(DataClassORJSONMixin):
+    hash_area_map: dict[int, int] = field(default_factory=dict)
+    ids: list[int] = field(default_factory=list)
+
+
+@dataclass
+class Events(DataClassORJSONMixin):
+    work_tasks_event: WorkTaskEvent = field(default_factory=WorkTaskEvent)