fix bug

2023-12-23 19:51:35 +08:00
parent dcf1985361
commit b3fe49684b
2 changed files with 6 additions and 3 deletions
--- a/app/db/models/transferhistory.py
+++ b/app/db/models/transferhistory.py
@ -103,8 +103,11 @@ class TransferHistory(Base):

    @staticmethod
    @db_query
-    def count(db: Session):
-        return db.query(func.count(TransferHistory.id)).first()[0]
+    def count(db: Session, status: bool = None):
+        if status is not None:
+            return db.query(func.count(TransferHistory.id)).filter(TransferHistory.status == status).first()[0]
+        else:
+            return db.query(func.count(TransferHistory.id)).first()[0]

    @staticmethod
    @db_query