| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- from __future__ import print_function
- from flask_security import UserMixin, RoleMixin
- from flask_security.utils import encrypt_password
- from project import db
- roles_users = db.Table('roles_users',
- db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
- db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))
- class Role(RoleMixin, db.Model):
- """Based on Flask-Security quickstart Role"""
- id = db.Column(db.Integer(), primary_key=True)
- name = db.Column(db.String(255), unique=True)
- description = db.Column(db.String(255), default='')
- @classmethod
- def createrole(cls, session, name, description=None):
- """Creates a new role instance."""
- o = cls(
- name=name,
- description=description
- )
- session.add(o)
- return o
- def __repr__(self):
- return u"{0} ({1})".format(self.name, self.description or 'Role')
- class User(UserMixin, db.Model):
- """Based on Flask-Security quickstart User"""
- id = db.Column(db.Integer, primary_key=True)
- email = db.Column(db.String(255), unique=True)
- password = db.Column(db.String(255), default='')
- roles = db.relationship('Role', secondary=roles_users,
- backref=db.backref('users', lazy='dynamic'))
- def set_password(self, session, password):
- """Encrypts a plain-text user password."""
- self.password = encrypt_password(password)
- session.add(self)
- @classmethod
- def createuser(cls, session, email, password, roles=None):
- """Creates a new user instance."""
- o = cls(
- email=email,
- password=encrypt_password(password),
- roles=roles,
- )
- session.add(o)
- return o
- def __repr__(self):
- return self.email
- @property
- def active(self):
- return True
- @property
- def has_admin_privs(self):
- """Determines if this user has admin privileges."""
- return self.active and self.has_role('admin')
|
