init

app/db/models/subscribe.py

@@ -0,0 +1,36 @@
+from sqlalchemy import Column, Integer, String, Sequence
+from sqlalchemy.orm import Session
+
+from app.db.models import Base
+
+
+class Subscribe(Base):
+    id = Column(Integer, Sequence('id'), primary_key=True, index=True)
+    name = Column(String, nullable=False, index=True)
+    year = Column(String)
+    type = Column(String)
+    keyword = Column(String)
+    tmdbid = Column(String, index=True)
+    doubanid = Column(String)
+    season = Column(Integer)
+    image = Column(String)
+    description = Column(String)
+    filter = Column(String)
+    include = Column(String)
+    exclude = Column(String)
+    total_episode = Column(Integer)
+    start_episode = Column(Integer)
+    lack_episode = Column(Integer)
+    note = Column(String)
+    state = Column(String, nullable=False, index=True, default='N')
+
+    @staticmethod
+    def exists(db: Session, tmdbid: str, season: int = None):
+        if season:
+            return db.query(Subscribe).filter(Subscribe.tmdbid == tmdbid,
+                                              Subscribe.season == season).first()
+        return db.query(Subscribe).filter(Subscribe.tmdbid == tmdbid).first()
+
+    @staticmethod
+    def get_by_state(db: Session, state: str):
+        return db.query(Subscribe).filter(Subscribe.state == state).all()