From 08a2b348d8fa90e09c5a7bb28b00c4e181fcce3b Mon Sep 17 00:00:00 2001
From: thsrite <thsrite@gmail.com>
Date: Sun, 23 Jun 2024 19:45:08 +0800
Subject: [PATCH] add get_by_dest

---
 app/db/models/transferhistory.py | 5 +++++
 app/db/transferhistory_oper.py   | 7 +++++++
 2 files changed, 12 insertions(+)

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查询转移记录