nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-04-05 03:38:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 优化临时文件的创建和上传逻辑

Browse Source 
- 使用 with 语句自动管理临时文件的创建和关闭，提高代码的可读性和安全性
- 优化了代码结构，减少了嵌套的 try 语句，使代码更加清晰
This commit is contained in:
developer-wlj
2025-09-14 16:46:27 +08:00
parent 3b194d282e
commit 77c0f8f39e
1 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/chain/media.py
View File

@@ -458,17 +458,16 @@ class MediaChain(ChainBase):
if not _fileitem or not _content or not _path:
return
# 使用tempfile创建临时文件，手动删除
tmp_file = NamedTemporaryFile(delete=False, suffix=_path.suffix)
tmp_file_path = Path(tmp_file.name)
try:
with NamedTemporaryFile(delete=False, suffix=_path.suffix) as tmp_file:
tmp_file_path = Path(tmp_file.name)
# 写入内容
if isinstance(_content, bytes):
tmp_file.write(_content)
else:
tmp_file.write(_content.encode('utf-8'))
tmp_file.flush()
tmp_file.close() # 关闭文件句柄
try:
# 上传文件
item = storagechain.upload_file(fileitem=_fileitem, path=tmp_file_path, new_name=_path.name)
if item:
@@ -497,16 +496,15 @@ class MediaChain(ChainBase):
with request_utils.get_stream(url=_url) as r:
if r and r.status_code == 200:
# 使用tempfile创建临时文件，手动删除
tmp_file = NamedTemporaryFile(delete=False, suffix=_path.suffix)
tmp_file_path = Path(tmp_file.name)
try:
with NamedTemporaryFile(delete=False, suffix=_path.suffix) as tmp_file:
tmp_file_path = Path(tmp_file.name)
# 流式写入文件
for chunk in r.iter_content(chunk_size=8192):
if chunk:
tmp_file.write(chunk)
tmp_file.flush()
tmp_file.close() # 关闭文件句柄
try:
# 上传文件
item = storagechain.upload_file(fileitem=_fileitem, path=tmp_file_path,
new_name=_path.name)