from app.config.DB import db_manager
from app.modules.user.user_model import UserModel


class UserRepository:

    def __init__(self):
        self.collection_name = "users"

    def get_collection(self):
        return db_manager.db[self.collection_name]

    async def create_user(self, user: UserModel):
        collection = self.get_collection()

        user_dict = user.model_dump()

        result = await collection.insert_one(user_dict)

        user_dict["_id"] = str(result.inserted_id)

        return user_dict
    async def find_by_phone(self, phone: str):
        collection = self.get_collection()
        return await collection.find_one({"phone": phone})