67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
# coding=utf-8
|
|
import os
|
|
import cv2
|
|
import logging
|
|
import pymysql
|
|
import threading
|
|
import json
|
|
from dotenv import load_dotenv
|
|
|
|
class Camera:
|
|
def __init__(self, id, rtsp, status):
|
|
self.id = id
|
|
self.rtsp = rtsp
|
|
self.status = status
|
|
|
|
def healthbeat(self):
|
|
cap = cv2.VideoCapture(self.rtsp)
|
|
ret, _ = cap.read()
|
|
cap.release()
|
|
|
|
# 摄像头状态: 1 在线, 2 离线
|
|
status = 1 if ret == True else 2
|
|
|
|
if status == 1:
|
|
logging.info("摄像头ID: %s, 状态: 在线" % (self.id))
|
|
else:
|
|
logging.warning("摄像头ID: %s, 状态: 离线" % (self.id))
|
|
|
|
if (status == 1 and self.status == 2) or (status == 2 and self.status == 1):
|
|
conn = get_connection()
|
|
|
|
with conn:
|
|
with conn.cursor() as cursor:
|
|
cursor.execute("update devices set state = %s where id = %s", (status, self.id))
|
|
conn.commit()
|
|
|
|
def init_logger():
|
|
logging.basicConfig(filename='healthbeat.log', level=logging.INFO, format='[%(asctime)s] %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
|
|
|
|
def get_connection():
|
|
return pymysql.connect(
|
|
host=os.environ.get("DB_HOST"),
|
|
port=int(os.environ.get("DB_PORT")),
|
|
user=os.environ.get("DB_USERNAME"),
|
|
password=os.environ.get("DB_PASSWORD"),
|
|
database=os.environ.get("DB_DATABASE"),
|
|
charset='utf8mb4'
|
|
)
|
|
|
|
if __name__ == '__main__':
|
|
load_dotenv()
|
|
# 初始化日志格式
|
|
init_logger()
|
|
|
|
conn = get_connection()
|
|
|
|
with conn:
|
|
with conn.cursor() as cursor:
|
|
cursor.execute("SELECT id, state, extends FROM devices WHERE type = 1 AND state in (1, 2) ORDER BY id ASC")
|
|
rows = cursor.fetchall()
|
|
|
|
for row in rows:
|
|
extends = json.loads(row[2])
|
|
c = Camera(row[0], extends["rtsp_url"], row[1])
|
|
t = threading.Thread(target=c.healthbeat)
|
|
t.start()
|
|
|