GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-05-11 18:10:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix umask

Browse Source
This commit is contained in:
景大侠
2025-09-15 22:01:00 +08:00
parent f8c682b183
commit 14961323c3
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/chain/media.py
View File

@@ -22,6 +22,8 @@ from app.utils.string import StringUtils
recognize_lock = Lock()
scraping_lock = Lock()
current_umask = os.umask(0)
os.umask(current_umask)
class MediaChain(ChainBase):
"""
@@ -459,7 +461,7 @@ class MediaChain(ChainBase):
if not _fileitem or not _content or not _path:
return
# 使用tempfile创建临时文件，自动删除
with NamedTemporaryFile(delete_on_close=False, suffix=_path.suffix) as tmp_file:
with NamedTemporaryFile(delete=True, delete_on_close=False, suffix=_path.suffix) as tmp_file:
tmp_file_path = Path(tmp_file.name)
# 写入内容
if isinstance(_content, bytes):
@@ -469,11 +471,9 @@ class MediaChain(ChainBase):
tmp_file.flush()
tmp_file.close() # 关闭文件句柄
current_umask = os.umask(0)
os.umask(current_umask)
# 刮削文件只需要读写权限
tmp_file_path.chmod(0o666 & ~current_umask)
# 上传文件
item = storagechain.upload_file(fileitem=_fileitem, path=tmp_file_path, new_name=_path.name)
if item:
@@ -496,7 +496,7 @@ class MediaChain(ChainBase):
with request_utils.get_stream(url=_url) as r:
if r and r.status_code == 200:
# 使用tempfile创建临时文件，自动删除
with NamedTemporaryFile(delete_on_close=False, suffix=_path.suffix) as tmp_file:
with NamedTemporaryFile(delete=True, delete_on_close=False, suffix=_path.suffix) as tmp_file:
tmp_file_path = Path(tmp_file.name)
# 流式写入文件
for chunk in r.iter_content(chunk_size=8192):
@@ -504,12 +504,10 @@ class MediaChain(ChainBase):
tmp_file.write(chunk)
tmp_file.flush()
tmp_file.close() # 关闭文件句柄
current_umask = os.umask(0)
os.umask(current_umask)
# 刮削的图片只需要读写权限
tmp_file_path.chmod(0o666 & ~current_umask)
# 上传文件
item = storagechain.upload_file(fileitem=_fileitem, path=tmp_file_path,
new_name=_path.name)