multiple fixes and improvements. ready to deploy
This commit is contained in:
@@ -3,9 +3,9 @@ import time
|
||||
import requests
|
||||
import numpy as np
|
||||
from datetime import datetime
|
||||
import os
|
||||
import json
|
||||
import logging as lg
|
||||
import logging.handlers as lg_handlers
|
||||
import argparse
|
||||
|
||||
class CameraMonitor:
|
||||
@@ -13,20 +13,28 @@ class CameraMonitor:
|
||||
self.config = config
|
||||
self.logger = logger
|
||||
self.last_detection = 0
|
||||
self.capture = self.initialize_camera()
|
||||
|
||||
def initialize_camera(self):
|
||||
"""
|
||||
Initialize the camera and return the VideoCapture object.
|
||||
"""
|
||||
cap = cv2.VideoCapture(self.config['camera_address'])
|
||||
cap = cv2.VideoCapture(
|
||||
self.config['camera_address'],
|
||||
cv2.CAP_V4L2
|
||||
)
|
||||
if not cap.isOpened():
|
||||
raise Exception(f"Failed to open camera at {self.config['camera_address']}")
|
||||
self.logger.info("Camera initialized")
|
||||
print("Camera initialized")
|
||||
return cap
|
||||
|
||||
def log_scene_change(self):
|
||||
"""
|
||||
Log a scene change notification to the Scenesense server.
|
||||
"""
|
||||
self.logger.info(f'Scene change detected')
|
||||
# print("Scene change detected")
|
||||
current_time = int(time.time())
|
||||
human_readable_time = datetime.utcfromtimestamp(
|
||||
current_time
|
||||
@@ -49,11 +57,20 @@ class CameraMonitor:
|
||||
headers=headers
|
||||
)
|
||||
response.raise_for_status()
|
||||
self.logger.info(f'Scene change logged: {description}')
|
||||
self.logger.info(
|
||||
f'Scene change logged to server db: {description}'
|
||||
)
|
||||
# print(f'Scene change logged: {description}')
|
||||
self.last_detection = current_time
|
||||
|
||||
|
||||
except requests.exceptions.RequestException as e:
|
||||
self.logger.error(f'Error connecting to Scenesense server: {e}')
|
||||
# print(f'Error connecting to Scenesense server: {e}')
|
||||
except requests.exceptions.HTTPError as e:
|
||||
self.logger.error(f'HTTP error: {e}')
|
||||
# print(f'HTTP error: {e}')
|
||||
|
||||
|
||||
def detect_scene_change(self, previous_frame, current_frame):
|
||||
"""
|
||||
@@ -61,29 +78,42 @@ class CameraMonitor:
|
||||
"""
|
||||
frame_diff = cv2.absdiff(previous_frame, current_frame)
|
||||
mean_diff = np.mean(frame_diff)
|
||||
return mean_diff > self.config['sensitivity_threshold']
|
||||
threshold = self.config['sensitivity_threshold']
|
||||
# self.logger.info(
|
||||
# f'Frame Diff: {frame_diff}\nMean Diff: {mean_diff}\nSensitivity Threshold: {threshold}'
|
||||
# )
|
||||
# print(
|
||||
# f'Frame Diff: {frame_diff}\nMean Diff: {mean_diff}\nSensitivity Threshold: {threshold}'
|
||||
# )
|
||||
return mean_diff > threshold
|
||||
|
||||
def run(self):
|
||||
self.logger.info("Camera Monitor started")
|
||||
print("Camera Monitor started")
|
||||
try:
|
||||
with self.initialize_camera() as cap:
|
||||
previous_frame = None
|
||||
previous_frame = None
|
||||
|
||||
while True:
|
||||
ret, current_frame = cap.read()
|
||||
while True:
|
||||
ret, current_frame = self.capture.read()
|
||||
|
||||
if not ret:
|
||||
raise Exception("Failed to read frame")
|
||||
if not ret:
|
||||
raise Exception("Failed to read frame")
|
||||
|
||||
if previous_frame is not None:
|
||||
if self.detect_scene_change(previous_frame, current_frame):
|
||||
self.log_scene_change()
|
||||
|
||||
previous_frame = current_frame
|
||||
if previous_frame is not None:
|
||||
if self.detect_scene_change(previous_frame, current_frame):
|
||||
self.log_scene_change()
|
||||
|
||||
|
||||
previous_frame = current_frame
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Camera monitoring stopped by user.")
|
||||
print("Camera monitoring stopped by user.")
|
||||
except Exception as e:
|
||||
self.logger.error(f"An error occurred: {e}")
|
||||
# print(f"An error occurred: {e}")
|
||||
finally:
|
||||
self.capture.release()
|
||||
|
||||
def parse_command_line_args():
|
||||
parser = argparse.ArgumentParser(description='Camera Monitor for Scenesense')
|
||||
@@ -100,14 +130,12 @@ def load_config(config_path):
|
||||
raise Exception(f"Invalid JSON format in config file: {config_path}")
|
||||
|
||||
def setup_logging(log_file='logs/camera_monitor.log'):
|
||||
# logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||
# return logging.getLogger('CameraMonitor')
|
||||
# Configure the logger
|
||||
logger = lg.getLogger('CameraMonitor')
|
||||
logger.setLevel(lg.INFO)
|
||||
|
||||
# Create a TimedRotatingFileHandler for log rotation
|
||||
file_handler = lg.handlers.TimedRotatingFileHandler(
|
||||
file_handler = lg_handlers.TimedRotatingFileHandler(
|
||||
log_file, when='midnight', backupCount=7
|
||||
) # Keeps logs for 7 days
|
||||
file_handler.suffix = '%Y-%m-%d' # Append date to log file names
|
||||
|
||||
Reference in New Issue
Block a user