GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-07 08:26:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix file upload

Browse Source
This commit is contained in:
jxxghp
2024-09-20 13:09:25 +08:00
parent 117bd80528
commit 8ac3fd46d2
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/chain/media.py
+10 -5
View File
@@ -364,11 +364,16 @@ class MediaChain(ChainBase, metaclass=Singleton):
""" """
tmp_file = settings.TEMP_PATH / _path.name tmp_file = settings.TEMP_PATH / _path.name
tmp_file.write_bytes(_content) tmp_file.write_bytes(_content)
StorageChain().upload_file(fileitem=_fileitem, path=tmp_file) upload_item = copy.deepcopy(_fileitem)
upload_item.path = str(_path)
upload_item.name = _path.name
upload_item.basename = _path.stem
upload_item.extension = _path.suffix
StorageChain().upload_file(fileitem=upload_item, path=tmp_file)
if tmp_file.exists(): if tmp_file.exists():
tmp_file.unlink() tmp_file.unlink()
def __save_image(_url: str) -> Optional[bytes]: def __download_image(_url: str) -> Optional[bytes]:
""" """
下载图片并保存 下载图片并保存
""" """
@@ -425,7 +430,7 @@ class MediaChain(ChainBase, metaclass=Singleton):
image_name = attr_name.replace("_path", "") + Path(attr_value).suffix image_name = attr_name.replace("_path", "") + Path(attr_value).suffix
image_path = filepath / image_name image_path = filepath / image_name
# 下载图片 # 下载图片
content = __save_image(_url=attr_value) content = __download_image(_url=attr_value)
# 写入nfo到根目录 # 写入nfo到根目录
__save_file(_fileitem=fileitem, _path=image_path, _content=content) __save_file(_fileitem=fileitem, _path=image_path, _content=content)
else: else:
@@ -474,7 +479,7 @@ class MediaChain(ChainBase, metaclass=Singleton):
for image_name, image_url in image_dict.items(): for image_name, image_url in image_dict.items():
image_path = filepath.with_name(image_name) image_path = filepath.with_name(image_name)
# 下载图片 # 下载图片
content = __save_image(image_url) content = __download_image(image_url)
# 保存图片文件到当前目录 # 保存图片文件到当前目录
__save_file(_fileitem=fileitem, _path=image_path, _content=content) __save_file(_fileitem=fileitem, _path=image_path, _content=content)
if season_meta.name: if season_meta.name:
@@ -492,7 +497,7 @@ class MediaChain(ChainBase, metaclass=Singleton):
for image_name, image_url in image_dict.items(): for image_name, image_url in image_dict.items():
image_path = filepath.parent.with_name(image_name) image_path = filepath.parent.with_name(image_name)
# 下载图片 # 下载图片
content = __save_image(image_url) content = __download_image(image_url)
# 保存图片文件到当前目录 # 保存图片文件到当前目录
__save_file(_fileitem=fileitem, _path=image_path, _content=content) __save_file(_fileitem=fileitem, _path=image_path, _content=content)
app/modules/filemanager/storages/local.py
+2 -1
View File
@@ -187,7 +187,8 @@ class LocalStorage(StorageBase):
""" """
上传文件 上传文件
""" """
pass file_path = path.rename(fileitem.path)
return self.__get_diritem(file_path)
def copy(self, fileitem: schemas.FileItem, target_file: Path) -> bool: def copy(self, fileitem: schemas.FileItem, target_file: Path) -> bool:
""" """