update

hik_push/read_event.py

@@ -0,0 +1,136 @@
+import asyncio
+import json
+from io import BytesIO
+
+import requests
+
+from event_map_mapping import mapping
+
+from redis import asyncio as aioredis
+
+headers = {"Authorization": "Bearer t-g1041oeFF3LTS5MA3HQYUHYFOK246EAEPFJWPRWP"}
+
+
+# 获取事件的图片
+def get_image_url(event):
+    analysis_key = mapping.get(event['eventType'])['analyze_key']
+    print(analysis_key)
+    if analysis_key is None:
+        return []
+    images = []
+    for el in event['data'][analysis_key]:
+        images.append(el['imageUrl'])
+    return images
+
+
+def get_value_by_nested_key(my_dict, nested_key):
+    ret = my_dict
+    for key in nested_key:
+        ret = ret[key]
+    return ret
+
+
+# 上次图片到飞书
+def upload_image(url):
+    image_content = requests.get(url).content
+    payload = {'image_type': 'message'}
+    files = [
+        ('image', ('file', BytesIO(image_content), 'application/octet-stream'))
+    ]
+    image = requests.request("POST", "https://open.feishu.cn/open-apis/im/v1/images", data=payload,
+                             files=files, headers=headers).content.decode('utf-8')
+    image_key = json.loads(image)['data']['image_key']
+    return image_key
+
+
+# 推送到飞书
+def push_to_feishu(data):
+    push_url = "https://open.feishu.cn/open-apis/message/v4/batch_send/"
+    # push_url = "http://192.168.20.115:8000/ipaasuat/engine_company/anycross/trigger/callback/MmRhZTE4OTRiYjVkZDQ5YWNmOGRmZDI0NjQ1MTBlODUw/1.0.0"
+    data = json.dumps(data)
+    r = requests.post(push_url, data=data, headers=headers).content.decode('utf-8')
+    print(r)
+
+
+async def read_event():
+    redis_client = await aioredis.from_url("redis://localhost")
+    while True:
+        data = await redis_client.brpop("event")
+        event_json = json.loads(data[1].decode('utf-8'))
+        try:
+            events = event_json["params"]["events"]
+            for event in events:
+                event_type = mapping.get(event['eventType'], {"type": "未知事件类型"})
+                print(event_type)
+                # TODO: 上传图片到飞书链接
+                images = get_image_url(event)
+                feishu_image_url = upload_image(images[0])
+                print(feishu_image_url)
+
+                # TODO: 根据ip_address获取对应userid
+                #
+                ip_address = event["data"]["ipAddress"]
+                # channel_name = get_value_by_nested_key(event, ("data", "channelName"))
+
+                # TODO: 向飞书推送消息
+                push_data = {
+                    "user_ids": [
+                        "8dg79c18"
+                    ],
+                    "msg_type": "interactive",
+                    "card": {
+                        "config": {
+                            "wide_screen_mode": True
+                        },
+                        "elements": [
+                            {
+                                "tag": "markdown",
+                                "content": "这里是卡片文本，支持使用markdown标签设置文本格式。例如：\n*斜体* 、**粗体**、~~删除线~~、[文字链接](https://www.feishu.cn)、<at id=all></at>、<font color='red'> 彩色文本 </font>"
+                            },
+                            {
+                                "alt": {
+                                    "content": "",
+                                    "tag": "plain_text"
+                                },
+                                "img_key": feishu_image_url,
+                                "tag": "img",
+                                "mode": "fit_horizontal",
+                                "compact_width": False
+                            },
+                            {
+                                "tag": "action",
+                                "actions": [
+                                    {
+                                        "tag": "button",
+                                        "text": {
+                                            "tag": "plain_text",
+                                            "content": "这是跳转按钮"
+                                        },
+                                        "type": "primary",
+                                        "url": "https://feishu.cn"
+                                    }
+                                ]
+                            }
+                        ],
+                        "header": {
+                            "template": "blue",
+                            "title": {
+                                "content": event_type["type"],
+                                "tag": "plain_text"
+                            }
+                        }
+                    }
+                }
+                push_to_feishu(push_data)
+                print(event_type["type"], ip_address)
+        except Exception as e:
+            print("error: ", e)
+            continue
+        # except Exception as e:
+        #     print("error", e)
+        #     # await redis_client.lpush("event", json.dumps(event_json))
+        #     continue
+
+
+if __name__ == "__main__":
+    asyncio.run(read_event())