github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-06-28 07:47:10 +00:00
Code Issues Projects Releases Wiki Activity

perf: clean mp4 replay file

Browse Source 
perf: refactor code to clean files
This commit is contained in:
Eric 2024-06-25 16:31:32 +08:00 committed by 老广
parent 7b4d3c44f8
commit 908181af64
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
apps/audits/tasks.py
View File

@ -92,6 +92,22 @@ def batch_delete(queryset, batch_size=3000):
model.objects.filter(id__in=list(pks)).delete() model.objects.filter(id__in=list(pks)).delete()
def remove_files_by_days(root_path, days, file_types=None):
if file_types is None:
file_types = ['.json', '.tar', '.gz', '.mp4']
need_rm_files = []
expire_date = timezone.now() - timezone.timedelta(days=days)
timestamp = expire_date.timestamp()
for root, dirs, files in os.walk(root_path):
for file in files:
if any(file.endswith(file_type) for file_type in file_types):
file_path = os.path.join(root, file)
if os.path.getmtime(file_path) <= timestamp:
need_rm_files.append(file_path)
for file in need_rm_files:
os.remove(file)
def clean_expired_session_period(): def clean_expired_session_period():
logger.info("Start clean expired session record, commands and replay") logger.info("Start clean expired session record, commands and replay")
days = get_log_keep_day('TERMINAL_SESSION_KEEP_DURATION') days = get_log_keep_day('TERMINAL_SESSION_KEEP_DURATION')
@ -105,10 +121,7 @@ def clean_expired_session_period():
logger.info("Clean session item done") logger.info("Clean session item done")
batch_delete(expired_commands) batch_delete(expired_commands)
logger.info("Clean session command done") logger.info("Clean session command done")
command = "find %s -mtime +%s \\( -name '*.json' -o -name '*.tar' -o -name '*.gz' \\) -exec rm -f {} \\;" % ( remove_files_by_days(replay_dir, days)
replay_dir, days
)
subprocess.call(command, shell=True)
command = "find %s -type d -empty -delete;" % replay_dir command = "find %s -type d -empty -delete;" % replay_dir
subprocess.call(command, shell=True) subprocess.call(command, shell=True)
logger.info("Clean session replay done") logger.info("Clean session replay done")