from pydantic import BaseModel from typing import List, Union, Optional import time import uuid import logging from peewee import * from apps.web.models.users import UserModel, Users from utils.utils import verify_password from apps.web.internal.db import DB from config import SRC_LOG_LEVELS log = logging.getLogger(__name__) log.setLevel(SRC_LOG_LEVELS["MODELS"]) #################### # DB MODEL #################### class Auth(Model): id = CharField(unique=True) email = CharField() password = CharField() active = BooleanField() api_key = CharField(null=True, unique=True) class Meta: database = DB class AuthModel(BaseModel): id: str email: str password: str active: bool = True api_key: Optional[str] = None #################### # Forms #################### class Token(BaseModel): token: str token_type: str class ApiKey(BaseModel): api_key: Optional[str] = None class UserResponse(BaseModel): id: str email: str name: str role: str profile_image_url: str class SigninResponse(Token, UserResponse): pass class SigninForm(BaseModel): email: str password: str class ProfileImageUrlForm(BaseModel): profile_image_url: str class UpdateProfileForm(BaseModel): profile_image_url: str name: str class UpdatePasswordForm(BaseModel): password: str new_password: str class SignupForm(BaseModel): name: str email: str password: str class AuthsTable: def __init__(self, db): self.db = db self.db.create_tables([Auth]) def insert_new_auth( self, email: str, password: str, name: str, role: str = "pending" ) -> Optional[UserModel]: log.info("insert_new_auth") id = str(uuid.uuid4()) auth = AuthModel( **{"id": id, "email": email, "password": password, "active": True} ) result = Auth.create(**auth.model_dump()) user = Users.insert_new_user(id, name, email, role) if result and user: return user else: return None def authenticate_user(self, email: str, password: str) -> Optional[UserModel]: log.info(f"authenticate_user: {email}") try: auth = Auth.get(Auth.email == email, Auth.active == True) if auth: if verify_password(password, auth.password): user = Users.get_user_by_id(auth.id) return user else: return None else: return None except: return None def authenticate_user_by_api_key(self, api_key: str) -> Optional[UserModel]: log.info(f"authenticate_user_by_api_key: {api_key}") # if no api_key, return None if not api_key: return None try: auth = Auth.get(Auth.api_key == api_key, Auth.active == True) if auth: user = Users.get_user_by_id(auth.id) return user else: return None except: return None def update_user_password_by_id(self, id: str, new_password: str) -> bool: try: query = Auth.update(password=new_password).where(Auth.id == id) result = query.execute() return True if result == 1 else False except: return False def update_email_by_id(self, id: str, email: str) -> bool: try: query = Auth.update(email=email).where(Auth.id == id) result = query.execute() return True if result == 1 else False except: return False def update_api_key_by_id(self, id: str, api_key: str) -> str: try: query = Auth.update(api_key=api_key).where(Auth.id == id) result = query.execute() return True if result == 1 else False except: return False def get_api_key_by_id(self, id: str) -> Optional[str]: try: auth = Auth.get(Auth.id == id) return auth.api_key except: return None def delete_auth_by_id(self, id: str) -> bool: try: # Delete User result = Users.delete_user_by_id(id) if result: # Delete Auth query = Auth.delete().where(Auth.id == id) query.execute() # Remove the rows, return number of rows removed. return True else: return False except: return False Auths = AuthsTable(DB)