from sqlalchemy.ext.hybrid import hybrid_property from flask_login import UserMixin from app import bcrypt, db class User(db.Model, UserMixin): __tablename__ = 'users' first_name = db.Column(db.String) last_name = db.Column(db.String) email = db.Column(db.String(120), primary_key=True) confirmation = db.Column(db.Boolean) paid = db.Column(db.Boolean) role = db.Column(db.String) team = db.Column(db.String) _password = db.Column(db.String) @property def full_name(self): return '{} {}'.format(self.first_name, self.last_name) @hybrid_property def password(self): return self._password @password.setter def password(self, plaintext): self._password = bcrypt.generate_password_hash(plaintext) def check_password(self, plaintext): return bcrypt.check_password_hash(self.password, plaintext) def get_id(self): return self.email def get_role(self): return self.role def get_team(self): return self.team def is_paid(self): return self.paid