pytest-bdd-example/pytest_bdd_example/dashboard/models.py

17 lines
522 B
Python

from flask.ext.security import UserMixin
from pytest_bdd_example.dashboard import db
ROLE_USER = 0
ROLE_ADMIN = 1
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), unique=True)
email = db.Column(db.String(120), unique=True)
password = db.Column(db.String(20))
role = db.Column(db.SmallInteger, default=ROLE_USER)
is_active = db.Column(db.Boolean, default=False)
def __repr__(self):
return '<User %r>' % (self.username)