mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-23 19:11:14 +01:00
125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
import datetime
|
|
import json
|
|
import logging
|
|
import os
|
|
import queue
|
|
import subprocess as sp
|
|
import threading
|
|
import time
|
|
from collections import defaultdict
|
|
from pathlib import Path
|
|
|
|
import psutil
|
|
|
|
from frigate.config import FrigateConfig
|
|
from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
SECONDS_IN_DAY = 60 * 60 * 24
|
|
|
|
def remove_empty_directories(directory):
|
|
# list all directories recursively and sort them by path,
|
|
# longest first
|
|
paths = sorted(
|
|
[x[0] for x in os.walk(RECORD_DIR)],
|
|
key=lambda p: len(str(p)),
|
|
reverse=True,
|
|
)
|
|
for path in paths:
|
|
# don't delete the parent
|
|
if path == RECORD_DIR:
|
|
continue
|
|
if len(os.listdir(path)) == 0:
|
|
os.rmdir(path)
|
|
|
|
class RecordingMaintainer(threading.Thread):
|
|
def __init__(self, config: FrigateConfig, stop_event):
|
|
threading.Thread.__init__(self)
|
|
self.name = 'recording_maint'
|
|
self.config = config
|
|
self.stop_event = stop_event
|
|
|
|
def move_files(self):
|
|
recordings = [d for d in os.listdir(RECORD_DIR) if os.path.isfile(os.path.join(RECORD_DIR, d)) and d.endswith(".mp4")]
|
|
|
|
files_in_use = []
|
|
for process in psutil.process_iter():
|
|
if process.name() != 'ffmpeg':
|
|
continue
|
|
try:
|
|
flist = process.open_files()
|
|
if flist:
|
|
for nt in flist:
|
|
if nt.path.startswith(RECORD_DIR):
|
|
files_in_use.append(nt.path.split('/')[-1])
|
|
except:
|
|
continue
|
|
|
|
for f in recordings:
|
|
if f in files_in_use:
|
|
continue
|
|
|
|
camera = '-'.join(f.split('-')[:-1])
|
|
start_time = datetime.datetime.strptime(f.split('-')[-1].split('.')[0], '%Y%m%d%H%M%S')
|
|
|
|
ffprobe_cmd = " ".join([
|
|
'ffprobe',
|
|
'-v',
|
|
'error',
|
|
'-show_entries',
|
|
'format=duration',
|
|
'-of',
|
|
'default=noprint_wrappers=1:nokey=1',
|
|
f"{os.path.join(RECORD_DIR,f)}"
|
|
])
|
|
p = sp.Popen(ffprobe_cmd, stdout=sp.PIPE, shell=True)
|
|
(output, err) = p.communicate()
|
|
p_status = p.wait()
|
|
if p_status == 0:
|
|
duration = float(output.decode('utf-8').strip())
|
|
else:
|
|
logger.info(f"bad file: {f}")
|
|
os.remove(os.path.join(RECORD_DIR,f))
|
|
continue
|
|
|
|
directory = os.path.join(RECORD_DIR, start_time.strftime('%Y-%m/%d/%H'), camera)
|
|
|
|
if not os.path.exists(directory):
|
|
os.makedirs(directory)
|
|
|
|
file_name = f"{start_time.strftime('%M.%S.mp4')}"
|
|
|
|
os.rename(os.path.join(RECORD_DIR,f), os.path.join(directory,file_name))
|
|
|
|
def expire_files(self):
|
|
delete_before = {}
|
|
for name, camera in self.config.cameras.items():
|
|
delete_before[name] = datetime.datetime.now().timestamp() - SECONDS_IN_DAY*camera.record.retain_days
|
|
|
|
for p in Path('/media/frigate/recordings').rglob("*.mp4"):
|
|
if not p.parent in delete_before:
|
|
continue
|
|
if p.stat().st_mtime < delete_before[p.parent]:
|
|
p.unlink(missing_ok=True)
|
|
|
|
def run(self):
|
|
counter = 0
|
|
self.expire_files()
|
|
while(True):
|
|
if self.stop_event.is_set():
|
|
logger.info(f"Exiting recording maintenance...")
|
|
break
|
|
|
|
# only expire events every 10 minutes, but check for new files every 10 seconds
|
|
time.sleep(10)
|
|
counter = counter + 1
|
|
if counter > 60:
|
|
self.expire_files()
|
|
remove_empty_directories(RECORD_DIR)
|
|
counter = 0
|
|
|
|
self.move_files()
|
|
|
|
|
|
|