meian

meian/devices/common_model.py

@@ -0,0 +1,114 @@
+# -*- coding:utf-8 -*-
+"""
+@File    :   common_model
+@Author  :   xuxingchen
+@Version :   1.0
+@Contact :   xuxingchen@sinochem.com
+@Desc    :   公用的数据结构
+"""
+import threading
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+# 错误码
+class ErrorCode(Enum):
+    SUCCESS = (0, "全部完成下发")
+    PART_SUCCESS = (10, "部分完成下发")
+    FAILURE = (20, "一个都没下发成功")
+    INPUT_TYPE_ERROR = (30, "入参数据类型异常")
+    NEVER_POST_FACE = (40, "用户未下发过人脸")
+    UNKNOWN = (90, "未知异常")
+
+    def __new__(cls, code, message):
+        obj = object.__new__(cls)
+        obj._value_ = code
+        obj.message = message
+        return obj
+
+
+class BaseInfo(BaseModel):
+    def check(self):
+        for attr in self.__dict__.keys():
+            # if property can be null, default value should not be set to None
+            if self.__dict__[attr] is None:
+                raise ValueError(f"{attr} not allowed to be set to None")
+
+
+class UserData:
+    def __init__(self):
+        self.topic: Optional[str] = None
+        self.topics: list = []
+        self.table_handler = None
+        self.code = None
+        self.token = None
+        self.status: dict = {}
+        self.clients: dict = {}
+        self.client_id = None
+        self.lock = threading.Lock()  # 添加一个锁用于线程同步
+
+    def set_topic(self, value: str):
+        with self.lock:
+            self.topic = value
+
+    def set_topics(self, value: list):
+        with self.lock:
+            self.topics = value
+
+    def set_table_handler(self, value):
+        with self.lock:
+            self.table_handler = value
+
+    def set_code(self, value):
+        with self.lock:
+            self.code = value
+
+    def set_token(self, value):
+        with self.lock:
+            self.token = value
+
+    def set_status_add(self, key, value):
+        with self.lock:
+            self.status[key] = value
+
+    def set_status_remove(self, key):
+        with self.lock:
+            if self.status and key in self.status.keys():
+                self.status.pop(key)
+
+    def get_status(self, key):
+        if self.status and key in self.status.keys():
+            return self.status[key]
+
+    def set_clients(self, value: dict):
+        with self.lock:
+            self.clients = value
+
+    def set_client_add(self, key, value):
+        with self.lock:
+            self.clients[key] = value
+
+    def set_client_id(self, key):
+        with self.lock:
+            self.client_id = key
+
+
+class Project(BaseModel):
+    project_name: str
+    device_id: str
+    subscribe_topic: str
+    publish_topic: str
+    gateway_id: str
+    gateway_sct: str
+    register_type: int
+
+
+class UserInfo(BaseModel):
+    user_id: str
+    device_id: str
+    name: str
+    user_type: int  # 0:业主 1:访客
+    qrcode: str = None  # 当使用二维码时，该值就是美安设备对应的userid
+    face_url: str = None