python-library-xiaomi-miot 0.1.0__py3-none-any.whl

python_library_xiaomi_miot-0.1.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: python-library-xiaomi-miot
+Version: 0.1.0
+Requires-Python: >=3.10
+Requires-Dist: pydantic
+Requires-Dist: python-miio

python_library_xiaomi_miot-0.1.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+xiaomi_miot/__init__.py,sha256=xMnX3Icdofj0fRNQC5_NZGyLq9G-6dqAX8l019TJ0dQ,432
+xiaomi_miot/_api.py,sha256=4dN17bEFnbkMFQkyyXK7FSyAngG0jcVnKNArIeJYZdQ,1838
+xiaomi_miot/device.py,sha256=YA8UVq0uIEoBLmvzMhUyIFKC_fQewpMHbTkl-otzjPI,1693
+xiaomi_miot/models.py,sha256=7X5L1ClsPYxJdDQrAEbqRl-PK49CLwGPB5ptbSaPZnE,739
+xiaomi_miot/extensions/__init__.py,sha256=CgSFv8opc31WFVXFga785p5lV3e2oPSNhucRgIvvZmE,146
+xiaomi_miot/extensions/_registry.py,sha256=vdbS7XAEqc-HyuGs22iBDiA4Gt2rW87QpyfDzKC7x-s,1064
+xiaomi_miot/extensions/switch.py,sha256=iV3xfu1mmTq9KxoWfMWnkTlKZOogIuCvS137ZP_D6uk,1300
+python_library_xiaomi_miot-0.1.0.dist-info/METADATA,sha256=-7TzQTGoWZMGvRtqMiowqWOpSo1JyGFPaEk2U1qABLo,145
+python_library_xiaomi_miot-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+python_library_xiaomi_miot-0.1.0.dist-info/RECORD,,

python_library_xiaomi_miot-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

xiaomi_miot/__init__.py

@@ -0,0 +1,17 @@
+import warnings
+warnings.filterwarnings("ignore", category=FutureWarning, module=r"miio\.miot_device")
+
+from .device import MiotDevice
+from .models import DeviceParams, GetParams, SetParams
+from .extensions import get_extension, list_extensions
+from ._api import execute
+
+__all__ = [
+    "MiotDevice",
+    "DeviceParams",
+    "GetParams",
+    "SetParams",
+    "get_extension",
+    "list_extensions",
+    "execute",
+]

xiaomi_miot/_api.py

@@ -0,0 +1,54 @@
+from .device import MiotDevice
+from .models import DeviceParams, GetParams, SetParams
+from .extensions import get_extension, list_extensions
+
+
+def execute(params: dict) -> dict:
+    """统一入口，用户输入永远是 dict
+
+    通用模式（无 type）:
+        {"ip": "...", "token": "...", "siid": 2, "piid": 1}              → 读属性
+        {"ip": "...", "token": "...", "siid": 2, "piid": 1, "value": 1}  → 写属性
+        {"ip": "...", "token": "...", "action": "info"}                   → 设备信息
+
+    扩展模式（有 type）:
+        {"type": "switch", "ip": "...", "token": "...", "on": True}       → 走开关扩展
+    """
+    device_type = params.get("type")
+
+    if device_type:
+        return _execute_extension(device_type, params)
+    return _execute_generic(params)
+
+
+def _execute_extension(device_type: str, params: dict) -> dict:
+    ext_cls = get_extension(device_type)
+    if ext_cls is None:
+        return {
+            "ok": False,
+            "error": f"未知设备类型: {device_type}",
+            "available": list_extensions(),
+        }
+
+    ext = ext_cls()
+    validated = ext.Params(**params)
+    device = MiotDevice(validated.ip, validated.token)
+    return ext.execute(device, validated)
+
+
+def _execute_generic(params: dict) -> dict:
+    action = params.get("action", "prop")
+
+    if action == "info":
+        dp = DeviceParams(**params)
+        device = MiotDevice(dp.ip, dp.token)
+        return {"ok": True, **device.info()}
+
+    if "value" in params:
+        sp = SetParams(**params)
+        device = MiotDevice(sp.ip, sp.token)
+        return device.set_prop(sp.did, sp.siid, sp.piid, sp.value)
+
+    gp = GetParams(**params)
+    device = MiotDevice(gp.ip, gp.token)
+    return device.get_prop(gp.did, gp.siid, gp.piid)

xiaomi_miot/device.py

