add upload api

This commit is contained in:
jxxghp
2024-06-21 08:08:23 +08:00
parent ae0e171dd2
commit 0133c6e60c
5 changed files with 162 additions and 69 deletions

View File

@ -47,10 +47,8 @@ def userinfo(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
查询用户信息
"""
aliyunhelper = AliyunHelper()
# 浏览一次文件确定token正确性
aliyunhelper.list_files()
# 查询用户信息返回
info = aliyunhelper.get_user_info()
info = aliyunhelper.user_info()
if info:
return schemas.Response(success=True, data=info)
return schemas.Response(success=False)
@ -76,7 +74,7 @@ def list_aliyun(fileitem: schemas.FileItem,
if sort == "time":
sort = "updated_at"
if fileitem.type == "file":
fileinfo = AliyunHelper().get_file_detail(fileitem.fileid)
fileinfo = AliyunHelper().detail(fileitem.fileid)
if fileinfo:
return [schemas.FileItem(
fileid=fileinfo.get("file_id"),
@ -91,7 +89,7 @@ def list_aliyun(fileitem: schemas.FileItem,
drive_id=fileinfo.get("drive_id"),
)]
return []
items = AliyunHelper().list_files(drive_id=fileitem.drive_id, parent_file_id=fileitem.fileid, order_by=sort)
items = AliyunHelper().list(drive_id=fileitem.drive_id, parent_file_id=fileitem.fileid, order_by=sort)
if not items:
return []
return [schemas.FileItem(
@ -131,7 +129,7 @@ def delete_aliyun(fileitem: schemas.FileItem,
"""
if not fileitem.fileid:
return schemas.Response(success=False)
result = AliyunHelper().delete_file(fileitem.fileid)
result = AliyunHelper().delete(fileitem.fileid)
if result:
return schemas.Response(success=True)
return schemas.Response(success=False)
@ -145,7 +143,7 @@ def download_aliyun(fileid: str,
"""
if not fileid:
return schemas.Response(success=False)
url = AliyunHelper().get_download_url(fileid)
url = AliyunHelper().download(fileid)
if url:
# 重定向
return Response(status_code=302, headers={"Location": url})
@ -162,7 +160,7 @@ def rename_aliyun(fileitem: schemas.FileItem,
"""
if not fileitem.fileid or not new_name:
return schemas.Response(success=False)
result = AliyunHelper().rename_file(fileitem.fileid, new_name)
result = AliyunHelper().rename(fileitem.fileid, new_name)
if result:
if recursive:
transferchain = TransferChain()
@ -214,7 +212,7 @@ def image_aliyun(fileid: str, _: schemas.TokenPayload = Depends(verify_uri_token
"""
if not fileid:
return schemas.Response(success=False)
url = AliyunHelper().get_download_url(fileid)
url = AliyunHelper().download_url(fileid)
if url:
# 重定向
return Response(status_code=302, headers={"Location": url})

View File

@ -46,7 +46,7 @@ def storage(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
"""
查询存储空间信息
"""
storage_info = U115Helper().get_storage()
storage_info = U115Helper().storage()
if storage_info:
return schemas.Response(success=True, data={
"total": storage_info[0],
@ -87,7 +87,7 @@ def list_115(fileitem: schemas.FileItem,
extension=suffix,
pickcode=fileitem.pickcode
)]
items = U115Helper().list_files(parent_file_id=fileid)
items = U115Helper().list(parent_file_id=fileid)
if not items:
return []
file_list = [schemas.FileItem(
@ -131,7 +131,7 @@ def delete_115(fileitem: schemas.FileItem,
"""
if not fileitem.fileid:
return schemas.Response(success=False)
result = U115Helper().delete_file(fileitem.fileid)
result = U115Helper().delete(fileitem.fileid)
if result:
return schemas.Response(success=True)
return schemas.Response(success=False)
@ -164,7 +164,7 @@ def rename_115(fileitem: schemas.FileItem,
"""
if not fileitem.fileid or not new_name:
return schemas.Response(success=False)
result = U115Helper().rename_file(fileitem.fileid, new_name)
result = U115Helper().rename(fileitem.fileid, new_name)
if result:
if recursive:
transferchain = TransferChain()