routes/user.py

from fastapi import APIRouter
from config.db import conn
from schemas.user import UserEntity, UserEntityList
from models.user import User, UsertoCreate, UsertoUpdate
from bson import ObjectId

user = APIRouter()

@user.get("/users")
def find_all_users():
    return UserEntityList(conn.Instant.Usuarios.find())

@user.post("/users")
def create_user(user: UsertoCreate):
    new_user = dict(user)

    new_user["contactos"] = []
    
    id = conn.Instant.Usuarios.insert_one(new_user).inserted_id
    
    return UserEntity(conn.Instant.Usuarios.find_one({"_id": ObjectId(id)}))

@user.get("/users/{telefono}")
def find_user_by_id(telefono: str):
    return UserEntity(conn.Instant.Usuarios.find_one({"telefono": telefono}))

@user.put("/users/{telefono}")
def update_user(telefono: str, user: UsertoUpdate):
    conn.Instant.Usuarios.update_one({"telefono": telefono}, {"$set": dict(user)})
    
    return UserEntity(conn.Instant.Usuarios.find_one({"telefono": telefono}))

@user.delete("/users/{telefono}")
def delete_user(telefono: str):
    conn.Instant.Usuarios.delete_one({"telefono": telefono})

    return "Deleted successfully"