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)

    def statistic(self, days: int = 7):
        """
        统计最近days天的下载历史数量
        """
        return TransferHistory.statistic(self._db, days)