mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-30 04:27:40 +08:00
214 lines
8.1 KiB
Python
214 lines
8.1 KiB
Python
from pathlib import Path
|
|
from typing import Any, List
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from starlette.responses import Response
|
|
|
|
from app import schemas
|
|
from app.chain.transfer import TransferChain
|
|
from app.core.config import settings
|
|
from app.core.metainfo import MetaInfoPath
|
|
from app.core.security import verify_token, verify_uri_token
|
|
from app.helper.progress import ProgressHelper
|
|
from app.helper.u115 import U115Helper
|
|
from app.schemas.types import ProgressKey
|
|
from app.utils.http import RequestUtils
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/qrcode", summary="生成二维码内容", response_model=schemas.Response)
|
|
def qrcode(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
生成二维码
|
|
"""
|
|
qrcode_data = U115Helper().generate_qrcode()
|
|
if qrcode_data:
|
|
return schemas.Response(success=True, data={
|
|
'codeContent': qrcode_data
|
|
})
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.get("/check", summary="二维码登录确认", response_model=schemas.Response)
|
|
def check(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
二维码登录确认
|
|
"""
|
|
data, errmsg = U115Helper().check_login()
|
|
if data:
|
|
return schemas.Response(success=True, data=data)
|
|
return schemas.Response(success=False, message=errmsg)
|
|
|
|
|
|
@router.get("/storage", summary="查询存储空间信息", response_model=schemas.Response)
|
|
def storage(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
查询存储空间信息
|
|
"""
|
|
storage_info = U115Helper().storage()
|
|
if storage_info:
|
|
return schemas.Response(success=True, data={
|
|
"total": storage_info[0],
|
|
"used": storage_info[1]
|
|
})
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.post("/list", summary="所有目录和文件（115网盘）", response_model=List[schemas.FileItem])
|
|
def list_115(fileitem: schemas.FileItem,
|
|
sort: str = 'updated_at',
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
查询当前目录下所有目录和文件
|
|
:param fileitem: 文件项
|
|
:param sort: 排序方式，name:按名称排序，time:按修改时间排序
|
|
:param _: token
|
|
:return: 所有目录和文件
|
|
"""
|
|
if not fileitem.fileid:
|
|
return []
|
|
if not fileitem.path:
|
|
path = "/"
|
|
else:
|
|
path = fileitem.path
|
|
if fileitem.fileid == "root":
|
|
fileid = "0"
|
|
else:
|
|
fileid = fileitem.fileid
|
|
if fileitem.type == "file":
|
|
name = Path(path).name
|
|
suffix = Path(name).suffix[1:]
|
|
return [schemas.FileItem(
|
|
fileid=fileid,
|
|
type="file",
|
|
path=path.rstrip('/'),
|
|
name=name,
|
|
extension=suffix,
|
|
pickcode=fileitem.pickcode
|
|
)]
|
|
file_list = U115Helper().list(parent_file_id=fileid, path=path)
|
|
if sort == "name":
|
|
file_list.sort(key=lambda x: x.name)
|
|
else:
|
|
file_list.sort(key=lambda x: x.modify_time, reverse=True)
|
|
return file_list
|
|
|
|
|
|
@router.post("/mkdir", summary="创建目录（115网盘）", response_model=schemas.Response)
|
|
def mkdir_115(fileitem: schemas.FileItem,
|
|
name: str,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
创建目录
|
|
"""
|
|
if not fileitem.fileid or not name:
|
|
return schemas.Response(success=False)
|
|
result = U115Helper().create_folder(parent_file_id=fileitem.fileid, name=name, path=fileitem.path)
|
|
if result:
|
|
return schemas.Response(success=True)
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.post("/delete", summary="删除文件或目录（115网盘）", response_model=schemas.Response)
|
|
def delete_115(fileitem: schemas.FileItem,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
删除文件或目录
|
|
"""
|
|
if not fileitem.fileid:
|
|
return schemas.Response(success=False)
|
|
result = U115Helper().delete(fileitem.fileid)
|
|
if result:
|
|
return schemas.Response(success=True)
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.get("/download", summary="下载文件（115网盘）")
|
|
def download_115(pickcode: str,
|
|
_: schemas.TokenPayload = Depends(verify_uri_token)) -> Any:
|
|
"""
|
|
下载文件或目录
|
|
"""
|
|
if not pickcode:
|
|
return schemas.Response(success=False)
|
|
ticket = U115Helper().download(pickcode)
|
|
if ticket:
|
|
# 请求数据，并以文件流的方式返回
|
|
res = RequestUtils(headers=ticket.headers).get_res(ticket.url)
|
|
if res:
|
|
return Response(content=res.content, media_type="application/octet-stream")
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.post("/rename", summary="重命名文件或目录（115网盘）", response_model=schemas.Response)
|
|
def rename_115(fileitem: schemas.FileItem,
|
|
new_name: str,
|
|
recursive: bool = False,
|
|
_: schemas.TokenPayload = Depends(verify_token)) -> Any:
|
|
"""
|
|
重命名文件或目录
|
|
"""
|
|
if not fileitem.fileid or not new_name:
|
|
return schemas.Response(success=False)
|
|
result = U115Helper().rename(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_115(fileitem)
|
|
if sub_files:
|
|
# 开始进度
|
|
progress = ProgressHelper()
|
|
progress.start(ProgressKey.BatchRename)
|
|
total = len(sub_files)
|
|
handled = 0
|
|
for sub_file in sub_files:
|
|
handled += 1
|
|
progress.update(value=handled / total * 100,
|
|
text=f"正在处理 {sub_file.name} ...",
|
|
key=ProgressKey.BatchRename)
|
|
if sub_file.type == "dir":
|
|
continue
|
|
if not sub_file.extension:
|
|
continue
|
|
if f".{sub_file.extension.lower()}" not in media_exts:
|
|
continue
|
|
sub_path = Path(f"{fileitem.path}{sub_file.name}")
|
|
meta = MetaInfoPath(sub_path)
|
|
mediainfo = transferchain.recognize_media(meta)
|
|
if not mediainfo:
|
|
progress.end(ProgressKey.BatchRename)
|
|
return schemas.Response(success=False, message=f"{sub_path.name} 未识别到媒体信息")
|
|
new_path = transferchain.recommend_name(meta=meta, mediainfo=mediainfo)
|
|
if not new_path:
|
|
progress.end(ProgressKey.BatchRename)
|
|
return schemas.Response(success=False, message=f"{sub_path.name} 未识别到新名称")
|
|
ret: schemas.Response = rename_115(fileitem=sub_file,
|
|
new_name=Path(new_path).name,
|
|
recursive=False)
|
|
if not ret.success:
|
|
progress.end(ProgressKey.BatchRename)
|
|
return schemas.Response(success=False, message=f"{sub_path.name} 重命名失败！")
|
|
progress.end(ProgressKey.BatchRename)
|
|
return schemas.Response(success=True)
|
|
return schemas.Response(success=False)
|
|
|
|
|
|
@router.get("/image", summary="读取图片（115网盘）")
|
|
def image_115(pickcode: str, _: schemas.TokenPayload = Depends(verify_uri_token)) -> Any:
|
|
"""
|
|
读取图片
|
|
"""
|
|
if not pickcode:
|
|
return schemas.Response(success=False)
|
|
ticket = U115Helper().download(pickcode)
|
|
if ticket:
|
|
# 请求数据，获取内容编码为图片base64返回
|
|
res = RequestUtils(headers=ticket.headers).get_res(ticket.url)
|
|
if res:
|
|
content_type = res.headers.get("Content-Type")
|
|
return Response(content=res.content, media_type=content_type)
|
|
raise HTTPException(status_code=500, detail="下载图片出错")
|