From 1223d312c8b2560a35f06bd785a5b1aa25a823e5 Mon Sep 17 00:00:00 2001
From: Jing Li <lym125@gmail.com>
Date: Thu, 7 Sep 2023 21:11:51 +0800
Subject: [PATCH] init

---
 .env.example     |  5 ++++
 .gitignore       |  2 ++
 Healthbeat.py    | 66 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  4 +++
 4 files changed, 77 insertions(+)
 create mode 100644 .env.example
 create mode 100644 .gitignore
 create mode 100644 Healthbeat.py
 create mode 100644 requirements.txt

diff --git a/.env.example b/.env.example
new file mode 100644
index 0000000..8453286
--- /dev/null
+++ b/.env.example
@@ -0,0 +1,5 @@
+DB_HOST=127.0.0.1
+DB_PORT=5432
+DB_DATABASE=postgres
+DB_USERNAME=postgres
+DB_PASSWORD=password
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..35b5bc4
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+.env
+healthbeat.log
\ No newline at end of file
diff --git a/Healthbeat.py b/Healthbeat.py
new file mode 100644
index 0000000..53ef93f
--- /dev/null
+++ b/Healthbeat.py
@@ -0,0 +1,66 @@
+# 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))
+
+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()
+    
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..5ba3c06
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,4 @@
+numpy==1.23.5
+opencv-python==4.6.0.66
+PyMySQL==1.1.0
+python-dotenv==0.21.0