github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: update code

pull/14313/head
Eric 2024-10-16 17:55:19 +08:00 committed by Bryan
parent d4f69a7ff8
commit b87650038f
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/common/storage/replay.py
View File

@ -1,6 +1,8 @@
import json import json
import os import os
import shutil
import tarfile import tarfile
import time
from itertools import chain from itertools import chain
from django.core.files.storage import default_storage from django.core.files.storage import default_storage
@ -62,16 +64,17 @@ class SessionPartReplayStorageHandler(object):
# 保存到storage的路径 # 保存到storage的路径
target_path = os.path.join(default_storage.base_location, local_path) target_path = os.path.join(default_storage.base_location, local_path)
target_tmp_path = target_path + f'.tmp{int(time.time())}'
target_dir = os.path.dirname(target_path) target_dir = os.path.dirname(target_path)
if not os.path.isdir(target_dir): if not os.path.isdir(target_dir):
make_dirs(target_dir, exist_ok=True) make_dirs(target_dir, exist_ok=True)
ok, err = storage.download(remote_path, target_path) ok, err = storage.download(remote_path, target_tmp_path)
if not ok: if not ok:
msg = 'Failed download {} file: {}'.format(part_filename, err) msg = 'Failed download {} file: {}'.format(part_filename, err)
logger.error(msg) logger.error(msg)
return None, msg return None, msg
shutil.move(target_tmp_path, target_path)
url = default_storage.url(local_path) url = default_storage.url(local_path)
return local_path, url return local_path, url