codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix：优化文件管理api

Browse Source
This commit is contained in:
jxxghp
2024-06-20 11:38:57 +08:00
parent 0fb12c77eb
commit 185b72dc8d
3 changed files with 78 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
app/api/endpoints/aliyun.py
View File

@ -56,29 +56,27 @@ def userinfo(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
return schemas.Response(success=False)
@router.get("/list", summary="所有目录和文件（阿里云盘）", response_model=List[schemas.FileItem])
def list_aliyun(path: str,
fileid: str,
filetype: str = "dir",
@router.post("/list", summary="所有目录和文件（阿里云盘）", response_model=List[schemas.FileItem])
def list_aliyun(fileitem: schemas.FileItem,
sort: str = 'updated_at',
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
查询当前目录下所有目录和文件
:param path: 当前路径
:param fileid: 文件ID
:param filetype: 文件类型
:param fileitem: 文件夹信息
:param sort: 排序方式，name:按名称排序，time:按修改时间排序
:param _: token
:return: 所有目录和文件
"""
if not fileid:
if not fileitem.fileid:
return []
if not path:
if not fileitem.path:
path = "/"
else:
path = fileitem.path
if sort == "time":
sort = "updated_at"
if filetype == "file":
fileinfo = AliyunHelper().get_file_detail(fileid)
if fileitem.type == "file":
fileinfo = AliyunHelper().get_file_detail(fileitem.fileid)
if fileinfo:
return [schemas.FileItem(
fileid=fileinfo.get("file_id"),
@ -92,7 +90,7 @@ def list_aliyun(path: str,
thumbnail=fileinfo.get("thumbnail")
)]
return []
items = AliyunHelper().list_files(parent_file_id=fileid, order_by=sort)
items = AliyunHelper().list_files(parent_file_id=fileitem.fileid, order_by=sort)
if not items:
return []
return [schemas.FileItem(
@ -108,30 +106,30 @@ def list_aliyun(path: str,
) for item in items]
@router.get("/mkdir", summary="创建目录（阿里云盘）", response_model=schemas.Response)
def mkdir_aliyun(fileid: str,
@router.post("/mkdir", summary="创建目录（阿里云盘）", response_model=schemas.Response)
def mkdir_aliyun(fileitem: schemas.FileItem,
name: str,
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
创建目录
"""
if not fileid or not name:
if not fileitem.fileid or not name:
return schemas.Response(success=False)
result = AliyunHelper().create_folder(parent_file_id=fileid, name=name)
result = AliyunHelper().create_folder(parent_file_id=fileitem.fileid, name=name)
if result:
return schemas.Response(success=True)
return schemas.Response(success=False)
@router.get("/delete", summary="删除文件或目录（阿里云盘）", response_model=schemas.Response)
def delete_aliyun(fileid: str,
@router.post("/delete", summary="删除文件或目录（阿里云盘）", response_model=schemas.Response)
def delete_aliyun(fileitem: schemas.FileItem,
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
删除文件或目录
"""
if not fileid:
if not fileitem.fileid:
return schemas.Response(success=False)
result = AliyunHelper().delete_file(fileid)
result = AliyunHelper().delete_file(fileitem.fileid)
if result:
return schemas.Response(success=True)
return schemas.Response(success=False)
@ -152,22 +150,23 @@ def download_aliyun(fileid: str,
raise HTTPException(status_code=500, detail="下载文件出错")
@router.get("/rename", summary="重命名文件或目录（阿里云盘）", response_model=schemas.Response)
def rename_aliyun(fileid: str, new_name: str, path: str,
@router.post("/rename", summary="重命名文件或目录（阿里云盘）", response_model=schemas.Response)
def rename_aliyun(fileitem: schemas.FileItem,
new_name: str,
recursive: bool = False,
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
重命名文件或目录
"""
if not fileid or not new_name:
if not fileitem.fileid or not new_name:
return schemas.Response(success=False)
result = AliyunHelper().rename_file(fileid, new_name)
result = AliyunHelper().rename_file(fileitem.fileid, new_name)
if result:
if recursive:
transferchain = TransferChain()
media_exts = settings.RMT_MEDIAEXT + settings.RMT_SUBEXT + settings.RMT_AUDIO_TRACK_EXT
# 递归修改目录内文件（智能识别命名）
sub_files: List[schemas.FileItem] = list_aliyun(path=path, fileid=fileid)
sub_files: List[schemas.FileItem] = list_aliyun(fileitem=fileitem)
if sub_files:
# 开始进度
progress = ProgressHelper()
@ -185,7 +184,7 @@ def rename_aliyun(fileid: str, new_name: str, path: str,
continue
if f".{sub_file.extension.lower()}" not in media_exts:
continue
sub_path = Path(f"{path}{sub_file.name}")
sub_path = Path(f"{fileitem.path}{sub_file.name}")
meta = MetaInfoPath(sub_path)
mediainfo = transferchain.recognize_media(meta)
if not mediainfo:
@ -195,8 +194,7 @@ def rename_aliyun(fileid: str, new_name: str, path: str,
if not new_path:
progress.end(ProgressKey.BatchRename)
return schemas.Response(success=False, message=f"{sub_path.name} 未识别到新名称")
ret: schemas.Response = rename_aliyun(fileid=sub_file.fileid,
path=path,
ret: schemas.Response = rename_aliyun(fileitem=sub_file,
new_name=Path(new_path).name,
recursive=False)
if not ret.success: