mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 20:17:22 +08:00
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from pathlib import Path
|
||
|
||
from app.actions import BaseAction
|
||
from app.schemas import ActionParams, ActionContext
|
||
from app.chain.storage import StorageChain
|
||
from app.chain.transfer import TransferChain
|
||
from app.log import logger
|
||
|
||
|
||
class TransferFileParams(ActionParams):
|
||
"""
|
||
整理文件参数
|
||
"""
|
||
pass
|
||
|
||
|
||
class TransferFileAction(BaseAction):
|
||
"""
|
||
整理文件
|
||
"""
|
||
|
||
__fileitems = []
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.transferchain = TransferChain()
|
||
self.storagechain = StorageChain()
|
||
|
||
@property
|
||
def name(self) -> str:
|
||
return "整理文件"
|
||
|
||
@property
|
||
def description(self) -> str:
|
||
return "转移和重命名文件"
|
||
|
||
@property
|
||
def data(self) -> dict:
|
||
return TransferFileParams().dict()
|
||
|
||
@property
|
||
def success(self) -> bool:
|
||
return True if self.__fileitems else False
|
||
|
||
async def execute(self, params: TransferFileParams, context: ActionContext) -> ActionContext:
|
||
"""
|
||
从downloads中整理文件，记录到fileitems
|
||
"""
|
||
for download in context.downloads:
|
||
if not download.completed:
|
||
logger.info(f"下载任务 {download.download_id} 未完成")
|
||
continue
|
||
fileitem = self.storagechain.get_file_item(storage="local", path=Path(download.path))
|
||
if not fileitem:
|
||
logger.info(f"文件 {download.path} 不存在")
|
||
continue
|
||
logger.info(f"开始整理文件 {download.path} ...")
|
||
state, errmsg = self.transferchain.do_transfer(fileitem, background=False)
|
||
if not state:
|
||
logger.error(f"整理文件 {download.path} 失败: {errmsg}")
|
||
continue
|
||
logger.info(f"整理文件 {download.path} 完成")
|
||
self.__fileitems.append(fileitem)
|
||
|
||
if self.__fileitems:
|
||
context.fileitems.extend(self.__fileitems)
|
||
|
||
self.job_done()
|
||
return context
|