fix #907

2024-01-08 13:05:48 +08:00
parent c674e32046
commit 4dcefb141a
4 changed files with 48 additions and 14 deletions
--- a/app/db/models/mediaserver.py
+++ b/app/db/models/mediaserver.py
@ -66,3 +66,11 @@ class MediaServerItem(Base):
        return db.query(MediaServerItem).filter(MediaServerItem.title == title,
                                                MediaServerItem.item_type == mtype,
                                                MediaServerItem.year == str(year)).first()
+
+    @staticmethod
+    @db_query
+    def list_by_type(db: Session, mtype: str = None):
+        if mtype:
+            return db.query(MediaServerItem).filter(MediaServerItem.item_type == mtype).all()
+        else:
+            return db.query(MediaServerItem).all()