from sqlalchemy import Column, Integer, String, Sequence, Float
from sqlalchemy.orm import Session

from app.db import db_query, Base


class UserRequest(Base):
    """
    用户请求表
    """
    # ID
    id = Column(Integer, Sequence('id'), primary_key=True, index=True)
    # 申请用户
    req_user = Column(String, index=True, nullable=False)
    # 申请时间
    req_time = Column(String)
    # 申请备注
    req_remark = Column(String)
    # 审批用户
    app_user = Column(String, index=True, nullable=False)
    # 审批时间
    app_time = Column(String)
    # 审批状态 0-待审批 1-通过 2-拒绝
    app_status = Column(Integer, default=0)
    # 类型
    type = Column(String)
    # 标题
    title = Column(String)
    # 年份
    year = Column(String)
    # 媒体ID
    tmdbid = Column(Integer)
    imdbid = Column(String)
    tvdbid = Column(Integer)
    doubanid = Column(String)
    bangumiid = Column(Integer)
    # 季号
    season = Column(Integer)
    # 海报
    poster = Column(String)
    # 背景图
    backdrop = Column(String)
    # 评分，float
    vote = Column(Float)
    # 简介
    description = Column(String)

    @staticmethod
    @db_query
    def get_by_req_user(db: Session, req_user: str, status: int = None):
        if status:
            return db.query(UserRequest).filter(UserRequest.req_user == req_user,
                                                UserRequest.app_status == status).all()
        else:
            return db.query(UserRequest).filter(UserRequest.req_user == req_user).all()

    @staticmethod
    @db_query
    def get_by_app_user(db: Session, app_user: str, status: int = None):
        if status:
            return db.query(UserRequest).filter(UserRequest.app_user == app_user,
                                                UserRequest.app_status == status).all()
        else:
            return db.query(UserRequest).filter(UserRequest.app_user == app_user).all()

    @staticmethod
    @db_query
    def get_by_status(db: Session, status: int):
        return db.query(UserRequest).filter(UserRequest.app_status == status).all()