|
| 1 | +# -*- encoding: utf-8 -*- |
| 2 | +""" |
| 3 | +Copyright (c) 2019 - present AppSeed.us |
| 4 | +""" |
| 5 | + |
| 6 | +from flask_login import UserMixin |
| 7 | + |
| 8 | +from apps import db, login_manager |
| 9 | + |
| 10 | +from apps.authentication.util import hash_pass |
| 11 | + |
| 12 | +class Users(db.Model, UserMixin): |
| 13 | + |
| 14 | + __tablename__ = 'Users' |
| 15 | + |
| 16 | + id = db.Column(db.Integer, primary_key=True) |
| 17 | + username = db.Column(db.String(64), unique=True) |
| 18 | + email = db.Column(db.String(64), unique=True) |
| 19 | + password = db.Column(db.LargeBinary) |
| 20 | + |
| 21 | + def __init__(self, **kwargs): |
| 22 | + for property, value in kwargs.items(): |
| 23 | + # depending on whether value is an iterable or not, we must |
| 24 | + # unpack it's value (when **kwargs is request.form, some values |
| 25 | + # will be a 1-element list) |
| 26 | + if hasattr(value, '__iter__') and not isinstance(value, str): |
| 27 | + # the ,= unpack of a singleton fails PEP8 (travis flake8 test) |
| 28 | + value = value[0] |
| 29 | + |
| 30 | + if property == 'password': |
| 31 | + value = hash_pass(value) # we need bytes here (not plain str) |
| 32 | + |
| 33 | + setattr(self, property, value) |
| 34 | + |
| 35 | + def __repr__(self): |
| 36 | + return str(self.username) |
| 37 | + |
| 38 | + |
| 39 | +@login_manager.user_loader |
| 40 | +def user_loader(id): |
| 41 | + return Users.query.filter_by(id=id).first() |
| 42 | + |
| 43 | + |
| 44 | +@login_manager.request_loader |
| 45 | +def request_loader(request): |
| 46 | + username = request.form.get('username') |
| 47 | + user = Users.query.filter_by(username=username).first() |
| 48 | + return user if user else None |
0 commit comments