alarm_plan.py 45 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152
  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 core.linkage_action import LinkageAction
  19. from db.db_process import (
  20. db_req_que, DBRequest
  21. )
  22. import db.db_sqls as sqls
  23. import mqtt.mqtt_send as mqtt_send
  24. import device.dev_mng as g_Dev
  25. from device.dev_mng import (
  26. Device, g_dev_mgr
  27. )
  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. linkage_action: LinkageAction = LinkageAction(),
  164. tenant_id:int = 0
  165. ):
  166. self.lock_ = Lock()
  167. self.plan_uuid_ = plan_uuid # 计划id
  168. self.name_ = name # 计划名称
  169. self.dev_id_ = dev_id # 设备id
  170. self.dev_name_ = dev_name # 设备名称
  171. self.enable_ = enable # 是否启用
  172. self.time_plan_ = time_plan # 时间计划
  173. self.param_ = param # 参数
  174. self.linkage_action_ = linkage_action # 联动动作
  175. self.tenant_id_ = tenant_id # 租户id
  176. # 维护状态(根据TimePlanu判断)
  177. self.status_ = 0 # 0未激活,1激活,-1过期
  178. self.status_update_ts_ = -1 # 状态更新时间,初始值为-1
  179. # 事件触发参数
  180. self.rect_ = rect # 检测区域 [left, top, width, height]
  181. self.threshold_time_ = threshold_time # 触发时间阈值
  182. self.merge_time_ = merge_time # 归并时间窗口
  183. self.event_type_ = event_type # 事件类型
  184. self.event_attr_ = self.init_event_attr() # 事件属性
  185. if self.event_attr_ is None:
  186. raise ValueError(f"Invalid event_type: {event_type}")
  187. # 计划任务的开始时间
  188. self.cron_ = cron # {“hour": 7, "minute": 0}
  189. def execute(self):
  190. if self.status_ != 1:
  191. return
  192. g_las.g_alarm_plan_disp.dispatch(self.event_type_, self)
  193. # 初始化事件属性
  194. def init_event_attr(self):
  195. try:
  196. event_cls = event_attr_map.get(self.event_type_)
  197. if event_cls is None:
  198. return None
  199. event_attr = event_cls(self.event_type_)
  200. if ((self.event_type_ == EventType.TOILETING_FREQUENCY_ABNORMAL.value) or
  201. (self.event_type_ == EventType.NIGHT_TOILETING_FREQUENCY_ABNORMAL.value)):
  202. event_attr.threshold_count_ = int(self.param_.get("count", 0))
  203. if ((self.event_type_ == EventType.TARGET_ABSENCE.value)):
  204. event_attr.time_threshold_ = int(self.param_.get("time_threshold", 0))
  205. return event_attr
  206. except json.JSONDecodeError as e:
  207. tb_info = traceback.extract_tb(e.__traceback__)
  208. for frame in tb_info:
  209. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  210. except Exception as e:
  211. tb_info = traceback.extract_tb(e.__traceback__)
  212. for frame in tb_info:
  213. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  214. # 更新激活状态
  215. def update_status(self, now: Optional[datetime] = None) -> None:
  216. now = now or datetime.now()
  217. old_status = self.status_
  218. if not self.enable_:
  219. self.status_ = 0
  220. else:
  221. now_fmt = now.strftime("%Y-%m-%d")
  222. # 过期
  223. if now_fmt > self.time_plan_.stop_date_:
  224. self.status_ = -1
  225. elif now_fmt < self.time_plan_.start_date_:
  226. self.status_ = 0
  227. elif self.time_plan_.is_active_now(now):
  228. self.status_ = 1
  229. else:
  230. self.status_ = 0
  231. if self.status_ != old_status:
  232. self.status_update_ts = int(now.timestamp())
  233. LOGINFO(f"[Status] plan {self.plan_uuid_} status_ changed {old_status} -> {self.status_}")
  234. def is_point_in_rect(self, x:float, y:float, rect:list) -> bool:
  235. rx, ry, rw, rh = rect
  236. x_min = min(rx, rx + rw)
  237. x_max = max(rx, rx + rw)
  238. y_min = min(ry, ry - rh)
  239. y_max = max(ry, ry - rh)
  240. bRet: bool = x_min <= x <= x_max and y_min <= y <= y_max
  241. return bRet
  242. # 查找最近 t 秒内,最后一个落在 rect 内的 target_point 的 rtd_unit
  243. def find_latest_rtd_in_region(self, device: Device, rect: list, now: int=None, t: int=1):
  244. now_s = now if now else get_utc_time_s()
  245. rtd_que_copy = device.get_rtd_que_copy()
  246. with self.lock_:
  247. for rtd_unit in reversed(rtd_que_copy): # 倒序扫描
  248. ts_s = int(rtd_unit.get("timestamp", 0))
  249. if now_s - ts_s > t:
  250. break # 已经超过 t 秒,可以直接结束
  251. # 检查点是否在区域内
  252. for pt in rtd_unit.get("target_point", []):
  253. if len(pt) >= 2:
  254. x, y = pt[0], pt[1]
  255. if self.is_point_in_rect(x, y, rect):
  256. return rtd_unit
  257. return None
  258. # 停留事件
  259. def handle_stay_detection(self):
  260. try:
  261. dev_id = self.dev_id_
  262. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  263. if not device:
  264. return
  265. now = get_utc_time_s()
  266. # 查找最新的落在检测区域的目标
  267. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  268. if rtd_unit:
  269. timestamp = rtd_unit["timestamp"]
  270. pose = rtd_unit["pose"]
  271. target_point = rtd_unit["target_point"]
  272. if self.event_attr_.enter_ts_ == -1:
  273. self.event_attr_.enter_ts_ = timestamp
  274. else:
  275. self.event_attr_.leave_ts_ = timestamp
  276. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  277. return
  278. # 归并时间内,不认为事件结束
  279. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  280. return
  281. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  282. stay_time =self.event_attr_.stay_time_
  283. # 时间小于触发时间阈值,忽略并重置
  284. if stay_time < self.threshold_time_ :
  285. self.event_attr_.reset()
  286. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  287. return
  288. # 构造事件
  289. # 入库
  290. info = {
  291. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  292. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  293. "stay_time": stay_time
  294. }
  295. event_uuid = str(uuid.uuid4())
  296. params = {
  297. "dev_id": dev_id,
  298. "uuid": event_uuid,
  299. "plan_uuid": self.plan_uuid_,
  300. "event_type": self.event_type_,
  301. "info": json.dumps(info),
  302. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  303. "is_handle": 0,
  304. "create_time": get_bj_time_s(),
  305. "is_deleted": 0,
  306. "tenant_id": self.tenant_id_,
  307. "remark": json.dumps({}, ensure_ascii=False)
  308. }
  309. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  310. # 通知
  311. linkage_action = {
  312. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  313. }
  314. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, linkage_action, "events")
  315. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  316. self.event_attr_.reset()
  317. except json.JSONDecodeError as e:
  318. tb_info = traceback.extract_tb(e.__traceback__)
  319. for frame in tb_info:
  320. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  321. except Exception as e:
  322. tb_info = traceback.extract_tb(e.__traceback__)
  323. for frame in tb_info:
  324. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  325. # 滞留事件
  326. def handle_retention_detection(self):
  327. try:
  328. dev_id = self.dev_id_
  329. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  330. if not device:
  331. return
  332. now = get_utc_time_s()
  333. # 查找最新的落在检测区域的目标
  334. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  335. if rtd_unit:
  336. timestamp = rtd_unit["timestamp"]
  337. pose = rtd_unit["pose"]
  338. target_point = rtd_unit["target_point"]
  339. if self.event_attr_.enter_ts_ == -1:
  340. self.event_attr_.enter_ts_ = timestamp
  341. else:
  342. self.event_attr_.leave_ts_ = timestamp
  343. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  344. return
  345. # 归并时间内,不认为事件结束
  346. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  347. return
  348. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  349. stay_time =self.event_attr_.stay_time_
  350. # 时间小于触发时间阈值,忽略并重置
  351. if stay_time < self.threshold_time_ :
  352. self.event_attr_.reset()
  353. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  354. return
  355. # 构造事件
  356. # 入库
  357. info = {
  358. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  359. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  360. "stay_time": stay_time
  361. }
  362. event_uuid = str(uuid.uuid4())
  363. params = {
  364. "dev_id": dev_id,
  365. "uuid": event_uuid,
  366. "plan_uuid": self.plan_uuid_,
  367. "event_type": self.event_type_,
  368. "info": json.dumps(info),
  369. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  370. "is_handle": 0,
  371. "create_time": get_bj_time_s(),
  372. "is_deleted": 0,
  373. "tenant_id": self.tenant_id_,
  374. "remark": json.dumps({}, ensure_ascii=False)
  375. }
  376. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  377. # 通知
  378. linkage_action = {
  379. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  380. }
  381. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, linkage_action, "events")
  382. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  383. self.event_attr_.reset()
  384. except json.JSONDecodeError as e:
  385. tb_info = traceback.extract_tb(e.__traceback__)
  386. for frame in tb_info:
  387. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  388. except Exception as e:
  389. tb_info = traceback.extract_tb(e.__traceback__)
  390. for frame in tb_info:
  391. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  392. # 如厕事件
  393. def handle_toileting_detection(self):
  394. try:
  395. dev_id = self.dev_id_
  396. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  397. if not device:
  398. return
  399. now = get_utc_time_s()
  400. # 查找最新的落在检测区域的目标
  401. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  402. if rtd_unit:
  403. timestamp = rtd_unit["timestamp"]
  404. pose = rtd_unit["pose"]
  405. target_point = rtd_unit["target_point"]
  406. if self.event_attr_.enter_ts_ == -1:
  407. self.event_attr_.enter_ts_ = timestamp
  408. else:
  409. self.event_attr_.leave_ts_ = timestamp
  410. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  411. return
  412. # 归并时间内,不认为事件结束
  413. if now - self.event_attr_.leave_ts_ < self.merge_time_:
  414. return
  415. self.event_attr_.stay_time_ = self.event_attr_.leave_ts_ - self.event_attr_.enter_ts_
  416. stay_time =self.event_attr_.stay_time_
  417. # 时间小于触发时间阈值,忽略并重置
  418. if stay_time < self.threshold_time_ :
  419. self.event_attr_.reset()
  420. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  421. return
  422. # 构造事件
  423. # 入库
  424. info = {
  425. "start_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  426. "end_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  427. "stay_time": stay_time
  428. }
  429. event_uuid = str(uuid.uuid4())
  430. params = {
  431. "dev_id": dev_id,
  432. "uuid": event_uuid,
  433. "plan_uuid": self.plan_uuid_,
  434. "event_type": self.event_type_,
  435. "info": json.dumps(info),
  436. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  437. "is_handle": 0,
  438. "create_time": get_bj_time_s(),
  439. "is_deleted": 0,
  440. "tenant_id": self.tenant_id_,
  441. "remark": json.dumps({}, ensure_ascii=False)
  442. }
  443. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  444. # 通知
  445. linkage_action = {
  446. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  447. }
  448. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, linkage_action, "events")
  449. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, stay_time: {stay_time}")
  450. self.event_attr_.reset()
  451. except json.JSONDecodeError as e:
  452. tb_info = traceback.extract_tb(e.__traceback__)
  453. for frame in tb_info:
  454. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  455. except Exception as e:
  456. tb_info = traceback.extract_tb(e.__traceback__)
  457. for frame in tb_info:
  458. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  459. # 如厕频次统计
  460. def handle_toileting_frequency(self):
  461. try:
  462. dev_id = self.dev_id_
  463. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  464. if not device:
  465. return
  466. start_dt, end_dt = helper.get_query_time_range(self.param_)
  467. params = {
  468. "dev_id" : self.dev_id_,
  469. "event_type": EventType.TOILETING_DETECTION.value,
  470. "start_dt": start_dt,
  471. "end_dt": end_dt
  472. }
  473. userdata = {
  474. "start_dt" : start_dt,
  475. "end_dt" : end_dt
  476. }
  477. db_req_que.put(DBRequest(
  478. sql=sqls.sql_query_events_by_datetime,
  479. params=params,
  480. callback=self.cb_toileting_frequency,
  481. userdata=userdata))
  482. except json.JSONDecodeError as e:
  483. tb_info = traceback.extract_tb(e.__traceback__)
  484. for frame in tb_info:
  485. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  486. except Exception as e:
  487. tb_info = traceback.extract_tb(e.__traceback__)
  488. for frame in tb_info:
  489. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  490. # 夜间如厕频次统计
  491. def handle_night_toileting_frequency(self):
  492. try:
  493. dev_id = self.dev_id_
  494. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  495. if not device:
  496. return
  497. start_dt, end_dt = helper.get_query_time_range(self.param_)
  498. params = {
  499. "dev_id" : self.dev_id_,
  500. "event_type": EventType.TOILETING_DETECTION.value,
  501. "start_dt": start_dt,
  502. "end_dt": end_dt
  503. }
  504. userdata = {
  505. "start_dt" : start_dt,
  506. "end_dt" : end_dt
  507. }
  508. db_req_que.put(DBRequest(
  509. sql=sqls.sql_query_events_by_datetime,
  510. params=params,
  511. callback=self.cb_night_toileting_frequency,
  512. userdata=userdata))
  513. except json.JSONDecodeError as e:
  514. tb_info = traceback.extract_tb(e.__traceback__)
  515. for frame in tb_info:
  516. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  517. except Exception as e:
  518. tb_info = traceback.extract_tb(e.__traceback__)
  519. for frame in tb_info:
  520. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  521. # 如厕频次异常
  522. def handle_toileting_frequency_abnormal(self):
  523. try:
  524. dev_id = self.dev_id_
  525. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  526. if not device:
  527. return
  528. start_dt, end_dt = helper.get_query_time_range(self.param_)
  529. params = {
  530. "dev_id" : self.dev_id_,
  531. "event_type": EventType.TOILETING_DETECTION.value,
  532. "start_dt": start_dt,
  533. "end_dt": end_dt
  534. }
  535. userdata = {
  536. "start_dt" : start_dt,
  537. "end_dt" : end_dt
  538. }
  539. db_req_que.put(DBRequest(
  540. sql=sqls.sql_query_events_by_datetime,
  541. params=params,
  542. callback=self.cb_toileting_frequency_abnormal,
  543. userdata=userdata))
  544. except json.JSONDecodeError as e:
  545. tb_info = traceback.extract_tb(e.__traceback__)
  546. for frame in tb_info:
  547. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  548. except Exception as e:
  549. tb_info = traceback.extract_tb(e.__traceback__)
  550. for frame in tb_info:
  551. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  552. # 起夜异常
  553. def handle_night_toileting_frequency_abnormal(self):
  554. try:
  555. dev_id = self.dev_id_
  556. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  557. if not device:
  558. return
  559. start_dt, end_dt = helper.get_query_time_range(self.param_)
  560. params = {
  561. "dev_id" : self.dev_id_,
  562. "event_type": EventType.TOILETING_DETECTION.value,
  563. "start_dt": start_dt,
  564. "end_dt": end_dt
  565. }
  566. userdata = {
  567. "start_dt" : start_dt,
  568. "end_dt" : end_dt
  569. }
  570. db_req_que.put(DBRequest(
  571. sql=sqls.sql_query_events_by_datetime,
  572. params=params,
  573. callback=self.cb_night_toileting_frequency_abnormal,
  574. userdata=userdata))
  575. except json.JSONDecodeError as e:
  576. tb_info = traceback.extract_tb(e.__traceback__)
  577. for frame in tb_info:
  578. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  579. except Exception as e:
  580. tb_info = traceback.extract_tb(e.__traceback__)
  581. for frame in tb_info:
  582. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  583. # 卫生间频次统计
  584. def handle_bathroom_stay_frequency(self):
  585. try:
  586. dev_id = self.dev_id_
  587. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  588. if not device:
  589. return
  590. start_dt, end_dt = helper.get_query_time_range(self.param_)
  591. params = {
  592. "dev_id" : self.dev_id_,
  593. "event_type": EventType.STAY_DETECTION.value,
  594. "start_dt": start_dt,
  595. "end_dt": end_dt
  596. }
  597. userdata = {
  598. "start_dt" : start_dt,
  599. "end_dt" : end_dt
  600. }
  601. db_req_que.put(DBRequest(
  602. sql=sqls.sql_query_events_by_datetime,
  603. params=params,
  604. callback=self.cb_bathroom_stay_frequency,
  605. userdata=userdata))
  606. except json.JSONDecodeError as e:
  607. tb_info = traceback.extract_tb(e.__traceback__)
  608. for frame in tb_info:
  609. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  610. except Exception as e:
  611. tb_info = traceback.extract_tb(e.__traceback__)
  612. for frame in tb_info:
  613. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  614. # 异常消失
  615. def handle_target_absence(self):
  616. try:
  617. dev_id = self.dev_id_
  618. device:Device = g_Dev.g_dev_mgr.find_dev_map(dev_id)
  619. if not device:
  620. return
  621. now = get_utc_time_s()
  622. threshold: int = self.event_attr_.time_threshold_
  623. # 查找最新的落在检测区域的目标
  624. rtd_unit = self.find_latest_rtd_in_region(device, self.rect_, now, 3)
  625. if rtd_unit:
  626. return
  627. timestamp = get_utc_time_s()
  628. if self.event_attr_.leave_ts_ == -1:
  629. self.event_attr_.leave_ts_ = timestamp
  630. else:
  631. self.event_attr_.enter_ts_ = timestamp
  632. if self.event_attr_.enter_ts_ == -1 or self.event_attr_.leave_ts_ == -1:
  633. return
  634. # 归并时间内,不认为事件结束
  635. if now - self.event_attr_.enter_ts_ < self.merge_time_:
  636. return
  637. self.event_attr_.absence_time_ = self.event_attr_.enter_ts_ - self.event_attr_.leave_ts_
  638. absence_time = self.event_attr_.absence_time_
  639. # 时间小于触发时间阈值,忽略并重置
  640. if absence_time < threshold :
  641. self.event_attr_.reset()
  642. LOGINFO(f"less than threshold_time, alarm_plan: {self.plan_uuid_}, event_type: {self.event_type_}")
  643. return
  644. # 构造事件
  645. # 入库
  646. info = {
  647. "start_time": utc_to_bj_s(self.event_attr_.leave_ts_),
  648. "end_time": utc_to_bj_s(self.event_attr_.enter_ts_),
  649. "absence_time": absence_time
  650. }
  651. event_uuid = str(uuid.uuid4())
  652. params = {
  653. "dev_id": dev_id,
  654. "uuid": event_uuid,
  655. "plan_uuid": self.plan_uuid_,
  656. "event_type": self.event_type_,
  657. "info": json.dumps(info),
  658. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  659. "is_handle": 0,
  660. "create_time": get_bj_time_s(),
  661. "is_deleted": 0,
  662. "tenant_id": self.tenant_id_,
  663. "remark": json.dumps({}, ensure_ascii=False)
  664. }
  665. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  666. # 通知
  667. linkage_action = {
  668. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  669. }
  670. mqtt_send.alarm_event(dev_id, self.dev_name_, event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], info, linkage_action, "events")
  671. LOGDBG(f"new event: {event_desc_map[self.event_type_]}, absence_time: {absence_time}")
  672. self.event_attr_.reset()
  673. except json.JSONDecodeError as e:
  674. tb_info = traceback.extract_tb(e.__traceback__)
  675. for frame in tb_info:
  676. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  677. except Exception as e:
  678. tb_info = traceback.extract_tb(e.__traceback__)
  679. for frame in tb_info:
  680. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  681. # 如厕频次统计回调
  682. def cb_toileting_frequency(self, result, userdata):
  683. try:
  684. if result:
  685. count = 0
  686. event_list = []
  687. for row in result:
  688. dev_id: str = row.get("dev_id")
  689. event_uuid: str = row.get("uuid")
  690. plan_uuid: str = row.get("plan_uuid")
  691. event_type: int = row.get("event_type")
  692. info: dict = json.loads(row["info"]) if row.get("info") else {}
  693. is_handle: str = row.get("is_handle")
  694. create_time: str = row.get("create_time")
  695. update_time: str = row.get("update_time")
  696. is_deleted: str = row.get("is_deleted")
  697. remark: str = row.get("remark")
  698. event_list.append(info)
  699. this_event_uuid = str(uuid.uuid4())
  700. last_info = {
  701. "start_time" : userdata["start_dt"],
  702. "end_time" : userdata["end_dt"],
  703. "count" : len(event_list),
  704. "event_list" : event_list
  705. }
  706. # 入库
  707. event_uuid = str(uuid.uuid4())
  708. params = {
  709. "dev_id": dev_id,
  710. "uuid": this_event_uuid,
  711. "plan_uuid": self.plan_uuid_,
  712. "event_type": self.event_type_,
  713. "info": json.dumps(last_info, ensure_ascii=False),
  714. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  715. "is_handle": 0,
  716. "create_time": get_bj_time_s(),
  717. "is_deleted": 0,
  718. "tenant_id": self.tenant_id_,
  719. "remark": json.dumps({}, ensure_ascii=False)
  720. }
  721. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  722. # 通知
  723. linkage_action = {
  724. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  725. }
  726. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, linkage_action, "events")
  727. LOGINFO(f"cb_toileting_frequency succeed")
  728. else:
  729. LOGDBG("cb_toileting_frequency, empty result")
  730. except json.JSONDecodeError as e:
  731. tb_info = traceback.extract_tb(e.__traceback__)
  732. for frame in tb_info:
  733. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  734. except Exception as e:
  735. tb_info = traceback.extract_tb(e.__traceback__)
  736. for frame in tb_info:
  737. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  738. return
  739. # 夜间如厕频次统计回调
  740. def cb_night_toileting_frequency(self, result, userdata):
  741. try:
  742. if result:
  743. count = 0
  744. event_list = []
  745. for row in result:
  746. dev_id: str = row.get("dev_id")
  747. event_uuid: str = row.get("uuid")
  748. plan_uuid: str = row.get("plan_uuid")
  749. event_type: int = row.get("event_type")
  750. info: dict = json.loads(row["info"]) if row.get("info") else {}
  751. is_handle: str = row.get("is_handle")
  752. create_time: str = row.get("create_time")
  753. update_time: str = row.get("update_time")
  754. is_deleted: str = row.get("is_deleted")
  755. remark: str = row.get("remark")
  756. event_list.append(info)
  757. this_event_uuid = str(uuid.uuid4())
  758. last_info = {
  759. "start_time" : userdata["start_dt"],
  760. "end_time" : userdata["end_dt"],
  761. "count" : len(event_list),
  762. "event_list" : event_list
  763. }
  764. # 入库
  765. event_uuid = str(uuid.uuid4())
  766. params = {
  767. "dev_id": dev_id,
  768. "uuid": this_event_uuid,
  769. "plan_uuid": self.plan_uuid_,
  770. "event_type": self.event_type_,
  771. "info": json.dumps(last_info, ensure_ascii=False),
  772. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  773. "is_handle": 0,
  774. "create_time": get_bj_time_s(),
  775. "is_deleted": 0,
  776. "tenant_id": self.tenant_id_,
  777. "remark": json.dumps({}, ensure_ascii=False)
  778. }
  779. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  780. # 通知
  781. linkage_action = {
  782. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  783. }
  784. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, linkage_action, "events")
  785. LOGINFO(f"cb_night_toileting_frequency succeed")
  786. else:
  787. LOGDBG("cb_night_toileting_frequency, empty result")
  788. except json.JSONDecodeError as e:
  789. tb_info = traceback.extract_tb(e.__traceback__)
  790. for frame in tb_info:
  791. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  792. except Exception as e:
  793. tb_info = traceback.extract_tb(e.__traceback__)
  794. for frame in tb_info:
  795. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  796. return
  797. # 如厕频次异常回调
  798. def cb_toileting_frequency_abnormal(self, result, userdata):
  799. try:
  800. if result:
  801. count = 0
  802. event_list = []
  803. for row in result:
  804. dev_id: str = row.get("dev_id")
  805. event_uuid: str = row.get("uuid")
  806. plan_uuid: str = row.get("plan_uuid")
  807. event_type: int = row.get("event_type")
  808. info: dict = json.loads(row["info"]) if row.get("info") else {}
  809. is_handle: str = row.get("is_handle")
  810. create_time: str = row.get("create_time")
  811. update_time: str = row.get("update_time")
  812. is_deleted: str = row.get("is_deleted")
  813. remark: str = row.get("remark")
  814. event_list.append(info)
  815. if len(event_list) < self.event_attr_.threshold_count_:
  816. return
  817. this_event_uuid = str(uuid.uuid4())
  818. last_info = {
  819. "start_time" : userdata["start_dt"],
  820. "end_time" : userdata["end_dt"],
  821. "count" : len(event_list),
  822. "event_list" : event_list
  823. }
  824. # 入库
  825. event_uuid = str(uuid.uuid4())
  826. params = {
  827. "dev_id": dev_id,
  828. "uuid": this_event_uuid,
  829. "plan_uuid": self.plan_uuid_,
  830. "event_type": self.event_type_,
  831. "info": json.dumps(last_info, ensure_ascii=False),
  832. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  833. "is_handle": 0,
  834. "create_time": get_bj_time_s(),
  835. "is_deleted": 0,
  836. "tenant_id": self.tenant_id_,
  837. "remark": json.dumps({}, ensure_ascii=False)
  838. }
  839. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  840. # 通知
  841. linkage_action = {
  842. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  843. }
  844. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], last_info, linkage_action, "events")
  845. LOGINFO(f"cb_toileting_frequency_abnormal succeed")
  846. else:
  847. LOGDBG("cb_toileting_frequency_abnormal, empty result")
  848. except json.JSONDecodeError as e:
  849. tb_info = traceback.extract_tb(e.__traceback__)
  850. for frame in tb_info:
  851. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  852. except Exception as e:
  853. tb_info = traceback.extract_tb(e.__traceback__)
  854. for frame in tb_info:
  855. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  856. return
  857. # 起夜异常回调
  858. def cb_night_toileting_frequency_abnormal(self, result, userdata):
  859. try:
  860. if result:
  861. count = 0
  862. event_list = []
  863. for row in result:
  864. dev_id: str = row.get("dev_id")
  865. event_uuid: str = row.get("uuid")
  866. plan_uuid: str = row.get("plan_uuid")
  867. event_type: int = row.get("event_type")
  868. info: dict = json.loads(row["info"]) if row.get("info") else {}
  869. is_handle: str = row.get("is_handle")
  870. create_time: str = row.get("create_time")
  871. update_time: str = row.get("update_time")
  872. is_deleted: str = row.get("is_deleted")
  873. remark: str = row.get("remark")
  874. event_list.append(info)
  875. if len(event_list) < self.event_attr_.threshold_count_:
  876. return
  877. this_event_uuid = str(uuid.uuid4())
  878. last_info = {
  879. "start_time" : userdata["start_dt"],
  880. "end_time" : userdata["end_dt"],
  881. "count" : len(event_list),
  882. "event_list" : event_list
  883. }
  884. # 入库
  885. event_uuid = str(uuid.uuid4())
  886. params = {
  887. "dev_id": dev_id,
  888. "uuid": this_event_uuid,
  889. "plan_uuid": self.plan_uuid_,
  890. "event_type": self.event_type_,
  891. "info": json.dumps(last_info, ensure_ascii=False),
  892. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  893. "is_handle": 0,
  894. "create_time": get_bj_time_s(),
  895. "is_deleted": 0,
  896. "tenant_id": self.tenant_id_,
  897. "remark": json.dumps({}, ensure_ascii=False)
  898. }
  899. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  900. # 通知
  901. linkage_action = {
  902. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  903. }
  904. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, self.plan_uuid_, event_desc_map[self.event_type_], last_info, linkage_action, "events")
  905. LOGINFO(f"cb_night_toileting_frequency_abnormal succeed")
  906. else:
  907. LOGDBG("cb_night_toileting_frequency_abnormal, empty result")
  908. except json.JSONDecodeError as e:
  909. tb_info = traceback.extract_tb(e.__traceback__)
  910. for frame in tb_info:
  911. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  912. except Exception as e:
  913. tb_info = traceback.extract_tb(e.__traceback__)
  914. for frame in tb_info:
  915. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  916. return
  917. # 卫生间频次统计回调
  918. def cb_bathroom_stay_frequency(self, result, userdata):
  919. try:
  920. if result:
  921. count = 0
  922. event_list = []
  923. for row in result:
  924. dev_id: str = row.get("dev_id")
  925. event_uuid: str = row.get("uuid")
  926. plan_uuid: str = row.get("plan_uuid")
  927. event_type: int = row.get("event_type")
  928. info: dict = json.loads(row["info"]) if row.get("info") else {}
  929. is_handle: str = row.get("is_handle")
  930. create_time: str = row.get("create_time")
  931. update_time: str = row.get("update_time")
  932. is_deleted: str = row.get("is_deleted")
  933. remark: str = row.get("remark")
  934. event_list.append(info)
  935. this_event_uuid = str(uuid.uuid4())
  936. last_info = {
  937. "start_time" : userdata["start_dt"],
  938. "end_time" : userdata["end_dt"],
  939. "count" : len(event_list),
  940. "event_list" : event_list
  941. }
  942. # 入库
  943. event_uuid = str(uuid.uuid4())
  944. params = {
  945. "dev_id": dev_id,
  946. "uuid": this_event_uuid,
  947. "plan_uuid": self.plan_uuid_,
  948. "event_type": self.event_type_,
  949. "info": json.dumps(last_info, ensure_ascii=False),
  950. "linkage_push_wechat_service": self.linkage_action_.wechat_service_,
  951. "is_handle": 0,
  952. "create_time": get_bj_time_s(),
  953. "is_deleted": 0,
  954. "tenant_id": self.tenant_id_,
  955. "remark": json.dumps({}, ensure_ascii=False)
  956. }
  957. db_req_que.put(DBRequest(sql=sqls.sql_insert_events, params=params, callback=None))
  958. # 通知
  959. linkage_action = {
  960. "linkage_push_wechat_service": self.linkage_action_.wechat_service_
  961. }
  962. mqtt_send.alarm_event(dev_id, self.dev_name_, this_event_uuid, plan_uuid, event_desc_map[self.event_type_], last_info, linkage_action, "events")
  963. LOGINFO(f"cb_bathroom_stay_frequency+- succeed")
  964. else:
  965. LOGDBG("cb_bathroom_stay_frequency, empty result")
  966. except json.JSONDecodeError as e:
  967. tb_info = traceback.extract_tb(e.__traceback__)
  968. for frame in tb_info:
  969. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error:{e}, {e.doc}")
  970. except Exception as e:
  971. tb_info = traceback.extract_tb(e.__traceback__)
  972. for frame in tb_info:
  973. LOGERR(f"[{frame.filename}:{frame.lineno}] @{frame.name}(), error: {e}")
  974. return