mirror of https://github.com/logos-co/open-law.git
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from uuid import uuid4
|
|
|
|
from flask_login import UserMixin, AnonymousUserMixin
|
|
from sqlalchemy import func
|
|
from sqlalchemy.ext.hybrid import hybrid_property
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
|
from app import db
|
|
from app.models.utils import BaseModel
|
|
from app.logger import log
|
|
from app import schema as s
|
|
|
|
|
|
def gen_password_reset_id() -> str:
|
|
return str(uuid4())
|
|
|
|
|
|
class User(BaseModel, UserMixin):
|
|
__tablename__ = "users"
|
|
|
|
username = db.Column(db.String(60), unique=True, nullable=False)
|
|
password_hash = db.Column(db.String(255), default="")
|
|
|
|
# Relationships
|
|
stars = db.relationship("Book", secondary="books_stars", back_populates="stars")
|
|
books = db.relationship("Book")
|
|
|
|
@hybrid_property
|
|
def password(self):
|
|
return self.password_hash
|
|
|
|
@password.setter
|
|
def password(self, password):
|
|
self.password_hash = generate_password_hash(password)
|
|
|
|
@classmethod
|
|
def authenticate(cls, user_id, password):
|
|
user = cls.query.filter(
|
|
func.lower(cls.username) == func.lower(user_id),
|
|
).first()
|
|
if not user:
|
|
log(log.WARNING, "user:[%s] not found", user_id)
|
|
|
|
if user is not None and check_password_hash(user.password, password):
|
|
return user
|
|
|
|
def __repr__(self):
|
|
return f"<{self.id}: {self.username}>"
|
|
|
|
@property
|
|
def json(self):
|
|
u = s.User.from_orm(self)
|
|
return u.json()
|
|
|
|
|
|
class AnonymousUser(AnonymousUserMixin):
|
|
pass
|