admin/marcel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update rtps_daily_recorder.py

Browse Source
This commit is contained in:
Metacube
2025-09-21 19:25:58 +02:00
committed by GitHub
parent 116d9ee7dd
commit 1d50147c75
1 changed files with 107 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rtps_daily_recorder.py
+107 -69
View File
@@ -15,15 +15,15 @@ except ImportError:
CONFIG = { CONFIG = {
"BASE_DIR": "/var/www/html/image/", # Für RTSP-Screenshots und Videos "BASE_DIR": "/var/www/html/image/", # Für RTSP-Screenshots und Videos
"RESIZE_DIR": "/var/www/html/images/", # Für die zu resizenden Bilder "RESIZE_DIR": "/var/www/html/images/", # Für die zu resizenden Bilder
"RTSP_URL": "rtsp://aurora:%2B61946194@192.168.1.133:88/videoMain", # @ escaped als %40 "RTSP_URL": "rtsp://aurora:%2B61946194@192.168.1.133:88/videoMain",
"LOG_FILE": "/var/www/html/rtsp-recorder.log", "LOG_FILE": "/var/www/html/rtsp-recorder.log",
"HOURS_TO_RUN": 24, "HOURS_TO_RUN": 24, # ✅ 24 Stunden für ein komplettes Tagesvideo
"SCREENSHOT_INTERVAL": 33, #3600, "SCREENSHOT_INTERVAL": 33, # Alle 33 Sekunden ein Screenshot
"VIDEO_FPS": 5, "VIDEO_FPS": 5,
"VIDEO_RETENTION_DAYS": 7, "VIDEO_RETENTION_DAYS": 7,
"TARGET_WIDTH": 274, "TARGET_WIDTH": 274,
"TARGET_HEIGHT": 52, "TARGET_HEIGHT": 52,
"SCREENSHOTS_PER_HOUR": 109, # 12 Screenshots pro Stunde (NEU) "SCREENSHOTS_PER_HOUR": 109, # ✅ Korrekt: 3600/33 = 109
} }
# Logging Setup # Logging Setup
@@ -37,7 +37,6 @@ logging.basicConfig(
] ]
) )
class CameraService: class CameraService:
def __init__(self): def __init__(self):
self.execution_count = 0 self.execution_count = 0
@@ -58,6 +57,76 @@ class CameraService:
return False return False
return True return True
def check_and_create_missing_videos(self):
"""Prüft ob Videos für die letzten 7 Tage existieren und erstellt fehlende"""
logging.info("Prüfe auf fehlende Videos der letzten 7 Tage...")
for days_ago in range(CONFIG["VIDEO_RETENTION_DAYS"]):
target_date = datetime.now() - timedelta(days=days_ago)
date_str = target_date.strftime("%Y%m%d")
# Prüfe ob Video für diesen Tag existiert
existing_videos = list(self.base_dir.glob(f"daily_video_{date_str}_*.mp4"))
if not existing_videos:
logging.info(f"Kein Video für {date_str} gefunden. Erstelle aus vorhandenen Screenshots...")
self.create_video_for_date(target_date)
else:
logging.info(f"Video für {date_str} bereits vorhanden: {existing_videos[0].name}")
def create_video_for_date(self, target_date):
"""Erstellt ein Video für ein spezifisches Datum aus vorhandenen Screenshots"""
date_str = target_date.strftime("%Y%m%d")
# Sammle alle Screenshots für diesen Tag
start_time = target_date.replace(hour=0, minute=0, second=0)
end_time = start_time + timedelta(days=1)
jpg_files = []
for jpg in sorted(self.base_dir.glob(f"screenshot_{date_str}_*.jpg")):
jpg_files.append(jpg)
if len(jpg_files) < 10: # Mindestens 10 Bilder für ein sinnvolles Video
logging.warning(f"Zu wenige Screenshots für {date_str} ({len(jpg_files)} gefunden)")
return False
# Erstelle Video
timestamp = target_date.strftime("%Y%m%d_%H%M%S")
output_file = self.base_dir / f"daily_video_{timestamp}.mp4"
temp_list = Path(f"/tmp/files_{timestamp}.txt")
try:
with temp_list.open('w') as f:
for jpg in jpg_files:
f.write(f"file '{jpg.absolute()}'\n")
f.write(f"duration 0.2\n")
cmd = [
'ffmpeg', '-y',
'-f', 'concat',
'-safe', '0',
'-i', str(temp_list),
'-vsync', 'vfr',
'-c:v', 'libx264',
'-pix_fmt', 'yuv420p',
'-preset', 'fast',
'-crf', '23',
str(output_file)
]
subprocess.run(cmd, check=True, capture_output=True)
if output_file.exists() and output_file.stat().st_size > 0:
logging.info(f"Nachträglich Video erstellt für {date_str}: {output_file}")
return True
except Exception as e:
logging.error(f"Fehler beim Erstellen des Videos für {date_str}: {e}")
return False
finally:
if temp_list.exists():
temp_list.unlink()
def resize_image(self, image_path): def resize_image(self, image_path):
"""Passt die Bildgröße an die Zielgröße an und behält das Seitenverhältnis bei""" """Passt die Bildgröße an die Zielgröße an und behält das Seitenverhältnis bei"""
try: try:
@@ -66,35 +135,24 @@ class CameraService:
target_width = CONFIG["TARGET_WIDTH"] target_width = CONFIG["TARGET_WIDTH"]
target_height = CONFIG["TARGET_HEIGHT"] target_height = CONFIG["TARGET_HEIGHT"]
# Berechne das Seitenverhältnis
aspect_ratio = current_width / current_height aspect_ratio = current_width / current_height
target_ratio = target_width / target_height target_ratio = target_width / target_height
if current_width != target_width or current_height != target_height: if current_width != target_width or current_height != target_height:
# Berechne neue Dimensionen mit Hintergrund
if aspect_ratio > target_ratio: if aspect_ratio > target_ratio:
# Bild ist breiter als Zielformat
new_width = target_width new_width = target_width
new_height = int(target_width / aspect_ratio) new_height = int(target_width / aspect_ratio)
else: else:
# Bild ist höher als Zielformat
new_height = target_height new_height = target_height
new_width = int(target_height * aspect_ratio) new_width = int(target_height * aspect_ratio)
# Erstelle neues Bild mit WEISSEM Hintergrund
background = Image.new('RGB', (target_width, target_height), (255, 255, 255)) background = Image.new('RGB', (target_width, target_height), (255, 255, 255))
# Resize Original
resized_img = img.resize((new_width, new_height), Image.Resampling.LANCZOS) resized_img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
# Berechne Position zum Zentrieren
x = (target_width - new_width) // 2 x = (target_width - new_width) // 2
y = (target_height - new_height) // 2 y = (target_height - new_height) // 2
# Füge das resized Bild in den Hintergrund ein
background.paste(resized_img, (x, y)) background.paste(resized_img, (x, y))
# Speichern
background.save(image_path, "PNG", optimize=True) background.save(image_path, "PNG", optimize=True)
logging.info(f"Bild angepasst: {image_path}") logging.info(f"Bild angepasst: {image_path}")
return True return True
@@ -103,15 +161,11 @@ class CameraService:
logging.error(f"Fehler bei Bildanpassung {image_path}: {e}") logging.error(f"Fehler bei Bildanpassung {image_path}: {e}")
return False return False
def resize_all_images(self): def resize_all_images(self):
"""Passt alle PNG-Bilder im RESIZE_DIR an""" """Passt alle PNG-Bilder im RESIZE_DIR an"""
for png_file in Path(CONFIG["RESIZE_DIR"]).glob("*.png"): # *.png statt *.jpg for png_file in Path(CONFIG["RESIZE_DIR"]).glob("*.png"):
self.resize_image(png_file) self.resize_image(png_file)
def take_screenshot(self): def take_screenshot(self):
"""Erstelle einen Screenshot von der RTSP-Kamera""" """Erstelle einen Screenshot von der RTSP-Kamera"""
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -120,16 +174,13 @@ class CameraService:
logging.info("Erstelle Screenshot...") logging.info("Erstelle Screenshot...")
try: try:
cmd = [ cmd = [
'ffmpeg', '-y', '-rtsp_transport', 'tcp', 'ffmpeg', '-y', '-rtsp_transport', 'tcp',
'-analyzeduration', '20M', '-probesize', '20M', '-analyzeduration', '20M', '-probesize', '20M',
'-i', CONFIG["RTSP_URL"], '-vframes', '1', '-q:v', '2', '-i', CONFIG["RTSP_URL"], '-vframes', '1', '-q:v', '2',
'-ss', '00:00:01', str(output_file) '-ss', '00:00:01', str(output_file)
] ]
# Verbose Logging hinzufügen
process = subprocess.run(cmd, capture_output=True, text=True) process = subprocess.run(cmd, capture_output=True, text=True)
if process.returncode != 0: if process.returncode != 0:
logging.error(f"FFmpeg Fehler: {process.stderr}") logging.error(f"FFmpeg Fehler: {process.stderr}")
@@ -146,18 +197,6 @@ class CameraService:
logging.error(f"Unerwarteter Fehler: {e}") logging.error(f"Unerwarteter Fehler: {e}")
return False return False
def resize_all_images(self):
"""Passt NUR die PNG-Bilder im RESIZE_DIR an"""
# Nur Bilder in /var/www/html/images/ werden resized
for png_file in Path(CONFIG["RESIZE_DIR"]).glob("*.png"):
self.resize_image(png_file)
logging.info("Website-Bilder in /images/ angepasst")
def create_daily_video(self): def create_daily_video(self):
"""Erstelle ein Video aus den gesammelten Screenshots""" """Erstelle ein Video aus den gesammelten Screenshots"""
logging.info("Starte Videoerstellung...") logging.info("Starte Videoerstellung...")
@@ -166,10 +205,6 @@ class CameraService:
output_file = self.base_dir / f"daily_video_{timestamp}.mp4" output_file = self.base_dir / f"daily_video_{timestamp}.mp4"
temp_list = Path(f"/tmp/files_{timestamp}.txt") temp_list = Path(f"/tmp/files_{timestamp}.txt")
# Finde das vorherige Video
previous_videos = sorted(self.base_dir.glob("daily_video_*.mp4"))
previous_video = previous_videos[-1] if previous_videos else None
cutoff_time = datetime.now() - timedelta(hours=CONFIG["HOURS_TO_RUN"]) cutoff_time = datetime.now() - timedelta(hours=CONFIG["HOURS_TO_RUN"])
jpg_files = sorted([ jpg_files = sorted([
f for f in self.base_dir.glob("screenshot_*.jpg") f for f in self.base_dir.glob("screenshot_*.jpg")
@@ -181,13 +216,11 @@ class CameraService:
return False return False
try: try:
# Erstelle die Dateiliste
with temp_list.open('w') as f: with temp_list.open('w') as f:
for jpg in jpg_files: for jpg in jpg_files:
f.write(f"file '{jpg.absolute()}'\n") f.write(f"file '{jpg.absolute()}'\n")
f.write(f"duration 0.2\n") # Jedes Bild 0.2 Sekunden zeigen f.write(f"duration 0.2\n")
# FFmpeg Befehl
cmd = [ cmd = [
'ffmpeg', '-y', 'ffmpeg', '-y',
'-f', 'concat', '-f', 'concat',
@@ -201,19 +234,15 @@ class CameraService:
str(output_file) str(output_file)
] ]
# Führe FFmpeg aus
subprocess.run(cmd, check=True, capture_output=True) subprocess.run(cmd, check=True, capture_output=True)
# Prüfe ob das neue Video erfolgreich erstellt wurde
if output_file.exists() and output_file.stat().st_size > 0: if output_file.exists() and output_file.stat().st_size > 0:
logging.info(f"Video erstellt: {output_file}") logging.info(f"Video erstellt: {output_file}")
# Lösche das alte Video erst nach erfolgreicher Erstellung des neuen # ✅ WICHTIG: Alte Videos NICHT löschen!
# if previous_video and previous_video != output_file: # Videos bleiben 7 Tage erhalten
# previous_video.unlink()
# logging.info(f"Altes Video gelöscht: {previous_video}")
# Lösche die verwendeten JPGs # Lösche nur die verwendeten JPGs um Platz zu sparen
for jpg in jpg_files: for jpg in jpg_files:
jpg.unlink() jpg.unlink()
logging.info(f"{len(jpg_files)} Screenshots verarbeitet und gelöscht") logging.info(f"{len(jpg_files)} Screenshots verarbeitet und gelöscht")
@@ -232,21 +261,21 @@ class CameraService:
if temp_list.exists(): if temp_list.exists():
temp_list.unlink() temp_list.unlink()
def cleanup_old_files(self): def cleanup_old_files(self):
"""Lösche alte Video- und Bilddateien""" """Lösche alte Video- und Bilddateien (nur älter als 7 Tage)"""
cutoff_time = datetime.now() - timedelta(days=CONFIG["VIDEO_RETENTION_DAYS"]) cutoff_time = datetime.now() - timedelta(days=CONFIG["VIDEO_RETENTION_DAYS"])
for video in self.base_dir.glob("daily_video_*.mp4"): # Lösche nur Videos älter als 7 Tage
if video.stat().st_mtime < cutoff_time.timestamp(): # for video in self.base_dir.glob("daily_video_*.mp4"):
video.unlink() # if video.stat().st_mtime < cutoff_time.timestamp():
logging.info(f"Altes Video gelöscht: {video}") # video.unlink()
# logging.info(f"Altes Video gelöscht (>7 Tage): {video}")
# Lösche verwaiste Screenshots älter als 7 Tage
for jpg in self.base_dir.glob("screenshot_*.jpg"): for jpg in self.base_dir.glob("screenshot_*.jpg"):
if jpg.stat().st_mtime < cutoff_time.timestamp(): if jpg.stat().st_mtime < cutoff_time.timestamp():
jpg.unlink() jpg.unlink()
logging.info(f"Verwaistes Bild gelöscht: {jpg}") logging.info(f"Verwaistes Bild gelöscht (>7 Tage): {jpg}")
def cleanup(self): def cleanup(self):
"""Aufräumen vor Beendigung""" """Aufräumen vor Beendigung"""
@@ -266,16 +295,21 @@ def main():
if not service.validate_config(): if not service.validate_config():
raise RuntimeError("Ungültige Konfiguration") raise RuntimeError("Ungültige Konfiguration")
# ✅ NEU: Beim Start prüfen und fehlende Videos erstellen
service.check_and_create_missing_videos()
while True: # Endlosschleife while True: # Endlosschleife
service.execution_count = 0 service.execution_count = 0
# ✅ KORRIGIERT: Verwende SCREENSHOTS_PER_HOUR statt 12
total_screenshots = CONFIG["HOURS_TO_RUN"] * CONFIG["SCREENSHOTS_PER_HOUR"] total_screenshots = CONFIG["HOURS_TO_RUN"] * CONFIG["SCREENSHOTS_PER_HOUR"]
logging.info(f"Starte neuen 24-Stunden-Zyklus mit {total_screenshots} Screenshots")
service.resize_all_images() service.resize_all_images()
while service.execution_count < total_screenshots: while service.execution_count < total_screenshots:
if service.take_screenshot(): if service.take_screenshot():
service.execution_count += 1 service.execution_count += 1
retry_count = 0 # Reset retry counter on success retry_count = 0
logging.info(f"Screenshot {service.execution_count} von {total_screenshots}") logging.info(f"Screenshot {service.execution_count} von {total_screenshots}")
else: else:
retry_count += 1 retry_count += 1
@@ -286,17 +320,23 @@ def main():
time.sleep(60) time.sleep(60)
continue continue
service.cleanup_old_files() # Aufräumen nur alle 100 Screenshots (nicht bei jedem)
if service.execution_count % 100 == 0:
service.cleanup_old_files()
if service.execution_count < total_screenshots: if service.execution_count < total_screenshots:
time.sleep(CONFIG["SCREENSHOT_INTERVAL"]) # Nutze den Wert aus CONFIG time.sleep(CONFIG["SCREENSHOT_INTERVAL"]) # 33 Sekunden warten
# Nach 24 Stunden: Video erstellen
if service.create_daily_video(): if service.create_daily_video():
logging.info("Täglicher Zyklus erfolgreich abgeschlossen") logging.info("24-Stunden-Video erfolgreich erstellt")
else: else:
logging.error("Fehler beim Erstellen des Tagesvideos") logging.error("Fehler beim Erstellen des Tagesvideos")
logging.info("Starte neuen 12-Stunden-Zyklus...") # ✅ NEU: Nach jedem Zyklus prüfen ob alle Tage Videos haben
service.check_and_create_missing_videos()
logging.info("Starte neuen 24-Stunden-Zyklus...")
except KeyboardInterrupt: except KeyboardInterrupt:
logging.info("Programm durch Benutzer beendet") logging.info("Programm durch Benutzer beendet")
@@ -306,7 +346,5 @@ def main():
finally: finally:
service.cleanup() service.cleanup()
if __name__ == "__main__": if __name__ == "__main__":
main() main()