69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import json
|
||
import socket
|
||
# 导入日志
|
||
from logger import LOGGER
|
||
|
||
|
||
class Api:
|
||
"""
|
||
API类，实现最简单的tcp通讯，如果需要自动重连，队列等业务代码请自行实现
|
||
"""
|
||
def __init__(self):
|
||
"""
|
||
Attributes:
|
||
self.ignore_result: 为True时，不发送检测结果，为False则发送检测结果
|
||
self.ignore_alert: 为True时，不发送告警信息，为False则发送告警信息
|
||
self.draw_image: 为True时，告警图片会画上告警信息，为False则不画
|
||
self.ignore_alert_video: 为True时，不发送告警视频，为False则发送
|
||
"""
|
||
self.ignore_result = True
|
||
self.ignore_alert = False
|
||
self.draw_image = True
|
||
self.ignore_alert_video = True
|
||
|
||
def send_result_callback(self, result):
|
||
"""
|
||
发送检测结果回调函数
|
||
Args:
|
||
result: 检测结果数据
|
||
Returns:
|
||
"""
|
||
pass
|
||
|
||
def send_alert_callback(self, alert):
|
||
"""
|
||
发送告警信息回调函数
|
||
Args:
|
||
alert: 告警数据
|
||
Returns:
|
||
"""
|
||
|
||
try:
|
||
LOGGER.info('发送TCP告警')
|
||
# 创建TCP socket
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
# 连接到目标服务器（这里用示例地址和端口）
|
||
s.connect(('192.168.0.4', 10001))
|
||
|
||
# 为了方便查看日志，去掉base64编码图片
|
||
alert.pop('image')
|
||
|
||
json_data = json.dumps(alert, ensure_ascii=False)
|
||
data = json_data.encode('utf-8')
|
||
|
||
# 发送完整数据
|
||
s.sendall(data)
|
||
|
||
LOGGER.info(f'告警已发送至TCP服务器: {data}')
|
||
except Exception as e:
|
||
LOGGER.error(f'发送TCP告警失败: {str(e)}')
|
||
|
||
def send_alert_video_callback(self, alert_video):
|
||
"""
|
||
发送告警视频回调函数
|
||
Args:
|
||
alert_video: 告警视频数据
|
||
Returns:
|
||
"""
|
||
pass
|