python sdk: IoT Hub 为设备HTTP接入提供测试代码

ccharlesren · ccharlesren · commit f93d367e6089 · 2022-01-13T17:51:14.000+08:00
Change-Id: Ib3c8c83a44994bb504c305aff64b6597e04dfd5d
diff --git a/hub/hub.py b/hub/hub.py
@@ -946,6 +946,41 @@ def publishDevice(self,topicName,signType,payload,qos,timeout=10):
                 self._logger.error('code: {}, error message: {}'.format(
                     err_code, err_code['Message']))
                 return -1, err_code['Message']
+            
+    def httpCallback(self, on_connect, on_disconnect,
+                            on_message, on_publish,
+                            on_subscribe, on_unsubscribe):
+        """Register user http callback
+
+        Register user http callback for http
+        Args:
+            on_connect: http connect callback
+            on_disconnect: http disconnect callback
+            on_message: http message callback
+            on_publish: http publish callback
+            on_subscribe: http subscribe callback
+            on_unsubscribe: http unsubscribe callback
+        Returns:
+            success: default
+            fail: default
+        """
+        self.__user_on_connect = on_connect
+        self.__user_on_disconnect = on_disconnect
+        self.__user_on_message = on_message
+        self.__user_on_publish = on_publish
+        self.__user_on_subscribe = on_subscribe
+        self.__user_on_unsubscribe = on_unsubscribe    
+    
+    def isHttpConnected(self):
+        """Is http connected
+
+        Is http connected
+        Args: None
+        Returns:
+            success: True/False
+        """
+        return self.__hub_state == self.HubState.CONNECTED    
+        
         
     def isSubdevStatusOnline(self, sub_productId, sub_devName):
         """Sub-device status
diff --git a/hub/sample/deviceReport/example_http_report.py b/hub/sample/deviceReport/example_http_report.py
diff --git a/hub/sample/httpAccess/example_http.py b/hub/sample/httpAccess/example_http.py
@@ -0,0 +1,65 @@
+import sys
+import logging
+import time
+from hub.hub import QcloudHub
+
+provider = QcloudHub(device_file="hub/sample/device_info.json", tls=True)
+qcloud = provider.hub
+logger = qcloud.logInit(qcloud.LoggerLevel.DEBUG, "logs/log", 1024*1024*10, 5, enable=True)
+
+def on_connect(flags, rc, userdata):
+    logger.debug("%s:flags:%d,rc:%d,userdata:%s" % (sys._getframe().f_code.co_name, flags, rc, userdata))
+    pass
+
+
+def on_disconnect(rc, userdata):
+    logger.debug("%s:rc:%d,userdata:%s" % (sys._getframe().f_code.co_name, rc, userdata))
+    pass
+
+
+def on_message(topic, payload, qos, userdata):
+    logger.debug("%s:topic:%s,payload:%s,qos:%s,userdata:%s" % (sys._getframe().f_code.co_name, topic, payload, qos, userdata))
+    pass
+
+
+def on_publish(mid, userdata):
+    logger.debug("%s:mid:%d,userdata:%s" % (sys._getframe().f_code.co_name, mid, userdata))
+    pass
+
+
+def on_subscribe(mid, granted_qos, userdata):
+    logger.debug("%s:mid:%d,granted_qos:%s,userdata:%s" % (sys._getframe().f_code.co_name, mid, granted_qos, userdata))
+    pass
+
+
+def on_unsubscribe(mid, userdata):
+    logger.debug("%s:mid:%d,userdata:%s" % (sys._getframe().f_code.co_name, mid, userdata))
+    pass
+
+def example_http():
+    logger.debug("\033[1;36m http test start...\033[0m")
+
+    qcloud.httpCallback(on_connect, on_disconnect,
+                            on_message, on_publish,
+                            on_subscribe, on_unsubscribe)
+    qcloud.connect()
+
+    count = 0
+    while True:
+        if qcloud.isHttpConnected():
+            break
+        else:
+            if count >= 3:
+                logger.error("\033[1;31m http test fail...\033[0m")
+                return False
+            time.sleep(1)
+            count += 1
+
+    timestamp = qcloud.getNtpAccurateTime()
+    dt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(timestamp/1000))
+    logger.debug("current time:%s" % dt)
+
+    # qcloud.disconnect()
+    logger.debug("\033[1;36m http test success...\033[0m")
+
+    return True
diff --git a/hub/sample/test.py b/hub/sample/test.py
@@ -7,6 +7,7 @@
 from broadcast import example_broadcast as broadcasttest
 from rrpc import example_rrpc as rrpctest
 from shadow import example_shadow as shadowtest
+from httpAccess import example_http as httptest
 
 class MyTestCase(unittest.TestCase):
 
@@ -32,6 +33,11 @@ def test_gateway(self):
         self.assertEqual(ret, True)
         pass
 
+    def test_http(self):
+        ret = httptest.example_http()
+        self.assertEqual(ret, True)
+        pass
+
     @unittest.skip("skipping")
     def test_ota(self):
         ret = otatest.example_ota()
@@ -53,6 +59,5 @@ def test_shadow(self):
         self.assertEqual(ret, True)
         pass
 
-
 if __name__ == '__main__':
     unittest.main()
