update extension v2

Author: wwj718

extensions_v2/eim_monitor.py

@@ -0,0 +1,2 @@
+def monitor(content,logger):
+    return content + ' from script'

extensions_v2/eim_trigger.py

@@ -0,0 +1,6 @@
+import time
+
+def trigger():
+    timestamp = time.time()
+    time.sleep(1)
+    return timestamp

extensions_v2/extension_HA.py

@@ -0,0 +1,17 @@
+from codelab_adapter.core_extension import ControllerExtension
+from codelab_adapter.utils import get_server_file_path
+
+
+class HAControllerExtension(ControllerExtension):
+    '''
+    use to control HA server
+    '''
+
+    def __init__(self):
+        super().__init__()
+        self.server_extension_id = "eim/HA_gateway"
+        self.EXTENSION_ID = f"{self.server_extension_id}/control"  # default eim
+        self.server_file = get_server_file_path("HA_server.py")
+
+
+export = HAControllerExtension

extensions_v2/extension_HCI.py

@@ -0,0 +1,38 @@
+'''
+HCI: human–machine interaction
+本插件支持将任何输入映射为鼠标键盘行为
+
+requirement:
+    pip3 install pyautogui --user
+
+PyAutoGUI only runs on Windows, Mac, and Linux.
+If you lose control and need to stop the current PyAutoGUI function, keep moving the mouse cursor up and to the left. 
+
+tips:
+    currentMouseX, currentMouseY = pyautogui.position()
+    pyautogui.moveTo(100, 150)
+    pyautogui.click()
+    pyautogui.moveRel(None, 10)  # move mouse 10 pixels down
+    pyautogui.typewrite('Hello world!', interval=0.25)
+
+todo:
+    在Scratch3.0中写一个能自己写程序的程序
+'''
+
+from codelab_adapter.core_extension import ControllerExtension
+from codelab_adapter.utils import get_server_file_path
+
+
+class HCIControllerExtension(ControllerExtension):
+    '''
+    use to control VectorNode(server)
+    '''
+
+    def __init__(self):
+        super().__init__()
+        self.server_extension_id = "eim/HCI"
+        self.EXTENSION_ID = f"{self.server_extension_id}/control"  # default eim
+        self.server_file = get_server_file_path("HCI_server.py")
+
+
+export = HCIControllerExtension

extensions_v2/extension_color_tracking.py

@@ -0,0 +1,27 @@
+import subprocess
+import time
+from codelab_adapter.core_extension import ControllerExtension
+from codelab_adapter.utils import get_server_file_path
+from codelab_adapter.utils import get_python3_path
+
+
+class ColorTrackingControllerExtension(ControllerExtension):
+    '''
+    use to control HA server
+    '''
+
+    def __init__(self):
+        super().__init__()
+        self.server_extension_id = "eim/ColorTracking"
+        self.EXTENSION_ID = f"{self.server_extension_id}/control"  # default eim
+        self.server_file = get_server_file_path("color_tracking_server.py")
+
+    def run(self):
+        python3_path = get_python3_path()
+        camera = 1 # 0: mac 1: usb
+        cmd = f'{python3_path} {self.server_file} --camera {camera}' # 0
+        self.server = subprocess.Popen(cmd,shell=True)  # 发送ctrl c，但使用消息意味着可以分布式
+        while self._running:
+            time.sleep(0.1)
+
+export = ColorTrackingControllerExtension

extensions_v2/extension_cozmo.py

@@ -0,0 +1,16 @@
+from codelab_adapter.core_extension import ControllerExtension
+from codelab_adapter.utils import get_server_file_path
+
+class CozmoControllerExtension(ControllerExtension):
+    '''
+    use to control CozmoNode(server)
+    '''
+
+    def __init__(self):
+        super().__init__()
+        self.server_extension_id = "eim/cozmo"
+        self.EXTENSION_ID = f"{self.server_extension_id}/control"  # default eim
+        self.server_file = get_server_file_path("cozmo_server.py")
+
+
+export = CozmoControllerExtension

extensions_v2/extension_eim.py

@@ -15,7 +15,7 @@ def __init__(self):
         self.EXTENSION_ID = "eim"
 
     def extension_message_handle(self, topic, payload):
-        print(topic, payload, type(payload))
+        # print(topic, payload, type(payload))
         if type(payload) == str:
             self.logger.info(f'scratch eim message:{payload}')
             return
