cxc/hik-push
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update version

Browse Source
This commit is contained in:
quantulr
2024-03-21 16:05:37 +08:00
parent ba00ccf642
commit b1549787df
8 changed files with 1 additions and 253 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
hik_push/read_event.py
View File

@ -1,209 +0,0 @@
import asyncio
import json
from pathlib import Path
import logging
import yaml
import requests
from redis import asyncio as aioredis
async def read_event():
logging.basicConfig(level=logging.INFO)
# 读取配置文件
home_dir = Path.home()
with open(
home_dir / ".config" / "hik-push" / "config.yaml", "r", encoding="utf-8"
) as f:
config = yaml.safe_load(f)
if "user_ids" not in config:
logging.error("请配置user_ids")
return
if "devices" not in config:
logging.error("请配置devices")
return
if "event_type" not in config:
logging.error("请配置event_type")
return
if "push_url" not in config:
logging.error("请配置push_url")
return
if "event_level" not in config:
logging.error("请配置event_level")
return
logging.info(config)
user_map = config["user_ids"]
device_map = config["devices"]
event_map = config["event_type"]
push_url = config["push_url"]
event_level = config["event_level"]
redis_client = await aioredis.Redis(
host="127.0.0.1", port=7019, password="SMHdFrlK"
)
while True:
try:
data = await redis_client.brpop("hik-sub-event")
sub_json = json.loads(data[1].decode("utf-8"))
events = sub_json["params"]["events"]
for event in events:
# 如果是ai事件
if event["srcType"] == "eventRule":
event_details = event["eventDetails"]
for event_detail in event_details:
# 设置默认事件等级
handle_event_level(
event_detail, event_level, event_detail["eventType"]
)
# 事件类型 id 替换为 str
event_type_str = event_map.get(
event_detail["eventType"], "未知事件类型"
)
event_detail["eventType"] = event_type_str
# 添加设备名称
src_index = event_detail["srcIndex"]
device_name = device_map.get(src_index, "未知设备")
event_detail["deviceName"] = device_name
# 添加regionName
if "srcName" in event_detail:
event_detail["regionName"] = event_detail["srcName"]
else:
event_detail["regionName"] = device_name
# 根据设备名称获取 user_ids
user_ids = []
# 高级
if event_detail["eventLvl"] == 3:
user_all = user_map.get("all", {}).get("high", [])
# 接受全部通知的用户
user_ids = user_map.get(device_name, {}).get("high", [])
user_ids = user_ids + user_all
# 中级
else:
user_all = user_map.get("all", {}).get("all", [])
# 接受全部通知的用户
user_ids = user_map.get(device_name, {}).get("all", [])
user_ids = user_ids + user_all
event_detail["userIds"] = user_ids
# 添加其他字段
event_detail["eventLvl"] = event["eventLvl"]
event_detail["happenTime"] = event["happenTime"]
# 如果存在 data 属性
if "data" in event_detail:
# 替换分析结果字段
if "eventType" in event_detail["data"]:
detection_field_name = event_detail["data"]["eventType"]
if detection_field_name in event_detail["data"]:
event_detail["data"]["_detectionResult"] = (
event_detail["data"].pop(detection_field_name)
)
event_detail["data"]["_detectionResult"]["typeName"] = event_type_str
if isinstance(
event_detail["data"]["_detectionResult"], list
) and len(event_detail["data"]["_detectionResult"]):
event_detail["data"]["_detectionResult"] = (
event_detail["data"]["_detectionResult"][0]
)
replace_image_host(
event_detail["data"]["_detectionResult"]
)
logging.info(event_detail)
try:
push_resp = requests.post(
push_url, json=event_detail
).content.decode("utf-8")
logging.info(push_resp)
except Exception as e:
logging.error(f"网络错误推送失败: {e}")
else:
# 设置默认事件等级
handle_event_level(event, event_level, event["eventType"])
# 将 eventType ID 替换为中文字符串
event_type = event["eventType"]
logging.info(event_type)
event_type_str = event_map.get(event_type, "未知事件类型")
event["eventType"] = event_type_str
# 添加设备名称
src_index = event["srcIndex"]
device_name = device_map.get(src_index, "未知设备")
# 添加regionName
if "srcName" in event:
event["regionName"] = event["srcName"]
else:
event["regionName"] = device_name
# 默认事件等级为2
if "eventLvl" not in event:
event["eventLvl"] = 2
# 根据设备名称获取 user_ids
user_ids = []
# 高级
if event["eventLvl"] == 3:
user_all = user_map.get("all", {}).get("high", [])
# 接受全部通知的用户
user_ids = user_map.get(device_name, {}).get("high", [])
user_ids = user_ids + user_all
# 中级
else:
user_all = user_map.get("all", {}).get("all", [])
# 接受全部通知的用户
user_ids = user_map.get(device_name, {}).get("all", [])
user_ids = user_ids + user_all
event["deviceName"] = device_name
event["userIds"] = user_ids
if "data" in event:
# 如果存在 data 属性
# 替换分析结果字段
if "eventType" in event["data"]:
detection_field_name = event["data"]["eventType"]
if detection_field_name in event["data"]:
event["data"]["_detectionResult"] = event["data"].pop(
detection_field_name
)
event["data"]["_detectionResult"]["typeName"] = event_type_str
if isinstance(
event["data"]["_detectionResult"], list
) and len(event["data"]["_detectionResult"]):
event["data"]["_detectionResult"] = event["data"][
"_detectionResult"
][0]
replace_image_host(event["data"]["_detectionResult"])
# 请求推送 api
logging.info(event)
try:
push_resp = requests.post(push_url, json=event).content.decode(
"utf-8"
)
logging.info(push_resp)
except Exception as e:
logging.error(f"网络错误推送失败: {e}")
except Exception as e:
logging.error("error: ", e)
continue
finally:
await asyncio.sleep(0.5)
def replace_image_host(detection_data):
if "imageUrl" in detection_data:
detection_data["imageUrl"] = detection_data["imageUrl"].replace(
"192.168.1.250", "192.168.11.180"
)
if "visiblePicUrl" in detection_data:
detection_data["visiblePicUrl"] = detection_data["visiblePicUrl"].replace(
"192.168.1.250", "192.168.11.180"
)
def handle_event_level(event_detail, event_levels, event_type):
if event_type in event_levels["high"]:
event_detail["eventLvl"] = 3
elif "eventLvl" not in event_detail:
event_detail["eventLvl"] = 2
def run_app():
asyncio.run(read_event())
if __name__ == "__main__":
asyncio.run(read_event())