latosa/user.py

from flask import Flask
from flask_bcrypt import Bcrypt
class User:
    def __init__(self, app: Flask, uid: str, display_name: str, email:str, password: str, admin: bool = False, groups: list[str] = []):
        self.display_name = display_name
        self.email = email
        self.groups = groups
        self.is_active = False
        self.is_admin = admin
        self.is_anonymous = False
        self.is_authenticated = False
        self.uid = uid
        self.bcrypt = Bcrypt(app)
        self.salt = self.get_salt()
        self.password_hash = self.bcrypt.generate_password_hash(password + self.salt).decode('utf-8')

    def check_password(self, password: str):
        return self.bcrypt.check_password_hash(self.password_hash, password + self.salt)

    def get_id(self):
        return self.uid

    def get_display_name(self):
        return self.display_name

    def get_email(self):
        return self.email

    def get_salt(self):
        return "salt"

    def set_active(self, active: bool):
        self.is_active = active

    def set_authenticated(self, authenticated: bool):
        self.is_authenticated = authenticated
        
    def set_anonymous(self, anonymous: bool):
        self.is_anonymous = anonymous
        
    def set_admin(self, admin: bool):
        self.is_admin = admin

    def set_email(self, email: str):
        self.email = email
        
    def set_password(self, password: str):
        self.password_hash = self.bcrypt.generate_password_hash(password + self.salt).decode('utf-8')