# -*- coding:utf-8 -*- """ @Author : xuxingchen @Contact : xuxingchen@sinochem.com @Desc : 由服务主动调用的一些方法，实现调用后自动连接mqtt进行任务下发和回馈处理，解耦服务端与mqtt """ import json import time import traceback from typing import Optional from pydantic import BaseModel, field_validator from config import (BROKER_HOST, BROKER_PORT, BROKER_USERNAME, BROKER_PASSWD, PREFIX_PATH, SUB_PATH, SLEEP_TIME, TIMEOUT_SECOND) from config import APP_HOST as HOST_IP from config import IMAGE_SERVER_PORT as PORT from utils import logger from utils.misc import generate_captcha_text, UserData, create_mqtt_client, on_publish, on_disconnect, now_tz_datetime class CallMessage: def __init__(self, service_code, params): self.messageId = generate_captcha_text(16) self.version = "1.0" self.time = int(time.time() * 1000) self.fromUid = "system" self.serviceCode = service_code self.params = params def dict(self): _dict = self.__dict__.copy() for key, value in self.__dict__.items(): if value is None: _dict.pop(key) _dict["from"] = "backend" return _dict def __call__(self, *args, **kwargs): return json.dumps(self.dict()) class AddFaceItem(BaseModel): name: str user_id: str | int device_ids: str start_date: str = now_tz_datetime() phone_number: str = "" face_url: str expire_date: Optional[str] = "2099-01-01T01:01:01.001Z" @field_validator("user_id") def check_user_id(cls, value): if isinstance(value, int): return str(value) return value class DelFaceItem(BaseModel): user_id: str | int device_ids: str @field_validator("user_id") def check_user_id(cls, value): if isinstance(value, int): return str(value) return value ACTION = { "insert": "新增", "update": "更新", "delete": "删除" } class VehicleRegistrationItem(BaseModel): car_plate_no: str # 车牌 registration_id: str owner_id: str owner_name: str registration_time: str begin_time: str end_time: str = "2099-01-01T01:01:01.001Z" action: str # insert, update, delete registration_type: str # 0: 业主，1: 访客 class UpdateFixedCardItem(BaseModel): vehicle_owner_id: str vehicle_owner_name: str vehicle_owner_phone: str plate_number: str # 车牌 effective_date_begin: str effective_date_end: str action: str # insert, update, delete card_id: str # 在新增时进行生成 def on_connect(client, userdata, _flags, rc): logger.Logger.debug(f"🔗 Mqtt connection! {{rc: {rc}}} 🔗") if userdata.topics: _topics = [(topic, 0) for topic in userdata.topics] client.subscribe(_topics) logger.Logger.debug(f"subscribe topics: {userdata.topics}") def on_message(_client, userdata: UserData, message): try: logger.Logger.debug("💡 接收到新数据，执行 on_message 中 ...", log_path=None) msg = json.loads(message.payload.decode().replace("\x00", "")) logger.Logger.info(f"{message.topic}: {msg}") userdata.set_topic(message.topic) if userdata.token: if "messageId" in msg.keys() and userdata.token == msg["messageId"]: # 兼容aiot平台 userdata.set_status_add("status", True) userdata.set_status_add("response", msg) except Exception as e: logger.Logger.error(f"{type(e).__name__}, {e}") if logger.DEBUG: traceback.print_exc() class ServicesCall: def __init__(self): # 创建mqtt连接 self.userdata = UserData() self.client = create_mqtt_client(BROKER_HOST, BROKER_PORT, self.userdata, on_message, on_publish, on_connect, on_disconnect, username=BROKER_USERNAME, password=BROKER_PASSWD) self.client.loop_start() def __del__(self): self.client.loop_stop() def add_face(self, device_id, obj: AddFaceItem) -> tuple[bool, int, str]: if obj.face_url.startswith("http"): face_filename = obj.face_url.split('/')[-1] else: face_filename = obj.face_url obj.face_url = f"http://{HOST_IP}:{PORT}{PREFIX_PATH}/{SUB_PATH}/{face_filename}" topic = f"/jmlink/{device_id}/tml/service/call" self.userdata.set_topic(topic) logger.Logger.debug(f"{'=' * 10} 🚩 AddFace 🚩 {'=' * 10}") logger.Logger.debug(f"{obj.device_ids} 添加用户 {obj.user_id} 的人脸授权") cm = CallMessage("add_face", obj.__dict__) self.userdata.set_message(cm.dict()) # 等待回传 self.userdata.set_token(cm.messageId) self.userdata.set_status_add("status", False) self.userdata.set_status_add("start_timestamp", time.time()) topic_resp = f"/jmlink/{device_id}/tml/service/call_resp" self.client.subscribe(topic_resp) self.client.publish(topic, cm()) try: while True: if self.userdata.status["status"]: if "code" in self.userdata.status["response"].keys(): if self.userdata.status["response"]["code"] != 0: error_code = self.userdata.status['response']['code'] logger.Logger.error( f"用户 - {obj.user_id} 在设备 - {device_id} 上人脸下置失败，错误码 {error_code}") return False, error_code, self.userdata.status['response']['message'] else: logger.Logger.info(f"设备 - {obj.device_ids} 用户({obj.user_id}) 人脸完成授权") return True, 0, "" if time.time() - self.userdata.status["start_timestamp"] > TIMEOUT_SECOND: # 超时跳出 logger.Logger.error("等待回复超时") return False, -1, "等待回复超时" time.sleep(SLEEP_TIME) finally: self.client.unsubscribe(topic_resp) logger.Logger.debug(f"移除订阅: {topic_resp}") self.userdata.set_token(None) self.userdata.set_status_remove("response") logger.Logger.debug(f"{'=' * 10} 🏁 AddFace 🏁 {'=' * 10}") def del_face(self, device_id, obj: DelFaceItem) -> tuple[bool, int, str]: topic = f"/jmlink/{device_id}/tml/service/call" self.userdata.set_topic(topic) logger.Logger.debug(f"{'=' * 10} 🚩 DelFace 🚩 {'=' * 10}") logger.Logger.debug(f"从 {obj.device_ids} 移除用户 {obj.user_id} 的人脸授权") cm = CallMessage("del_face", obj.__dict__) self.userdata.set_message(cm.dict()) # 等待回传 self.userdata.set_token(cm.messageId) self.userdata.set_status_add("status", False) self.userdata.set_status_add("start_timestamp", time.time()) topic_resp = f"/jmlink/{device_id}/tml/service/call_resp" self.client.subscribe(topic_resp) self.client.publish(topic, cm()) try: while True: if self.userdata.status["status"]: if "code" in self.userdata.status["response"].keys(): if self.userdata.status["response"]["code"] != 0: error_code = self.userdata.status['response']['code'] logger.Logger.error( f"用户 - {obj.user_id} 在设备 - {device_id} 上人脸移除失败，错误码 {error_code}") return False, error_code, self.userdata.status['response']['message'] else: logger.Logger.info(f"设备 - {obj.device_ids} 用户({obj.user_id}) 人脸完成移除") return True, 0, "" if time.time() - self.userdata.status["start_timestamp"] > TIMEOUT_SECOND: # 超时跳出 logger.Logger.error("等待回复超时") return False, -1, "等待回复超时" time.sleep(SLEEP_TIME) finally: self.client.unsubscribe(topic_resp) logger.Logger.debug(f"移除订阅: {topic_resp}") self.userdata.set_token(None) self.userdata.set_status_remove("response") logger.Logger.debug(f"{'=' * 10} 🏁 DelFace 🏁 {'=' * 10}") def vehicle_registration(self, device_id, obj: VehicleRegistrationItem) -> tuple[bool, int, str]: topic = f"/jmlink/{device_id}/tml/service/call" self.userdata.set_topic(topic) logger.Logger.debug(f"{'=' * 10} 🚩 VehicleRegistration 🚩 {'=' * 10}") logger.Logger.debug(f"{device_id} {ACTION[obj.action]}车辆 {obj.car_plate_no} 信息") cm = CallMessage("vehicle_registration", obj.__dict__) self.userdata.set_message(cm.dict()) # 等待回传 self.userdata.set_token(cm.messageId) self.userdata.set_status_add("status", False) self.userdata.set_status_add("start_timestamp", time.time()) topic_resp = f"/jmlink/{device_id}/tml/service/call_resp" self.client.subscribe(topic_resp) self.client.publish(topic, cm()) try: while True: if self.userdata.status["status"]: if "code" in self.userdata.status["response"].keys(): if self.userdata.status["response"]["code"] != 0: error_code = self.userdata.status['response']['code'] logger.Logger.error(f"{topic_resp} 返回错误码: {error_code}") return False, error_code, self.userdata.status['response']['message'] else: logger.Logger.info(f"车辆({obj.car_plate_no}) 信息完成{ACTION[obj.action]}") return True, 0, "" if time.time() - self.userdata.status["start_timestamp"] > TIMEOUT_SECOND: # 超时跳出 logger.Logger.error("等待回复超时") return False, -1, "等待回复超时" time.sleep(SLEEP_TIME) finally: self.client.unsubscribe(topic_resp) logger.Logger.debug(f"移除订阅: {topic_resp}") self.userdata.set_token(None) self.userdata.set_status_remove("response") logger.Logger.debug(f"{'=' * 10} 🏁 VehicleRegistration 🏁 {'=' * 10}") def update_fixed_card(self, device_id, obj: UpdateFixedCardItem) -> tuple[bool, int, str]: logger.Logger.debug(f"{'=' * 10} 🚩 UpdateFixedCard 🚩 {'=' * 10}") topic = f"/jmlink/{device_id}/tml/service/call" self.userdata.set_topic(topic) logger.Logger.debug(f"{device_id} {ACTION[obj.action]}用户 {obj.plate_number} 的月卡信息") cm = CallMessage("update_fixed_card", obj.__dict__) self.userdata.set_message(cm.dict()) # 等待回传 self.userdata.set_token(cm.messageId) self.userdata.set_status_add("status", False) self.userdata.set_status_add("start_timestamp", time.time()) topic_resp = f"/jmlink/{device_id}/tml/service/call_resp" self.client.subscribe(topic_resp) self.client.publish(topic, cm()) try: while True: if self.userdata.status["status"]: if "code" in self.userdata.status["response"].keys(): if self.userdata.status["response"]["code"] != 0: error_code = self.userdata.status['response']['code'] logger.Logger.error(f"{topic_resp} 返回错误码: {error_code}") return False, error_code, self.userdata.status['response']['message'] else: logger.Logger.info(f"车辆({obj.plate_number}) 月卡信息完成修改") return True, 0, "" if time.time() - self.userdata.status["start_timestamp"] > TIMEOUT_SECOND: # 超时跳出 logger.Logger.error("等待回复超时") return False, -1, "等待回复超时" time.sleep(SLEEP_TIME) finally: self.client.unsubscribe(topic_resp) logger.Logger.debug(f"移除订阅: {topic_resp}") self.userdata.set_token(None) self.userdata.set_status_remove("response") logger.Logger.debug(f"{'=' * 10} 🏁 UpdateFixedCard 🏁 {'=' * 10}") if __name__ == '__main__': sc = ServicesCall() sc.del_face(2, DelFaceItem( user_id="1234", device_ids=str([2]) )) sc.add_face(2, AddFaceItem( name="测试", user_id="ceshi", face_url="http://ip:port/api/v1/faces/person_demo.jpg", device_ids=str([2, 321]), start_date="2024-01-01T18:00:00.000Z" )) sc.vehicle_registration(2, VehicleRegistrationItem( car_plate_no="鲁G95339", registration_id="5a12265cf73f261148770c12328bb195", owner_id="1244668310399733760", owner_name="测试业主", registration_time="2024-05-27T16:12:26.895Z", begin_time="2024-05-27T16:12:26.895Z", end_time="2034-05-27T16:12:26.895Z", action="insert", registration_type='0' )) sc.update_fixed_card(2, UpdateFixedCardItem( vehicle_owner_id="1244668310399733760", vehicle_owner_name="测试业主", vehicle_owner_phone="1244668310399733760", action="insert", card_id=str(int(time.time() * 1000)), effective_date_begin=now_tz_datetime(), effective_date_end=now_tz_datetime(365 * 10), plate_number="鲁G95339" ))