@@ -0,0 +1,48 @@
+from miio import Device
+
+
+class MiotDevice:
+    """纯局域网 MIoT 设备操作，不依赖云端"""
+
+    def __init__(self, ip: str, token: str):
+        self._dev = Device(ip, token)
+
+    def info(self) -> dict:
+        """获取设备基本信息（model, firmware, hardware 等）"""
+        raw = self._dev.info()
+        return {
+            "model": raw.model,
+            "mac": raw.mac_address,
+            "firmware": raw.firmware_version,
+            "hardware": raw.hardware_version,
+            "raw": str(raw),
+        }
+
+    def get_prop(self, did: str, siid: int, piid: int) -> dict:
+        results = self._dev.send("get_properties", [
+            {"did": did, "siid": siid, "piid": piid}
+        ])
+        r = results[0]
+        if r.get("code") == 0:
+            return {"ok": True, "value": r["value"]}
+        return {"ok": False, "error": r}
+
+    def get_props(self, props: list[dict]) -> list[dict]:
+        return self._dev.send("get_properties", props)
+
+    def set_prop(self, did: str, siid: int, piid: int, value) -> dict:
+        results = self._dev.send("set_properties", [
+            {"did": did, "siid": siid, "piid": piid, "value": value}
+        ])
+        r = results[0]
+        if r.get("code") == 0:
+            return {"ok": True}
+        return {"ok": False, "error": r}
+
+    def set_props(self, props: list[dict]) -> list[dict]:
+        return self._dev.send("set_properties", props)
+
+    def call_action(self, did: str, siid: int, aiid: int, params: list | None = None) -> dict:
+        return self._dev.send("action", {
+            "did": did, "siid": siid, "aiid": aiid, "in": params or []
+        })

xiaomi_miot/extensions/__init__.py

@@ -0,0 +1,7 @@
+from ._registry import extension, get_extension, list_extensions
+
+__all__ = [
+    "extension",
+    "get_extension",
+    "list_extensions",
+]

xiaomi_miot/extensions/_registry.py

@@ -0,0 +1,44 @@
+import importlib
+import pkgutil
+from pathlib import Path
+
+_extensions: dict[str, type] = {}
+_discovered = False
+
+
+def extension(name: str):
+    """装饰器：注册设备扩展
+
+    @extension("switch")
+    class SwitchExtension:
+        class Params(DeviceParams): ...
+        def execute(self, device, params): ...
+    """
+    def decorator(cls):
+        _extensions[name] = cls
+        return cls
+    return decorator
+
+
+def get_extension(name: str):
+    discover()
+    return _extensions.get(name)
+
+
+def list_extensions() -> list[str]:
+    discover()
+    return list(_extensions.keys())
+
+
+def discover():
+    """动态扫描 extensions/ 目录下所有模块，导入以触发 @extension 注册"""
+    global _discovered
+    if _discovered:
+        return
+    _discovered = True
+
+    ext_dir = Path(__file__).parent
+    for finder, module_name, is_pkg in pkgutil.iter_modules([str(ext_dir)]):
+        if module_name.startswith("_"):
+            continue
+        importlib.import_module(f"{__package__}.{module_name}")

xiaomi_miot/extensions/switch.py

@@ -0,0 +1,31 @@
+from pydantic import BaseModel, Field
+from xiaomi_miot.device import MiotDevice
+from xiaomi_miot.extensions import extension
+from xiaomi_miot.models import DeviceParams
+
+class SwitchParams(DeviceParams):
+    type: str = "switch"
+    on: bool | None = Field(default=None,description="None=查询状态, True=开, False=关")
+    siid: int = Field(default=2,description="默认 service：Switch")
+    piid: int = Field(default=1,description="默认 property：On")
+
+@extension("switch")
+class SwitchExtension:
+    Params = SwitchParams
+
+    def execute(self, device: MiotDevice, params: SwitchParams) -> dict:
+        if params.on is None:
+            return self._get_state(device, params)
+        return self._set_state(device, params)
+
+    def _get_state(self, device: MiotDevice, params: SwitchParams) -> dict:
+        result = device.get_prop("switch", params.siid, params.piid)
+        if result["ok"]:
+            result["state"] = "开启" if result["value"] else "关闭"
+        return result
+
+    def _set_state(self, device: MiotDevice, params: SwitchParams) -> dict:
+        result = device.set_prop("switch", params.siid, params.piid, params.on)
+        if result["ok"]:
+            result["state"] = "开启" if params.on else "关闭"
+        return result

xiaomi_miot/models.py

@@ -0,0 +1,22 @@
+from typing import Any
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class DeviceParams(BaseModel):
+    """最基础的设备参数：局域网连接所需"""
+    model_config = ConfigDict(extra="ignore")
+
+    ip: str = Field(description="设备 IP 地址")
+    token: str = Field(description="设备 Token")
+    type: str | None = Field(default=None,description="设备类型")
+
+class GetParams(DeviceParams):
+    """通用属性读取"""
+    did: str = Field(default="prop",description="默认 did：prop")
+    siid: int = Field(description="service ID")
+    piid: int = Field(description="property ID")
+
+
+class SetParams(GetParams):
+    """通用属性设置"""
+    value: Any = Field(description="属性值")