formatting cleanup

frigate/record.py

@@ -18,41 +18,47 @@ 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)
+    # 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.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")]
+        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():
             try:
-                if process.name() != 'ffmpeg':
+                if process.name() != "ffmpeg":
                     continue
                 flist = process.open_files()
                 if flist:
                     for nt in flist:
                         if nt.path.startswith(RECORD_DIR):
-                            files_in_use.append(nt.path.split('/')[-1])
+                            files_in_use.append(nt.path.split("/")[-1])
             except:
                 continue
 
@@ -60,44 +66,53 @@ class RecordingMaintainer(threading.Thread):
             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)}"
-            ])
+            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())
+                duration = float(output.decode("utf-8").strip())
             else:
                 logger.info(f"bad file: {f}")
-                os.remove(os.path.join(RECORD_DIR,f))
+                os.remove(os.path.join(RECORD_DIR, f))
                 continue
 
-            directory = os.path.join(RECORD_DIR, start_time.strftime('%Y-%m/%d/%H'), camera)
+            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))
+            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
+            delete_before[name] = (
+                datetime.datetime.now().timestamp()
+                - SECONDS_IN_DAY * camera.record.retain_days
+            )
 
-        for p in Path('/media/frigate/recordings').rglob("*.mp4"):
+        for p in Path("/media/frigate/recordings").rglob("*.mp4"):
             if not p.parent.name in delete_before:
                 continue
             if p.stat().st_mtime < delete_before[p.parent.name]:
@@ -106,7 +121,7 @@ class RecordingMaintainer(threading.Thread):
     def run(self):
         counter = 0
         self.expire_files()
-        while(True):
+        while True:
             if self.stop_event.is_set():
                 logger.info(f"Exiting recording maintenance...")
                 break
@@ -120,6 +135,3 @@ class RecordingMaintainer(threading.Thread):
                 counter = 0
 
             self.move_files()
-
-
-