| 
 | 1 | +import uuid  | 
 | 2 | +from typing import Any  | 
 | 3 | + | 
 | 4 | +from sqlalchemy import func  | 
 | 5 | +from sqlmodel import Session, select  | 
 | 6 | + | 
 | 7 | +from app.core.security import get_password_hash  | 
 | 8 | +from app.models import User, UserCreate, UserUpdate  | 
 | 9 | + | 
 | 10 | + | 
 | 11 | +class UserModel:  | 
 | 12 | +    def __init__(self, session: Session):  | 
 | 13 | +        self.session = session  | 
 | 14 | + | 
 | 15 | +    def create(cls, user_create: UserCreate) -> "User":  | 
 | 16 | +        db_obj = cls.model_validate(  | 
 | 17 | +            user_create,  | 
 | 18 | +            update={"hashed_password": get_password_hash(user_create.password)},  | 
 | 19 | +        )  | 
 | 20 | +        cls.session.add(db_obj)  | 
 | 21 | +        cls.session.commit()  | 
 | 22 | +        cls.session.refresh(db_obj)  | 
 | 23 | +        return db_obj  | 
 | 24 | + | 
 | 25 | +    def update(cls, db_user: "User", user_in: UserUpdate) -> Any:  | 
 | 26 | +        user_data = user_in.model_dump(exclude_unset=True)  | 
 | 27 | +        extra_data = {}  | 
 | 28 | +        if "password" in user_data:  | 
 | 29 | +            password = user_data["password"]  | 
 | 30 | +            hashed_password = get_password_hash(password)  | 
 | 31 | +            extra_data["hashed_password"] = hashed_password  | 
 | 32 | +        db_user.sqlmodel_update(user_data, update=extra_data)  | 
 | 33 | +        cls.session.add(db_user)  | 
 | 34 | +        cls.session.commit()  | 
 | 35 | +        cls.session.refresh(db_user)  | 
 | 36 | +        return db_user  | 
 | 37 | + | 
 | 38 | +    def get_by_email(cls, email: str) -> "User | None":  | 
 | 39 | +        statement = select(cls.__class__).where(cls.__class__.email == email)  | 
 | 40 | +        return cls.session.exec(statement).first()  | 
 | 41 | + | 
 | 42 | +    def get_by_id(cls, user_id: str) -> "User | None":  | 
 | 43 | +        statement = select(cls.__class__).where(cls.__class__.id == uuid.UUID(user_id))  | 
 | 44 | +        return cls.session.exec(statement).first()  | 
 | 45 | + | 
 | 46 | +    def get_users(cls, skip: int = 0, limit: int = 100) -> dict:  | 
 | 47 | +        count_statement = select(func.count()).select_from(cls.__class__)  | 
 | 48 | +        count = cls.session.exec(count_statement).one()  | 
 | 49 | +        statement = select(cls.__class__).offset(skip).limit(limit)  | 
 | 50 | +        users = cls.session.exec(statement).all()  | 
 | 51 | +        return {"data": users, "count": count}  | 
 | 52 | + | 
 | 53 | +    def delete_user(cls, user_id: str) -> None:  | 
 | 54 | +        user = cls.get_by_id(user_id)  | 
 | 55 | +        if user:  | 
 | 56 | +            cls.session.delete(user)  | 
 | 57 | +            cls.session.commit()  | 
0 commit comments