diff --git a/app/db/models/transferhistory.py b/app/db/models/transferhistory.py index 5868e8e9..e001431d 100644 --- a/app/db/models/transferhistory.py +++ b/app/db/models/transferhistory.py @@ -89,6 +89,11 @@ class TransferHistory(Base): def get_by_src(db: Session, src: str): return db.query(TransferHistory).filter(TransferHistory.src == src).first() + @staticmethod + @db_query + def get_by_dest(db: Session, dest: str): + return db.query(TransferHistory).filter(TransferHistory.dest == dest).first() + @staticmethod @db_query def list_by_hash(db: Session, download_hash: str): diff --git a/app/db/transferhistory_oper.py b/app/db/transferhistory_oper.py index eedac77a..add5be05 100644 --- a/app/db/transferhistory_oper.py +++ b/app/db/transferhistory_oper.py @@ -36,6 +36,13 @@ class TransferHistoryOper(DbOper): """ return TransferHistory.get_by_src(self._db, src) + def get_by_dest(self, dest: str) -> TransferHistory: + """ + 按转移路径查询转移记录 + :param dest: 数据key + """ + return TransferHistory.get_by_dest(self._db, dest) + def list_by_hash(self, download_hash: str) -> List[TransferHistory]: """ 按种子hash查询转移记录