Cleanup directory remover

This commit is contained in:
Nick Mowen 2023-04-25 08:09:03 -06:00
parent b6a78f82a4
commit 34767270bd

View File

@ -2,20 +2,18 @@
import os import os
from frigate.const import RECORD_DIR
def remove_empty_directories(directory: str) -> None:
def remove_empty_directories(directory):
# list all directories recursively and sort them by path, # list all directories recursively and sort them by path,
# longest first # longest first
paths = sorted( 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)), key=lambda p: len(str(p)),
reverse=True, reverse=True,
) )
for path in paths: for path in paths:
# don't delete the parent # don't delete the parent
if path == RECORD_DIR: if path == directory:
continue continue
if len(os.listdir(path)) == 0: if len(os.listdir(path)) == 0:
os.rmdir(path) os.rmdir(path)