示例#1
0
文件: models.py 项目: Rdbaker/Rank
class User(UserMixin, CRUDMixin, SurrogatePK, db.Model):
    __tablename__ = 'users'
    username = db.Column(db.String(80), unique=True, nullable=False)
    #: The hashed password
    password = db.Column(db.String(128), nullable=True)
    created_at = db.Column(db.DateTime, nullable=False, default=dt.datetime.utcnow)
    active = db.Column(db.Boolean(), default=False)
    is_admin = db.Column(db.Boolean(), default=False, nullable=False)
    game_id = db.Column(db.Integer, db.ForeignKey("games.id"))
    game = relationship("Game", backref=backref("user", uselist=False), cascade="delete")

    def __init__(self, username, password=None, **kwargs):
        db.Model.__init__(self, username=username, **kwargs)
        if password:
            self.set_password(password)
        else:
            self.password = None

    def set_password(self, password):
        self.password = bcrypt.generate_password_hash(password)

    def check_password(self, value):
        return bcrypt.check_password_hash(self.password, value)

    def requests_today(self):
        midnight = dt.datetime.combine(dt.date.today(), dt.time())
        if not self.is_admin:
            return UserRequest.query.filter(
                UserRequest.game_id == self.game.id,
                UserRequest.time_requested > midnight).order_by(UserRequest.time_requested.desc())
        else:
            return UserRequest.query.filter(
                UserRequest.time_requested > midnight).order_by(UserRequest.time_requested.desc())

    def requests_this_week(self):
        midnight = dt.datetime.combine(dt.date.today(), dt.time())
        seven_days_ago = midnight - dt.timedelta(days=7)
        if not self.is_admin:
            return UserRequest.query.filter(
                UserRequest.game_id == self.game.id,
                UserRequest.time_requested > seven_days_ago).order_by(UserRequest.time_requested.desc())
        else:
            return UserRequest.query.filter(
                UserRequest.time_requested > seven_days_ago).order_by(UserRequest.time_requested.desc())

    def request_count_today(self):
        midnight = dt.datetime.combine(dt.date.today(), dt.time())
        if not self.is_admin:
            return UserRequest.query.filter(
                UserRequest.game_id == self.game.id,
                UserRequest.time_requested > midnight).count()
        else:
            return UserRequest.query.filter(
                UserRequest.time_requested > midnight).count()

    def __repr__(self):
        return '<User({username!r})>'.format(username=self.username)
示例#2
0
文件: models.py 项目: Rdbaker/Rank
class Role(SurrogatePK, db.Model):
    __tablename__ = 'roles'
    name = db.Column(db.String(80), nullable=False)
    user_id = ReferenceCol('users', nullable=True)
    user = db.relationship('User', backref='roles')

    def __init__(self, name, **kwargs):
        db.Model.__init__(self, name=name, **kwargs)

    def __repr__(self):
        return '<Role({name})>'.format(name=self.name)