@@ -29,7 +29,8 @@ def run(self):
         '''
         i = 0
         while self._running:
-            message = {"payload": {"content": str(i)}} # topic可选
+            message = self.message_template()
+            message["payload"]["content"] = str(i) # topic可选
             self.publish(message)
             time.sleep(1)
             i += 1

extensions_v2/extension_eim_monitor.py

@@ -0,0 +1,42 @@
+import time
+import importlib, sys
+from codelab_adapter import settings
+from codelab_adapter.core_extension import Extension
+import queue
+
+
+class EimMonitorExtension(Extension):
+    def __init__(self):
+        '''
+        参考 home assistant
+        '''
+        super().__init__()
+        self.EXTENSION_ID = "eim"
+        self.q = queue.Queue()
+
+    def extension_message_handle(self, topic, payload):
+        self.q.put(payload)
+
+    def run(self):
+        module_name = "eim_monitor"
+        try:
+            importlib.import_module(module_name)
+        except Exception as e:
+            self.pub_notification(f'{e}')
+            return
+        module = sys.modules["eim_monitor"]
+        importlib.reload(module)
+
+        while self._running:
+            if not self.q.empty():
+                payload = self.q.get()
+                content = payload["content"]
+                response = sys.modules["eim_monitor"].monitor(
+                    content, self.logger)
+                payload["content"] = response
+                message = {"payload": payload}
+                self.publish(message)
+            time.sleep(0.1)
+
+
+export = EimMonitorExtension

extensions_v2/extension_eim_trigger.py

@@ -0,0 +1,42 @@
+import time
+import importlib, sys
+from codelab_adapter import settings
+from codelab_adapter.core_extension import Extension
+import zmq
+'''
+检测文件是否存在 不存在就创建 都在插件里做
+'''
+
+
+class EimTriggerExtension(Extension):
+    def __init__(self):
+        super().__init__()
+        self.EXTENSION_ID = "eim"
+
+    def run(self):
+        module_name = "eim_trigger"
+        try:
+            importlib.import_module(module_name)
+        except Exception as e:
+            self.pub_notification(f'{e}')
+            return
+        module = sys.modules[module_name]
+        importlib.reload(module)
+        while self._running:
+            # monitor返回值被pub
+            # rate = 10
+            # time.sleep(1/rate) # 默认频率是每秒运行十次这个函数
+            try:
+                response = sys.modules[
+                    module_name].trigger()  # 休眠1s，阻塞，stop会报错， 等待退出，todo 非守护进程
+                if response:
+                    message = {"payload": {"content": response}}
+                    self.publish(message)
+            except zmq.error.ZMQError as e:
+                self.logger.error(f'{e}') # 终止之后 发不出消息
+            except Exception as e:
+                self.logger.error(f'{e}')
+                # self.pub_notification(f'{e}')
+
+
+export = EimTriggerExtension

extensions_v2/extension_joystick.py

@@ -0,0 +1,17 @@
+from codelab_adapter.core_extension import ControllerExtension
+from codelab_adapter.utils import get_server_file_path
+
+
+class JoystickControllerExtension(ControllerExtension):
+    '''
+    use to control VectorNode(server)
+    '''
+
+    def __init__(self):
+        super().__init__()
+        self.server_extension_id = "eim/joystick"
+        self.EXTENSION_ID = f"{self.server_extension_id}/control"  # default eim
+        self.server_file = get_server_file_path("joystick_server.py")
+
+
+export = JoystickControllerExtension

extensions_v2/extension_mqtt_gateway.py

@@ -0,0 +1,93 @@
+'''
+to_mqtt
+from_mqtt
+'''
+
+import asyncio
+import os
+import time
+import json
+# from hbmqtt.client import MQTTClient, ClientException
+# from hbmqtt.mqtt.constants import QOS_1, QOS_2
+
+from codelab_adapter.settings import FROM_MQTT_TOPIC, TO_MQTT_TOPIC
+from codelab_adapter.core_extension import Extension
+from codelab_adapter.utils import threaded
+import paho.mqtt.client as mqtt
+
+
+class MqttGatewayExtension(Extension):
+    '''
+    gateway的职责: 转发进出的消息: from_mqtt/to_mqtt
+
+    hbmqtt_pub --url mqtt://guest:[email protected] -t 'eim/mqtt_gateway' -m "hello from hbmqtt"
+    '''
+
+    def __init__(self):
+        self.EXTENSION_ID = "eim/mqtt_gateway"
+        super().__init__()
+        self.set_subscriber_topic(TO_MQTT_TOPIC)  # sub zmq message
+        self.mqtt_sub_topics = [FROM_MQTT_TOPIC]
+
+        # mqtt settings
+        self.mqtt_addr = "iot.codelab.club"
+        self.username = "guest"
+        self.password = "test"
+        self.mqtt_port = 1883
+
+        # mqtt client
+        self.client = mqtt.Client()
+        self.client.on_connect = self.mqtt_on_connect
+        self.client.on_message = self.mqtt_on_message
+        self.client.username_pw_set(self.username, self.password)
+        self.client.connect(self.mqtt_addr, self.mqtt_port, 60)
+        self.client.loop_start()  # as thread
+
+    def exit_message_handle(self, topic, payload):
+        # stop mqtt client
+        self.client.loop_stop()
+        self.terminate()
+
+    def extension_message_handle(self, topic, payload):
+        '''
+        zmq -> mqtt
+        注意 进出消息格式不同
+
+        Q: 无法进入这里吧， 这里要求extension_id一致
+        '''
+        if topic == TO_MQTT_TOPIC:
+            self.logger.info(topic, payload)
+            payload = json.dumps(payload).encode()
+            self.client.publish(topic, payload)
+
+    def mqtt_on_connect(self, client, userdata, flags, rc):
+        self.logger.info(
+            "MQTT Gateway Connected to MQTT {}:{} with result code {}.".format(
+                self.mqtt_addr, self.mqtt_port, str(rc)))
+        # when mqtt is connected to subscribe to mqtt topics
+        if self.mqtt_sub_topics:
+            for sub in self.mqtt_sub_topics:
+                self.client.subscribe(sub)
+
+    def mqtt_on_message(self, client, userdata, msg):
+        '''
+        mqtt -> zmq
+        传递json 而不是content
+        '''
+        topic = msg.topic  # str
+        if topic == FROM_MQTT_TOPIC:
+            m = msg.payload.decode()  # 在通道的两端做好decode和encode
+            payload = json.loads(m)  # json
+            zmq_topic = payload["zmq_topic"]
+            zmq_payload = payload["zmq_payload"]
+            self.logger.info(f'mqtt topic:{topic} payload: {payload}')
+            self.publish_payload(zmq_payload, zmq_topic)
+
+    def run(self):
+
+        while self._running:
+            # to publish mqtt message
+            time.sleep(0.1)
+
+
+export = MqttGatewayExtension