alarm_plan.py 42 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103
  1. from typing import List, Tuple, Optional
  2. from datetime import datetime, time, date
  3. from threading import Thread, Lock
  4. from enum import Enum
  5. import uuid
  6. import json
  7. import traceback
  8. from datetime import datetime, timezone, timedelta
  9. from common.sys_comm import (
  10. LOGDBG, LOGINFO, LOGWARN, LOGERR,
  11. get_utc_time_ms, get_utc_time_s, get_bj_time_ms, get_bj_time_s,
  12. utc_to_bj_ms, bj_to_utc_ms, utc_to_bj_s, bj_to_utc_s
  13. )
  14. from core.time_plan import TimePlan
  15. from core.event_type import EventType, event_desc_map
  16. import core.g_LAS as g_las
  17. import core.alarm_plan_helper as helper
  18. from device.dev_mng import (
  19. Device,
  20. dev_map_push, dev_map_pop, dev_map_find, dev_map_delete
  21. )
  22. from db.db_process import (
  23. db_req_que, DBRequest
  24. )
  25. import db.db_sqls as sqls
  26. import mqtt.mqtt_send as mqtt_send
  27. from device.dev_mng import g_dev_map, g_dev_map_lock
  28. class EventAttr_Base:
  29. def __init__(self):
  30. return
  31. # 事件属性 事件事件
  32. class EventAttr_StayDetection(EventAttr_Base):
  33. def __init__(self, event_type):
  34. self.event_type_ = event_type
  35. self.enter_ts_: int = -1 # 进入时间(s)
  36. self.leave_ts_: int = -1 # 离开时间(s)
  37. self.stay_time_: int = -1 # 停留时长(s)
  38. return
  39. def reset(self):
  40. self.enter_ts_ = -1
  41. self.leave_ts_ = -1
  42. self.stay_time_ = -1
  43. # 事件属性 滞留事件
  44. class EventAttr_RetentionDetection(EventAttr_Base):
  45. def __init__(self, event_type):
  46. self.event_type_ = event_type
  47. self.enter_ts_: int = -1 # 进入时间(s)
  48. self.leave_ts_: int = -1 # 离开时间(s)
  49. self.stay_time_: int = -1 # 停留时长(s)
  50. return
  51. def reset(self):
  52. self.enter_ts_ = -1
  53. self.leave_ts_ = -1
  54. self.stay_time_ = -1
  55. # 事件属性 如厕事件
  56. class EventAttr_ToiletingDetection(EventAttr_Base):
  57. def __init__(self, event_type):
  58. self.event_type_ = event_type
  59. self.enter_ts_: int = -1 # 进入时间(ms)
  60. self.leave_ts_: int = -1 # 离开时间(ms)
  61. self.stay_time_: int = -1 # 停留时长(ms)
  62. return
  63. def reset(self):
  64. self.enter_ts_ = -1
  65. self.leave_ts_ = -1
  66. self.stay_time_ = -1
  67. # 事件属性 如厕频次统计
  68. class EventAttr_ToiletingFrequency(EventAttr_Base):
  69. def __init__(self, event_type):
  70. self.event_type_ = event_type
  71. self.count_: int = 0 # 统计次数
  72. self.event_list: list = []
  73. return
  74. def reset(self):
  75. self.count_ = 0
  76. self.event_list = []
  77. # 事件属性 夜间如厕频次统计
  78. class EventAttr_NightToiletingFrequency(EventAttr_Base):
  79. def __init__(self, event_type):
  80. self.event_type_ = event_type
  81. self.count_: int = 0 # 统计次数
  82. self.event_list: list = []
  83. return
  84. def reset(self):
  85. self.count_ = 0
  86. self.event_list = []
  87. # 事件属性 如厕频次异常
  88. class EventAttr_ToiletingFrequencyAbnormal(EventAttr_Base):
  89. def __init__(self, event_type):
  90. self.event_type_ = event_type
  91. self.count_: int = 0 # 统计次数
  92. self.threshold_count_: int = 0 # 异常阈值
  93. self.event_list: list = []
  94. return
  95. def reset(self):
  96. self.count_ = 0
  97. self.event_list = []
  98. # 事件属性 起夜异常
  99. class EventAttr_NightToiletingFrequencyAbnormal(EventAttr_Base):
  100. def __init__(self, event_type):
  101. self.event_type_ = event_type
  102. self.count_: int = 0 # 统计次数
  103. self.threshold_count_: int = 0 # 异常阈值
  104. self.event_list: list = []
  105. return
  106. def reset(self):
  107. self.count_ = 0
  108. self.event_list = []
  109. # 事件属性 卫生间频次统计
  110. class EventAttr_BathroomStayFrequency(EventAttr_Base):
  111. def __init__(self, event_type):
  112. self.event_type_ = event_type
  113. self.count_: int = 0 # 统计次数
  114. self.event_list: list = []
  115. return
  116. def reset(self):
  117. self.count_ = 0
  118. self.event_list = []
  119. # 事件属性 异常消失
  120. class EventAttr_TargetAbsence(EventAttr_Base):
  121. def __init__(self, event_type):
  122. self.event_type_ = event_type
  123. self.leave_ts_: int = -1 # 离开时间(ms)
  124. self.enter_ts_: int = -1 # 进入时间(ms)
  125. self.absence_time_: int = -1 # 消失时长(ms)
  126. self.time_threshold_: int = 300 # 触发消失时间阈值(ms)
  127. return
  128. def reset(self):
  129. self.leave_ts_ = -1
  130. self.enter_ts_ = -1
  131. self.absence_time_ = -1
  132. # 事件属性表
  133. event_attr_map = {
  134. EventType.STAY_DETECTION.value : EventAttr_StayDetection,
  135. EventType.RETENTION_DETECTION.value : EventAttr_RetentionDetection,
  136. EventType.TOILETING_DETECTION.value : EventAttr_ToiletingDetection,
  137. EventType.TOILETING_FREQUENCY.value : EventAttr_ToiletingFrequency,
  138. EventType.NIGHT_TOILETING_FREQUENCY.value : EventAttr_NightToiletingFrequency,
  139. EventType.TOILETING_FREQUENCY_ABNORMAL.value : EventAttr_ToiletingFrequencyAbnormal,
  140. EventType.NIGHT_TOILETING_FREQUENCY_ABNORMAL.value: EventAttr_NightToiletingFrequencyAbnormal,
  141. EventType.BATHROOM_STAY_FREQUENCY.value : EventAttr_BathroomStayFrequency,
  142. EventType.TARGET_ABSENCE.value : EventAttr_TargetAbsence,
  143. }
  144. class Cron:
  145. def __init__(self, h, m, s):
  146. self.h_ = None
  147. self.m_ = None
  148. self.s_ = None
  149. class AlarmPlan:
  150. def __init__(self,
  151. plan_uuid: str,
  152. name: str,
  153. dev_id: str,
  154. dev_name: str,
  155. enable: bool,
  156. time_plan: TimePlan,
  157. rect: list,
  158. event_type: int,
  159. threshold_time: int,
  160. merge_time: int,
  161. param: dict,
  162. cron: Optional[dict] = None
  163. ):
  164. self.lock_ = Lock()
  165. self.plan_uuid_ = plan_uuid # 计划id
  166. self.name_ = name # 计划名称
  167. self.dev_id_ = dev_id # 设备id
  168. self.dev_name_ = dev_name # 设备名称
  169. self.enable_ = enable # 是否启用
  170. self.time_plan_ = time_plan # 时间计划
  171. self.param_ = param # 参数
  172. # 维护状态(根据TimePlanu判断)
  173. self.status_ = 0 # 0未激活,1激活,-1过期
  174. self.status_update_ts_ = -1 # 状态更新时间,初始值为-1
  175. # 事件触发参数
  176. self.rect_ = rect # 检测区域 [left, top, width, height]
  177. self.threshold_time_ = threshold_time # 触发时间阈值
  178. self.merge_time_ = merge_time # 归并时间窗口
  179. self.event_type_ = event_type # 事件类型
  180. self.event_attr_ = self.init_event_attr() # 事件属性
  181. if self.event_attr_ is None:
  182. raise ValueError(f"Invalid event_type: {event_type}")
  183. self.cron_ = cron # {“hour": 6, "minute": 0}
  184. def execute(self):
  185. if self.status_ != 1:
  186. return
  187. g_las.g_event_dispatcher.dispatch(self.event_type_, self)
  188. # 初始化事件属性
  189. def init_event_attr(self):
  190. try:
  191. event_cls = event_attr_map.get(self.event_type_)
  192. if event_cls is None:
  193. return None
  194. event_attr = event_cls(self.event_type_)
  195. if ((self.event_type_ == EventType.TOILETING_FREQUENCY_ABNORMAL.value) or
  196. (self.event_type_ == EventType.NIGHT_TOILETING_FREQUENCY_ABNORMAL.value)):
  197. event_attr.threshold_count_ = int(self.param_.get("count", 0))
  198. if ((self.event_type_ == EventType.TARGET_ABSENCE.value)):
  199. event_attr.time_threshold_ = int(self.param_.get("time_threshold", 0))
  200. return event_attr
  201. except json.JSONDecodeError as e:
  202. tb_info = traceback.extract_tb(e.__traceback__)
  203. for frame in tb_info:
  204. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  205. except Exception as e:
  206. tb_info = traceback.extract_tb(e.__traceback__)
  207. for frame in tb_info:
  208. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  209. # 更新激活状态
  210. def update_status(self, now: Optional[datetime] = None) -> None:
  211. now = now or datetime.now()
  212. old_status = self.status_
  213. if not self.enable_:
  214. self.status_ = 0
  215. else:
  216. now_fmt = now.strftime("%Y-%m-%d")
  217. # 过期
  218. if now_fmt > self.time_plan_.stop_date_:
  219. self.status_ = -1
  220. elif now_fmt < self.time_plan_.start_date_:
  221. self.status_ = 0
  222. elif self.time_plan_.is_active_now(now):
  223. self.status_ = 1
  224. else:
  225. self.status_ = 0
  226. if self.status_ != old_status:
  227. self.status_update_ts = int(now.timestamp())
  228. LOGINFO(f"[Status] plan {self.plan_uuid_} status_ changed {old_status} -> {self.status_}")
  229. def is_point_in_rect(self, x:float, y:float, rect:list) -> bool:
  230. rx, ry, rw, rh = rect
  231. x_min = min(rx, rx + rw)
  232. x_max = max(rx, rx + rw)
  233. y_min = min(ry, ry - rh)
  234. y_max = max(ry, ry - rh)
  235. bRet: bool = x_min <= x <= x_max and y_min <= y <= y_max
  236. return bRet
  237. # 查找最近 t 秒内,最后一个落在 rect 内的 target_point 的 rtd_unit
  238. def find_latest_rtd_in_region(self, device: Device, rect: list, now: int=None, t: int=1):
  239. now_s = now if now else get_utc_time_s()
  240. rtd_que_copy = device.get_rtd_que_copy()
  241. with self.lock_:
  242. for rtd_unit in reversed(rtd_que_copy): # 倒序扫描
  243. ts_s = int(rtd_unit.get("timestamp", 0))
  244. if now_s - ts_s > t:
  245. break # 已经超过 t 秒,可以直接结束
  246. # 检查点是否在区域内
  247. for pt in rtd_unit.get("target_point", []):
  248. if len(pt) >= 2:
  249. x, y = pt[0], pt[1]
  250. if self.is_point_in_rect(x, y, rect):
  251. return rtd_unit
  252. return None
  253. # 停留事件
  254. def handle_stay_detection(self):
  255. try:
  256. dev_id = self.dev_id_
  257. device:Device = dev_map_find(dev_id)
  258. if not device:
  259. return
  260. now = get_utc_time_s()
  261. # 查找最新的落在检测区域的目标
  262. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  263. if rtd_unit:
  264. timestamp = rtd_unit["timestamp"]
  265. pose = rtd_unit["pose"]
  266. target_point = rtd_unit["target_point"]
  267. if self.event_attr_.enter_ts_ == -1:
  268. self.event_attr_.enter_ts_ = timestamp
  269. else:
  270. self.event_attr_.leave_ts_ = timestamp
  271. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  272. return
  273. # 归并时间内,不认为事件结束
  274. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  275. return
  276. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  277. stay_time =self.event_attr_.stay_time_
  278. # 时间小于触发时间阈值,忽略并重置
  279. if stay_time < self.threshold_time_ :
  280. self.event_attr_.reset()
  281. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  282. return
  283. # 构造事件
  284. # 入库
  285. info = {
  286. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  287. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  288. "stay_time": stay_time
  289. }
  290. event_uuid = str(uuid.uuid4())
  291. params = {
  292. "dev_id": dev_id,
  293. "uuid": event_uuid,
  294. "plan_uuid": self.plan_uuid_,
  295. "event_type": event_desc_map[self.event_type_],
  296. "info": json.dumps(info),
  297. "is_handle": 0,
  298. "create_time": get_bj_time_s(),
  299. "is_deleted": 0,
  300. "remark": json.dumps({}, ensure_ascii=False)
  301. }
  302. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  303. # 通知
  304. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, "events")
  305. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  306. self.event_attr_.reset()
  307. except json.JSONDecodeError as e:
  308. tb_info = traceback.extract_tb(e.__traceback__)
  309. for frame in tb_info:
  310. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  311. except Exception as e:
  312. tb_info = traceback.extract_tb(e.__traceback__)
  313. for frame in tb_info:
  314. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  315. # 滞留事件
  316. def handle_retention_detection(self):
  317. try:
  318. dev_id = self.dev_id_
  319. device:Device = dev_map_find(dev_id)
  320. if not device:
  321. return
  322. now = get_utc_time_s()
  323. # 查找最新的落在检测区域的目标
  324. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  325. if rtd_unit:
  326. timestamp = rtd_unit["timestamp"]
  327. pose = rtd_unit["pose"]
  328. target_point = rtd_unit["target_point"]
  329. if self.event_attr_.enter_ts_ == -1:
  330. self.event_attr_.enter_ts_ = timestamp
  331. else:
  332. self.event_attr_.leave_ts_ = timestamp
  333. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  334. return
  335. # 归并时间内,不认为事件结束
  336. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  337. return
  338. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  339. stay_time =self.event_attr_.stay_time_
  340. # 时间小于触发时间阈值,忽略并重置
  341. if stay_time < self.threshold_time_ :
  342. self.event_attr_.reset()
  343. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  344. return
  345. # 构造事件
  346. # 入库
  347. info = {
  348. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  349. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  350. "stay_time": stay_time
  351. }
  352. event_uuid = str(uuid.uuid4())
  353. params = {
  354. "dev_id": dev_id,
  355. "uuid": event_uuid,
  356. "plan_uuid": self.plan_uuid_,
  357. "event_type": event_desc_map[self.event_type_],
  358. "info": json.dumps(info),
  359. "is_handle": 0,
  360. "create_time": get_bj_time_s(),
  361. "is_deleted": 0,
  362. "remark": json.dumps({}, ensure_ascii=False)
  363. }
  364. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  365. # 通知
  366. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, "events")
  367. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  368. self.event_attr_.reset()
  369. except json.JSONDecodeError as e:
  370. tb_info = traceback.extract_tb(e.__traceback__)
  371. for frame in tb_info:
  372. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  373. except Exception as e:
  374. tb_info = traceback.extract_tb(e.__traceback__)
  375. for frame in tb_info:
  376. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  377. # 如厕事件
  378. def handle_toileting_detection(self):
  379. try:
  380. dev_id = self.dev_id_
  381. device:Device = dev_map_find(dev_id)
  382. if not device:
  383. return
  384. now = get_utc_time_s()
  385. # 查找最新的落在检测区域的目标
  386. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  387. if rtd_unit:
  388. timestamp = rtd_unit["timestamp"]
  389. pose = rtd_unit["pose"]
  390. target_point = rtd_unit["target_point"]
  391. if self.event_attr_.enter_ts_ == -1:
  392. self.event_attr_.enter_ts_ = timestamp
  393. else:
  394. self.event_attr_.leave_ts_ = timestamp
  395. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  396. return
  397. # 归并时间内,不认为事件结束
  398. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  399. return
  400. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  401. stay_time =self.event_attr_.stay_time_
  402. # 时间小于触发时间阈值,忽略并重置
  403. if stay_time < self.threshold_time_ :
  404. self.event_attr_.reset()
  405. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  406. return
  407. # 构造事件
  408. # 入库
  409. info = {
  410. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  411. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  412. "stay_time": stay_time
  413. }
  414. event_uuid = str(uuid.uuid4())
  415. params = {
  416. "dev_id": dev_id,
  417. "uuid": event_uuid,
  418. "plan_uuid": self.plan_uuid_,
  419. "event_type": event_desc_map[self.event_type_],
  420. "info": json.dumps(info),
  421. "is_handle": 0,
  422. "create_time": get_bj_time_s(),
  423. "is_deleted": 0,
  424. "remark": json.dumps({}, ensure_ascii=False)
  425. }
  426. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  427. # 通知
  428. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, "events")
  429. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  430. self.event_attr_.reset()
  431. except json.JSONDecodeError as e:
  432. tb_info = traceback.extract_tb(e.__traceback__)
  433. for frame in tb_info:
  434. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  435. except Exception as e:
  436. tb_info = traceback.extract_tb(e.__traceback__)
  437. for frame in tb_info:
  438. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  439. # 如厕频次统计
  440. def handle_toileting_frequency(self):
  441. try:
  442. dev_id = self.dev_id_
  443. device:Device = dev_map_find(dev_id)
  444. if not device:
  445. return
  446. start_dt, end_dt = helper.get_query_time_range(self.param_)
  447. params = {
  448. "dev_id" : self.dev_id_,
  449. "event_type": event_desc_map[EventType.TOILETING_DETECTION.value],
  450. "start_dt": start_dt,
  451. "end_dt": end_dt
  452. }
  453. userdata = {
  454. "start_dt" : start_dt,
  455. "end_dt" : end_dt
  456. }
  457. db_req_que.put(DBRequest(
  458. sql=sqls.sql_query_events_by_datetime,
  459. params=params,
  460. callback=self.cb_toileting_frequency,
  461. userdata=userdata))
  462. except json.JSONDecodeError as e:
  463. tb_info = traceback.extract_tb(e.__traceback__)
  464. for frame in tb_info:
  465. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  466. except Exception as e:
  467. tb_info = traceback.extract_tb(e.__traceback__)
  468. for frame in tb_info:
  469. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  470. # 夜间如厕频次统计
  471. def handle_night_toileting_frequency(self):
  472. try:
  473. dev_id = self.dev_id_
  474. device:Device = dev_map_find(dev_id)
  475. if not device:
  476. return
  477. start_dt, end_dt = helper.get_query_time_range(self.param_)
  478. params = {
  479. "dev_id" : self.dev_id_,
  480. "event_type": event_desc_map[EventType.TOILETING_DETECTION.value],
  481. "start_dt": start_dt,
  482. "end_dt": end_dt
  483. }
  484. userdata = {
  485. "start_dt" : start_dt,
  486. "end_dt" : end_dt
  487. }
  488. db_req_que.put(DBRequest(
  489. sql=sqls.sql_query_events_by_datetime,
  490. params=params,
  491. callback=self.cb_night_toileting_frequency,
  492. userdata=userdata))
  493. except json.JSONDecodeError as e:
  494. tb_info = traceback.extract_tb(e.__traceback__)
  495. for frame in tb_info:
  496. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  497. except Exception as e:
  498. tb_info = traceback.extract_tb(e.__traceback__)
  499. for frame in tb_info:
  500. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  501. # 如厕频次异常
  502. def handle_toileting_frequency_abnormal(self):
  503. try:
  504. dev_id = self.dev_id_
  505. device:Device = dev_map_find(dev_id)
  506. if not device:
  507. return
  508. start_dt, end_dt = helper.get_query_time_range(self.param_)
  509. params = {
  510. "dev_id" : self.dev_id_,
  511. "event_type": event_desc_map[EventType.TOILETING_DETECTION.value],
  512. "start_dt": start_dt,
  513. "end_dt": end_dt
  514. }
  515. userdata = {
  516. "start_dt" : start_dt,
  517. "end_dt" : end_dt
  518. }
  519. db_req_que.put(DBRequest(
  520. sql=sqls.sql_query_events_by_datetime,
  521. params=params,
  522. callback=self.cb_toileting_frequency_abnormal,
  523. userdata=userdata))
  524. except json.JSONDecodeError as e:
  525. tb_info = traceback.extract_tb(e.__traceback__)
  526. for frame in tb_info:
  527. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  528. except Exception as e:
  529. tb_info = traceback.extract_tb(e.__traceback__)
  530. for frame in tb_info:
  531. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  532. # 起夜异常
  533. def handle_night_toileting_frequency_abnormal(self):
  534. try:
  535. dev_id = self.dev_id_
  536. device:Device = dev_map_find(dev_id)
  537. if not device:
  538. return
  539. start_dt, end_dt = helper.get_query_time_range(self.param_)
  540. params = {
  541. "dev_id" : self.dev_id_,
  542. "event_type": event_desc_map[EventType.TOILETING_DETECTION.value],
  543. "start_dt": start_dt,
  544. "end_dt": end_dt
  545. }
  546. userdata = {
  547. "start_dt" : start_dt,
  548. "end_dt" : end_dt
  549. }
  550. db_req_que.put(DBRequest(
  551. sql=sqls.sql_query_events_by_datetime,
  552. params=params,
  553. callback=self.cb_toileting_frequency_abnormal,
  554. userdata=userdata))
  555. except json.JSONDecodeError as e:
  556. tb_info = traceback.extract_tb(e.__traceback__)
  557. for frame in tb_info:
  558. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  559. except Exception as e:
  560. tb_info = traceback.extract_tb(e.__traceback__)
  561. for frame in tb_info:
  562. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  563. # 卫生间频次统计
  564. def handle_bathroom_stay_frequency(self):
  565. try:
  566. dev_id = self.dev_id_
  567. device:Device = dev_map_find(dev_id)
  568. if not device:
  569. return
  570. start_dt, end_dt = helper.get_query_time_range(self.param_)
  571. params = {
  572. "dev_id" : self.dev_id_,
  573. "event_type": event_desc_map[EventType.STAY_DETECTION.value],
  574. "start_dt": start_dt,
  575. "end_dt": end_dt
  576. }
  577. userdata = {
  578. "start_dt" : start_dt,
  579. "end_dt" : end_dt
  580. }
  581. db_req_que.put(DBRequest(
  582. sql=sqls.sql_query_events_by_datetime,
  583. params=params,
  584. callback=self.cb_bathroom_stay_frequency,
  585. userdata=userdata))
  586. except json.JSONDecodeError as e:
  587. tb_info = traceback.extract_tb(e.__traceback__)
  588. for frame in tb_info:
  589. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  590. except Exception as e:
  591. tb_info = traceback.extract_tb(e.__traceback__)
  592. for frame in tb_info:
  593. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  594. # 异常消失
  595. def handle_target_absence(self):
  596. try:
  597. dev_id = self.dev_id_
  598. device:Device = dev_map_find(dev_id)
  599. if not device:
  600. return
  601. now = get_utc_time_s()
  602. threshold: int = self.event_attr_.time_threshold_
  603. # 查找最新的落在检测区域的目标
  604. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  605. if rtd_unit:
  606. return
  607. timestamp = get_utc_time_s()
  608. if self.event_attr_.leave_ts_ == -1:
  609. self.event_attr_.leave_ts_ = timestamp
  610. else:
  611. self.event_attr_.enter_ts_ = timestamp
  612. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  613. return
  614. # 归并时间内,不认为事件结束
  615. if now - self.event_attr_.enter_ts_ < self.merge_time_:
  616. return
  617. self.event_attr_.absence_time_ = self.event_attr_.enter_ts_ - self.event_attr_.leave_ts_
  618. absence_time = self.event_attr_.absence_time_
  619. # 时间小于触发时间阈值,忽略并重置
  620. if absence_time < threshold :
  621. self.event_attr_.reset()
  622. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  623. return
  624. # 构造事件
  625. # 入库
  626. info = {
  627. "start_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  628. "end_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  629. "absence_time": absence_time
  630. }
  631. event_uuid = str(uuid.uuid4())
  632. params = {
  633. "dev_id": dev_id,
  634. "uuid": event_uuid,
  635. "plan_uuid": self.plan_uuid_,
  636. "event_type": event_desc_map[self.event_type_],
  637. "info": json.dumps(info),
  638. "is_handle": 0,
  639. "create_time": get_bj_time_s(),
  640. "is_deleted": 0,
  641. "remark": json.dumps({}, ensure_ascii=False)
  642. }
  643. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  644. # 通知
  645. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, "events")
  646. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, absence_time: {absence_time}")
  647. self.event_attr_.reset()
  648. except json.JSONDecodeError as e:
  649. tb_info = traceback.extract_tb(e.__traceback__)
  650. for frame in tb_info:
  651. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  652. except Exception as e:
  653. tb_info = traceback.extract_tb(e.__traceback__)
  654. for frame in tb_info:
  655. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  656. # 如厕频次统计回调
  657. def cb_toileting_frequency(self, result, userdata):
  658. try:
  659. if result:
  660. count = 0
  661. event_list = []
  662. for row in result:
  663. dev_id: str = row.get("dev_id")
  664. event_uuid: str = row.get("uuid")
  665. plan_uuid: str = row.get("plan_uuid")
  666. event_type: int = row.get("event_type")
  667. info: dict = json.loads(row["info"]) if row.get("info") else {}
  668. is_handle: str = row.get("is_handle")
  669. create_time: str = row.get("create_time")
  670. update_time: str = row.get("update_time")
  671. is_deleted: str = row.get("is_deleted")
  672. remark: str = row.get("remark")
  673. event_list.append(info)
  674. this_event_uuid = str(uuid.uuid4())
  675. last_info = {
  676. "start_time" : userdata["start_dt"],
  677. "end_time" : userdata["end_dt"],
  678. "count" : len(event_list),
  679. "event_list" : event_list
  680. }
  681. # 入库
  682. event_uuid = str(uuid.uuid4())
  683. params = {
  684. "dev_id": dev_id,
  685. "uuid": this_event_uuid,
  686. "plan_uuid": self.plan_uuid_,
  687. "event_type": event_desc_map[self.event_type_],
  688. "info": json.dumps(last_info, ensure_ascii=False),
  689. "is_handle": 0,
  690. "create_time": get_bj_time_s(),
  691. "is_deleted": 0,
  692. "remark": json.dumps({}, ensure_ascii=False)
  693. }
  694. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  695. # 通知
  696. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, "events")
  697. LOGINFO(f"EventAttr_ToiletingFrequency succeed")
  698. else:
  699. LOGDBG("EventAttr_ToiletingFrequency, empty result")
  700. except json.JSONDecodeError as e:
  701. tb_info = traceback.extract_tb(e.__traceback__)
  702. for frame in tb_info:
  703. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  704. except Exception as e:
  705. tb_info = traceback.extract_tb(e.__traceback__)
  706. for frame in tb_info:
  707. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  708. return
  709. # 夜间如厕频次统计回调
  710. def cb_night_toileting_frequency(self, result, userdata):
  711. try:
  712. if result:
  713. count = 0
  714. event_list = []
  715. for row in result:
  716. dev_id: str = row.get("dev_id")
  717. event_uuid: str = row.get("uuid")
  718. plan_uuid: str = row.get("plan_uuid")
  719. event_type: int = row.get("event_type")
  720. info: dict = json.loads(row["info"]) if row.get("info") else {}
  721. is_handle: str = row.get("is_handle")
  722. create_time: str = row.get("create_time")
  723. update_time: str = row.get("update_time")
  724. is_deleted: str = row.get("is_deleted")
  725. remark: str = row.get("remark")
  726. event_list.append(info)
  727. this_event_uuid = str(uuid.uuid4())
  728. last_info = {
  729. "start_time" : userdata["start_dt"],
  730. "end_time" : userdata["end_dt"],
  731. "count" : len(event_list),
  732. "event_list" : event_list
  733. }
  734. # 入库
  735. event_uuid = str(uuid.uuid4())
  736. params = {
  737. "dev_id": dev_id,
  738. "uuid": this_event_uuid,
  739. "plan_uuid": self.plan_uuid_,
  740. "event_type": event_desc_map[self.event_type_],
  741. "info": json.dumps(last_info, ensure_ascii=False),
  742. "is_handle": 0,
  743. "create_time": get_bj_time_s(),
  744. "is_deleted": 0,
  745. "remark": json.dumps({}, ensure_ascii=False)
  746. }
  747. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  748. # 通知
  749. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, "events")
  750. LOGINFO(f"EventAttr_ToiletingFrequency succeed")
  751. else:
  752. LOGDBG("EventAttr_ToiletingFrequency, empty result")
  753. except json.JSONDecodeError as e:
  754. tb_info = traceback.extract_tb(e.__traceback__)
  755. for frame in tb_info:
  756. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  757. except Exception as e:
  758. tb_info = traceback.extract_tb(e.__traceback__)
  759. for frame in tb_info:
  760. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  761. return
  762. # 如厕频次异常回调
  763. def cb_toileting_frequency_abnormal(self, result, userdata):
  764. try:
  765. if result:
  766. count = 0
  767. event_list = []
  768. for row in result:
  769. dev_id: str = row.get("dev_id")
  770. event_uuid: str = row.get("uuid")
  771. plan_uuid: str = row.get("plan_uuid")
  772. event_type: int = row.get("event_type")
  773. info: dict = json.loads(row["info"]) if row.get("info") else {}
  774. is_handle: str = row.get("is_handle")
  775. create_time: str = row.get("create_time")
  776. update_time: str = row.get("update_time")
  777. is_deleted: str = row.get("is_deleted")
  778. remark: str = row.get("remark")
  779. event_list.append(info)
  780. if len(event_list) < self.event_attr_.threshold_count_:
  781. return
  782. this_event_uuid = str(uuid.uuid4())
  783. last_info = {
  784. "start_time" : userdata["start_dt"],
  785. "end_time" : userdata["end_dt"],
  786. "count" : len(event_list),
  787. "event_list" : event_list
  788. }
  789. # 入库
  790. event_uuid = str(uuid.uuid4())
  791. params = {
  792. "dev_id": dev_id,
  793. "uuid": this_event_uuid,
  794. "plan_uuid": self.plan_uuid_,
  795. "event_type": event_desc_map[self.event_type_],
  796. "info": json.dumps(last_info, ensure_ascii=False),
  797. "is_handle": 0,
  798. "create_time": get_bj_time_s(),
  799. "is_deleted": 0,
  800. "remark": json.dumps({}, ensure_ascii=False)
  801. }
  802. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  803. # 通知
  804. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], last_info, "events")
  805. LOGINFO(f"EventAttr_ToiletingFrequency succeed")
  806. else:
  807. LOGDBG("EventAttr_ToiletingFrequency, empty result")
  808. except json.JSONDecodeError as e:
  809. tb_info = traceback.extract_tb(e.__traceback__)
  810. for frame in tb_info:
  811. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  812. except Exception as e:
  813. tb_info = traceback.extract_tb(e.__traceback__)
  814. for frame in tb_info:
  815. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  816. return
  817. # 起夜异常回调
  818. def cb_toileting_frequency_abnormal(self, result, userdata):
  819. try:
  820. if result:
  821. count = 0
  822. event_list = []
  823. for row in result:
  824. dev_id: str = row.get("dev_id")
  825. event_uuid: str = row.get("uuid")
  826. plan_uuid: str = row.get("plan_uuid")
  827. event_type: int = row.get("event_type")
  828. info: dict = json.loads(row["info"]) if row.get("info") else {}
  829. is_handle: str = row.get("is_handle")
  830. create_time: str = row.get("create_time")
  831. update_time: str = row.get("update_time")
  832. is_deleted: str = row.get("is_deleted")
  833. remark: str = row.get("remark")
  834. event_list.append(info)
  835. if len(event_list) < self.event_attr_.threshold_count_:
  836. return
  837. this_event_uuid = str(uuid.uuid4())
  838. last_info = {
  839. "start_time" : userdata["start_dt"],
  840. "end_time" : userdata["end_dt"],
  841. "count" : len(event_list),
  842. "event_list" : event_list
  843. }
  844. # 入库
  845. event_uuid = str(uuid.uuid4())
  846. params = {
  847. "dev_id": dev_id,
  848. "uuid": this_event_uuid,
  849. "plan_uuid": self.plan_uuid_,
  850. "event_type": event_desc_map[self.event_type_],
  851. "info": json.dumps(last_info, ensure_ascii=False),
  852. "is_handle": 0,
  853. "create_time": get_bj_time_s(),
  854. "is_deleted": 0,
  855. "remark": json.dumps({}, ensure_ascii=False)
  856. }
  857. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  858. # 通知
  859. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], last_info, "events")
  860. LOGINFO(f"EventAttr_ToiletingFrequency succeed")
  861. else:
  862. LOGDBG("EventAttr_ToiletingFrequency, empty result")
  863. except json.JSONDecodeError as e:
  864. tb_info = traceback.extract_tb(e.__traceback__)
  865. for frame in tb_info:
  866. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  867. except Exception as e:
  868. tb_info = traceback.extract_tb(e.__traceback__)
  869. for frame in tb_info:
  870. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  871. return
  872. # 如厕频次统计回调
  873. def cb_bathroom_stay_frequency(self, result, userdata):
  874. try:
  875. if result:
  876. count = 0
  877. event_list = []
  878. for row in result:
  879. dev_id: str = row.get("dev_id")
  880. event_uuid: str = row.get("uuid")
  881. plan_uuid: str = row.get("plan_uuid")
  882. event_type: int = row.get("event_type")
  883. info: dict = json.loads(row["info"]) if row.get("info") else {}
  884. is_handle: str = row.get("is_handle")
  885. create_time: str = row.get("create_time")
  886. update_time: str = row.get("update_time")
  887. is_deleted: str = row.get("is_deleted")
  888. remark: str = row.get("remark")
  889. event_list.append(info)
  890. this_event_uuid = str(uuid.uuid4())
  891. last_info = {
  892. "start_time" : userdata["start_dt"],
  893. "end_time" : userdata["end_dt"],
  894. "count" : len(event_list),
  895. "event_list" : event_list
  896. }
  897. # 入库
  898. event_uuid = str(uuid.uuid4())
  899. params = {
  900. "dev_id": dev_id,
  901. "uuid": this_event_uuid,
  902. "plan_uuid": self.plan_uuid_,
  903. "event_type": event_desc_map[self.event_type_],
  904. "info": json.dumps(last_info, ensure_ascii=False),
  905. "is_handle": 0,
  906. "create_time": get_bj_time_s(),
  907. "is_deleted": 0,
  908. "remark": json.dumps({}, ensure_ascii=False)
  909. }
  910. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  911. # 通知
  912. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, "events")
  913. LOGINFO(f"EventAttr_ToiletingFrequency succeed")
  914. else:
  915. LOGDBG("EventAttr_ToiletingFrequency, empty result")
  916. except json.JSONDecodeError as e:
  917. tb_info = traceback.extract_tb(e.__traceback__)
  918. for frame in tb_info:
  919. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  920. except Exception as e:
  921. tb_info = traceback.extract_tb(e.__traceback__)
  922. for frame in tb_info:
  923. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  924. return