From 34767270bd941b136d04a4f5dd428182c087e5e3 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Tue, 25 Apr 2023 08:09:03 -0600 Subject: [PATCH] Cleanup directory remover --- frigate/record/util.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/frigate/record/util.py b/frigate/record/util.py index d02f83f55..d9692c25e 100644 --- a/frigate/record/util.py +++ b/frigate/record/util.py @@ -2,20 +2,18 @@ import os -from frigate.const import RECORD_DIR - -def remove_empty_directories(directory): +def remove_empty_directories(directory: str) -> None: # list all directories recursively and sort them by path, # longest first paths = sorted( - [x[0] for x in os.walk(RECORD_DIR)], + [x[0] for x in os.walk(directory)], key=lambda p: len(str(p)), reverse=True, ) for path in paths: # don't delete the parent - if path == RECORD_DIR: + if path == directory: continue if len(os.listdir(path)) == 0: os.rmdir(path)