pymammotion 0.2.62__py3-none-any.whl → 0.5.51__py3-none-any.whl

pymammotion/__init__.py

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

pymammotion/aliyun/client.py

@@ -0,0 +1,235 @@
+from datetime import UTC, datetime
+import time
+
+from Tea.exceptions import UnretryableException
+from Tea.request import TeaRequest
+
+from pymammotion.aliyun.tea.core import TeaCore
+
+try:
+    from typing import Dict
+except ImportError:
+    pass
+
+from alibabacloud_apigateway_util.client import Client as APIGatewayUtilClient
+from alibabacloud_tea_util.client import Client as UtilClient
+
+
+class Client:
+    """test"""
+
+    _app_key = None  # type: str
+    _app_secret = None  # type: str
+    _protocol = None  # type: str
+    _user_agent = None  # type: str
+    _read_timeout = None  # type: int
+    _connect_timeout = None  # type: int
+    _http_proxy = None  # type: str
+    _https_proxy = None  # type: str
+    _no_proxy = None  # type: str
+    _max_idle_conns = None  # type: int
+    _domain = None  # type: str
+
+    def __init__(self, config) -> None:
+        self._domain = config.domain
+        self._app_key = config.app_key
+        self._app_secret = config.app_secret
+        self._protocol = config.protocol
+        self._read_timeout = config.read_timeout
+        self._connect_timeout = config.connect_timeout
+        self._http_proxy = config.http_proxy
+        self._https_proxy = config.https_proxy
+        self._no_proxy = config.no_proxy
+        self._max_idle_conns = config.max_idle_conns
+
+    def do_request(self, pathname, protocol, method, header, body, runtime):
+        """Send request
+
+        @type pathname: str
+        @param pathname: the url path
+
+        @type protocol: str
+        @param protocol: http or https
+
+        @type method: str
+        @param method: example GET
+
+        @type header: Dict[str, str]
+        @param header: request header
+
+        @type body: iot_api_gateway_models.IoTApiRequest
+        @param body: the object of IoTApiRequest
+
+        @type runtime: util_models.RuntimeOptions
+        @param runtime: which controls some details of call api, such as retry times
+
+        @rtype: TeaResponse
+        @return: the response
+        """
+        body.validate()
+        runtime.validate()
+        _runtime = {
+            "timeouted": "retry",
+            "readTimeout": UtilClient.default_number(runtime.read_timeout, self._read_timeout),
+            "connectTimeout": UtilClient.default_number(runtime.connect_timeout, self._connect_timeout),
+            "httpProxy": UtilClient.default_string(runtime.http_proxy, self._http_proxy),
+            "httpsProxy": UtilClient.default_string(runtime.https_proxy, self._https_proxy),
+            "noProxy": UtilClient.default_string(runtime.no_proxy, self._no_proxy),
+            "maxIdleConns": UtilClient.default_number(runtime.max_idle_conns, self._max_idle_conns),
+            "retry": {
+                "retryable": runtime.autoretry,
+                "maxAttempts": UtilClient.default_number(runtime.max_attempts, 3),
+            },
+            "backoff": {
+                "policy": UtilClient.default_string(runtime.backoff_policy, "yes"),
+                "period": UtilClient.default_number(runtime.backoff_period, 1),
+            },
+            "ignoreSSL": runtime.ignore_ssl,
+        }
+        _last_request = None
+        _last_exception = None
+        _now = time.time()
+        _retry_times = 0
+        while TeaCore.allow_retry(_runtime.get("retry"), _retry_times, _now):
+            if _retry_times > 0:
+                _backoff_time = TeaCore.get_backoff_time(_runtime.get("backoff"), _retry_times)
+                if _backoff_time > 0:
+                    TeaCore.sleep(_backoff_time)
+            _retry_times = _retry_times + 1
+            try:
+                _request = TeaRequest()
+                _request.protocol = UtilClient.default_string(self._protocol, protocol)
+                _request.method = UtilClient.default_string(method, "POST")
+                _request.pathname = pathname
+                _request.headers = TeaCore.merge(
+                    {
+                        "host": self._domain,
+                        "date": datetime.now(UTC).strftime("%a, %d %b %Y %H:%M:%S GMT"),
+                        "x-ca-nonce": UtilClient.get_nonce(),
+                        "x-ca-key": self._app_key,
+                        "x-ca-signaturemethod": "HmacSHA256",
+                        "accept": "application/json",
+                        "user-agent": self.get_user_agent(),
+                    },
+                    header,
+                )
+                if UtilClient.empty(body.id):
+                    body.id = UtilClient.get_nonce()
+                if not UtilClient.is_unset(body):
+                    _request.headers["content-type"] = "application/octet-stream"
+                    _request.headers["content-md5"] = APIGatewayUtilClient.get_content_md5(
+                        UtilClient.to_jsonstring(TeaCore.to_map(body))
+                    )
+                    _request.body = UtilClient.to_jsonstring(TeaCore.to_map(body))
+                _request.headers["x-ca-signature"] = APIGatewayUtilClient.get_signature(_request, self._app_secret)
+                _last_request = _request
+                _response = TeaCore.do_action(_request, _runtime)
+                if _response.body.get("code") == 20056:
+                    raise Exception("Gateway timeout.")
+
+                return _response
+            except Exception as e:
+                if TeaCore.is_retryable(e):
+                    _last_exception = e
+                    continue
+                raise e
+        raise UnretryableException(_last_request, _last_exception)
+
+    async def async_do_request(self, pathname, protocol, method, header, body, runtime):
+        """Send request
+
+        @type pathname: str
+        @param pathname: the url path
+
+        @type protocol: str
+        @param protocol: http or https
+
+        @type method: str
+        @param method: example GET
+
+        @type header: Dict[str, str]
+        @param header: request header
+
+        @type body: iot_api_gateway_models.IoTApiRequest
+        @param body: the object of IoTApiRequest
+
+        @type runtime: util_models.RuntimeOptions
+        @param runtime: which controls some details of call api, such as retry times
+
+        @rtype: TeaResponse
+        @return: the response
+        """
+        body.validate()
+        runtime.validate()
+        _runtime = {
+            "timeouted": "retry",
+            "readTimeout": UtilClient.default_number(runtime.read_timeout, self._read_timeout),
+            "connectTimeout": UtilClient.default_number(runtime.connect_timeout, self._connect_timeout),
+            "httpProxy": UtilClient.default_string(runtime.http_proxy, self._http_proxy),
+            "httpsProxy": UtilClient.default_string(runtime.https_proxy, self._https_proxy),
+            "noProxy": UtilClient.default_string(runtime.no_proxy, self._no_proxy),
+            "maxIdleConns": UtilClient.default_number(runtime.max_idle_conns, self._max_idle_conns),
+            "retry": {
+                "retryable": runtime.autoretry,
+                "maxAttempts": UtilClient.default_number(runtime.max_attempts, 3),
+            },
+            "backoff": {
+                "policy": UtilClient.default_string(runtime.backoff_policy, "yes"),
+                "period": UtilClient.default_number(runtime.backoff_period, 1),
+            },
+            "ignoreSSL": runtime.ignore_ssl,
+        }
+        _last_request = None
+        _last_exception = None
+        _now = time.time()
+        _retry_times = 0
+        while TeaCore.allow_retry(_runtime.get("retry"), _retry_times, _now):
+            if _retry_times > 0:
+                _backoff_time = TeaCore.get_backoff_time(_runtime.get("backoff"), _retry_times)
+                if _backoff_time > 0:
+                    await TeaCore.sleep_async(_backoff_time)
+            _retry_times = _retry_times + 1
+            try:
+                _request = TeaRequest()
+                _request.protocol = UtilClient.default_string(self._protocol, protocol)
+                _request.method = UtilClient.default_string(method, "POST")
+                _request.pathname = pathname
+                _request.headers = TeaCore.merge(
+                    {
+                        "host": self._domain,
+                        "date": UtilClient.get_date_utcstring(),
+                        "x-ca-nonce": UtilClient.get_nonce(),
+                        "x-ca-key": self._app_key,
+                        "x-ca-signaturemethod": "HmacSHA256",
+                        "accept": "application/json",
+                        "user-agent": self.get_user_agent(),
+                    },
+                    header,
+                )
+                if UtilClient.empty(body.id):
+                    body.id = UtilClient.get_nonce()
+                if not UtilClient.is_unset(body):
+                    _request.headers["content-type"] = "application/octet-stream"
+                    _request.headers["content-md5"] = APIGatewayUtilClient.get_content_md5(
+                        UtilClient.to_jsonstring(TeaCore.to_map(body))
+                    )
+                    _request.body = UtilClient.to_jsonstring(TeaCore.to_map(body))
+                _request.headers["x-ca-signature"] = APIGatewayUtilClient.get_signature(_request, self._app_secret)
+                _last_request = _request
+                _response = await TeaCore.async_do_action(_request, _runtime)
+                return _response
+            except Exception as e:
+                if TeaCore.is_retryable(e):
+                    _last_exception = e
+                    continue
+                raise e
+        raise UnretryableException(_last_request, _last_exception)
+
+    def get_user_agent(self):
+        """Get user agent
+
+        @rtype: str
+        @return: user agent
+        """
+        user_agent = UtilClient.get_user_agent(self._user_agent)
+        return user_agent