MoviePilot/app/db/transferhistory_oper.py
2023-06-28 12:49:38 +08:00

32 lines
885 B
Python

import time
from typing import Any
from app.db import DbOper
from app.db.models.transferhistory import TransferHistory
class TransferHistoryOper(DbOper):
"""
转移历史管理
"""
def get_by_title(self, title: str) -> Any:
"""
按标题查询转移记录
:param title: 数据key
"""
return TransferHistory.search_by_title(self._db, title)
def add(self, **kwargs):
"""
新增转移历史
"""
if kwargs.get("download_hash"):
transferhistory = TransferHistory.get_by_hash(self._db, kwargs.get("download_hash"))
if transferhistory:
transferhistory.delete(self._db, transferhistory.id)
kwargs.update({
"date": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
})
return TransferHistory(**kwargs).create(self._db)