init
This commit is contained in:
22
python/libraries/LICENSE
Normal file
22
python/libraries/LICENSE
Normal file
@@ -0,0 +1,22 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2026 QmiTek (https://qmikeji.cn)
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
|
||||
7
python/libraries/makergen/__init__.py
Normal file
7
python/libraries/makergen/__init__.py
Normal file
@@ -0,0 +1,7 @@
|
||||
"""适配makergen的Python SDK"""
|
||||
|
||||
__version__ = '0.1.0'
|
||||
|
||||
from makergen.core import Device
|
||||
|
||||
__all__ = ["Device"]
|
||||
85
python/libraries/makergen/core.py
Normal file
85
python/libraries/makergen/core.py
Normal file
@@ -0,0 +1,85 @@
|
||||
import json
|
||||
import threading
|
||||
import time
|
||||
from json import JSONDecodeError
|
||||
from typing import Callable, Optional
|
||||
|
||||
import websocket
|
||||
|
||||
|
||||
class ServerException(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class Device:
|
||||
def __init__(self, device_id: str, device_key: str,
|
||||
on_msg_callback: Optional[Callable[[str, any], None]] = lambda point, value : print(point, value),
|
||||
on_error_callback: Optional[Callable[[Exception], None]] = lambda error: print(error),
|
||||
base_url: str = "wss://api.makergen.cn"):
|
||||
"""
|
||||
创建一个设备对象。需要调用connect方法才能真正连接。
|
||||
:param device_id: 设备ID,在控制台设备管理中显示
|
||||
:param device_key: Device Key,在控制台设备管理中显示
|
||||
:param on_msg_callback: 当收到消息后的回调。接收两个参数,第一个参数为控制点,字符串型;第二个参数为控制值,任意类型
|
||||
:param on_error_callback: 当遇到错误时的回调。接收一个参数,为错误对象;若遇到服务器或页面发送的消息不合规,传入ServerException对象
|
||||
:param base_url: makergen的基础URL,通常为wss://api.makergen.cn
|
||||
"""
|
||||
url = f"{base_url if base_url[-1] == '/' else base_url + "/"}ws/device/{device_id}?key={device_key}"
|
||||
print(url)
|
||||
self.ws = websocket.WebSocketApp(
|
||||
url,
|
||||
on_open=self._on_open,
|
||||
on_message=self._on_message,
|
||||
on_close=self._on_close,
|
||||
on_error=self._on_error,
|
||||
)
|
||||
self.on_msg_callback = on_msg_callback if on_msg_callback is not None else lambda _, __ : None
|
||||
self.on_error_callback = on_error_callback if on_error_callback is not None else lambda _ : None
|
||||
self.is_connected = False
|
||||
|
||||
def _on_open(self, ws):
|
||||
self.is_connected = True
|
||||
def _on_close(self, ws, close_status_code, close_msg):
|
||||
self.is_connected = False
|
||||
|
||||
def _on_message(self, ws, message):
|
||||
try:
|
||||
message_str = json.loads(message)["payload"]
|
||||
self.on_msg_callback(message_str["point"], message_str["value"])
|
||||
except (KeyError, JSONDecodeError):
|
||||
self.on_error_callback(ServerException("Page or server sent an invalid message"))
|
||||
except Exception as e:
|
||||
self.on_error_callback(e)
|
||||
|
||||
def _on_error(self, ws, error):
|
||||
self.on_error_callback(error)
|
||||
|
||||
def send_data(self, point: str, value: str):
|
||||
self.ws.send(json.dumps({
|
||||
"type": "data",
|
||||
"payload": {
|
||||
"point": point,
|
||||
"value": value
|
||||
},
|
||||
"timestamp": int(time.time() * 1000)
|
||||
}))
|
||||
|
||||
def connect(self, timeout: float = 5.0):
|
||||
"""
|
||||
在新线程中运行客户端循环(推荐)
|
||||
"""
|
||||
wst = threading.Thread(target=self.ws.run_forever)
|
||||
wst.daemon = True
|
||||
wst.start()
|
||||
start_time = time.time()
|
||||
while not self.is_connected:
|
||||
if time.time() - start_time > timeout:
|
||||
if not self.is_connected:
|
||||
self.on_error_callback(TimeoutError(f"WebSocket connection timed out after {timeout} seconds"))
|
||||
time.sleep(0.05)
|
||||
|
||||
def main_loop(self):
|
||||
"""
|
||||
在当前线程中运行客户端循环(不推荐)
|
||||
"""
|
||||
self.ws.run_forever()
|
||||
1
python/libraries/requirements.txt
Normal file
1
python/libraries/requirements.txt
Normal file
@@ -0,0 +1 @@
|
||||
websocket-client==1.9.0
|
||||
Reference in New Issue
Block a user