69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
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(Integer, index=True)
|
||
imdbid = Column(String)
|
||
tvdbid = Column(Integer, 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)
|
||
# 状态：N-新建， R-订阅中
|
||
state = Column(String, nullable=False, index=True, default='N')
|
||
|
||
@staticmethod
|
||
def exists(db: Session, tmdbid: int, 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()
|
||
|
||
@staticmethod
|
||
def get_by_tmdbid(db: Session, tmdbid: int):
|
||
return db.query(Subscribe).filter(Subscribe.tmdbid == tmdbid).all()
|
||
|
||
@staticmethod
|
||
def get_by_title(db: Session, title: str):
|
||
return db.query(Subscribe).filter(Subscribe.name == title).first()
|
||
|
||
@staticmethod
|
||
def get_by_tvdbid(db: Session, tvdbid: int):
|
||
return db.query(Subscribe).filter(Subscribe.tvdbid == tvdbid).first()
|