from fastapi import APIRouter, Depends, status from fastapi.responses import JSONResponse from app.backend.db.database import get_db from sqlalchemy.orm import Session from app.backend.schemas import CustomerList, StoreCustomer, UpdateCustomer, UserLogin from app.backend.classes.customer_class import CustomerClass from app.backend.classes.user_class import UserClass from app.backend.classes.school_class import SchoolClass from app.backend.classes.rol_class import RolClass from app.backend.classes.teaching_class import TeachingClass from app.backend.db.models import SchoolModel from app.backend.auth.auth_user import get_current_active_user from datetime import datetime as dt customers = APIRouter( prefix="/customers", tags=["Customers"] ) @customers.post("/") def index(customer_list: CustomerList, session_user: UserLogin = Depends(get_current_active_user), db: Session = Depends(get_db)): page_value = 0 if customer_list.page is None else customer_list.page result = CustomerClass(db).get_all( page=page_value, items_per_page=customer_list.per_page, identification_number=customer_list.identification_number, names=customer_list.names, company_name=customer_list.company_name ) if isinstance(result, dict) and result.get("status") == "error": return JSONResponse( status_code=status.HTTP_404_NOT_FOUND, content={ "status": 404, "message": result.get("message", "Error"), "data": None } ) message = "Complete customers list retrieved successfully" if customer_list.page is None else "Customers retrieved successfully" return JSONResponse( status_code=status.HTTP_200_OK, content={ "status": 200, "message": message, "data": result } ) @customers.post("/store") def store(customer: StoreCustomer, session_user: UserLogin = Depends(get_current_active_user), db: Session = Depends(get_db)): customer_inputs = customer.dict() # Extraer campos para crear usuario email = customer_inputs.get('email') password = customer_inputs.get('password') rol_id = customer_inputs.get('rol_id') # Extraer schools antes de crear customer (no es campo del modelo Customer) schools = customer_inputs.pop('schools', None) # Crear el customer result = CustomerClass(db).store(customer_inputs) if isinstance(result, dict) and result.get("status") == "error": return JSONResponse( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content={ "status": 500, "message": result.get("message", "Error creating customer"), "data": None } ) # Si el customer se creó exitosamente, crear solo el usuario (sin rol ni rol_permission) if isinstance(result, dict) and result.get("status") == "success": customer_id = result.get("customer_id") # Crear el usuario vinculado al customer user_inputs = { "customer_id": customer_id, "school_id": None, "rol_id": rol_id, "rut": customer_inputs.get("identification_number"), "full_name": f"{customer_inputs.get('names', '')} {customer_inputs.get('lastnames', '')}" .strip(), "email": email, "password": password, "phone": customer_inputs.get("phone"), "branch_office_id": None } user_result = UserClass(db).store(user_inputs) if user_result == 0: return JSONResponse( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content={ "status": 500, "message": "Customer created but error creating user", "data": result } ) # Guardar schools si vienen en el request if schools and isinstance(schools, list): school_class = SchoolClass(db) rol_class = RolClass(db) teaching_class = TeachingClass(db) for school_name in schools: if school_name and school_name.strip(): school_inputs = { "customer_id": customer_id, "school_name": school_name.strip(), "school_address": None, "director_name": None, "community_school_password": None } school_result = school_class.store(school_inputs) # Si el school se creó exitosamente, crear los roles y enseñanzas automáticos if isinstance(school_result, dict) and school_result.get("status") == "success": school_id = school_result.get("school_id") # Crear rol "Profesional" con permisos: 40 (Ver cursos), 41 (Filtrar cursos) rol_profesional_inputs = { "customer_id": customer_id, "school_id": school_id, "rol": "Profesional", "permissions": [40, 41] } rol_class.store(rol_profesional_inputs) # Crear rol "Coordinador" con permisos múltiples incluyendo 40 (Ver cursos) y 41 (Filtrar cursos) rol_coordinador_inputs = { "customer_id": customer_id, "school_id": school_id, "rol": "Coordinador", "permissions": [1, 2, 3, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 40, 41] } rol_class.store(rol_coordinador_inputs) # Crear enseñanzas automáticas: Pre Básica, Básica y Media teachings_to_create = [ {"teaching_name": "Pre Básica", "teaching_type_id": 1}, {"teaching_name": "Básica", "teaching_type_id": 2}, {"teaching_name": "Media", "teaching_type_id": 3} ] for teaching_data in teachings_to_create: teaching_inputs = { "school_id": school_id, "teaching_type_id": teaching_data["teaching_type_id"], "teaching_name": teaching_data["teaching_name"] } teaching_class.store(teaching_inputs) return JSONResponse( status_code=status.HTTP_201_CREATED, content={ "status": 201, "message": "Customer, user, schools and roles created successfully", "data": result } ) @customers.get("/edit/{id}") def edit(id: int, session_user: UserLogin = Depends(get_current_active_user), db: Session = Depends(get_db)): result = CustomerClass(db).get(id) if isinstance(result, dict) and (result.get("error") or result.get("status") == "error"): return JSONResponse( status_code=status.HTTP_404_NOT_FOUND, content={ "status": 404, "message": result.get("error") or result.get("message", "Customer not found"), "data": None } ) # Obtener los colegios asociados al customer schools_result = SchoolClass(db).get_all(page=0, customer_id=id) schools_list = [] if isinstance(schools_result, list): schools_list = [school.get("school_name") for school in schools_result if school.get("school_name")] # Agregar schools al resultado if isinstance(result, dict) and result.get("customer_data"): result["customer_data"]["schools"] = schools_list return JSONResponse( status_code=status.HTTP_200_OK, content={ "status": 200, "message": "Customer retrieved successfully", "data": result } ) @customers.delete("/{id}") @customers.delete("/delete/{id}") def delete(id: int, session_user: UserLogin = Depends(get_current_active_user), db: Session = Depends(get_db)): result = CustomerClass(db).delete(id) if isinstance(result, dict) and result.get("status") == "error": return JSONResponse( status_code=status.HTTP_404_NOT_FOUND, content={ "status": 404, "message": result.get("message", "Customer not found"), "data": None } ) return JSONResponse( status_code=status.HTTP_200_OK, content={ "status": 200, "message": "Customer deleted successfully", "data": result } ) @customers.put("/update/{id}") def update(id: int, customer: UpdateCustomer, session_user: UserLogin = Depends(get_current_active_user), db: Session = Depends(get_db)): customer_inputs = customer.dict(exclude_unset=True) # Extraer schools antes de actualizar customer (no es campo del modelo Customer) schools = customer_inputs.pop('schools', None) result = CustomerClass(db).update(id, customer_inputs) if isinstance(result, dict) and result.get("status") == "error": return JSONResponse( status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content={ "status": 500, "message": result.get("message", "Error updating customer"), "data": None } ) # Actualizar schools si vienen en el request if schools is not None: school_class = SchoolClass(db) # Marcar todos los schools existentes como eliminados (deleted_status_id = 1) existing_schools = db.query(SchoolModel).filter( SchoolModel.customer_id == id, SchoolModel.deleted_status_id == 0 ).all() for school in existing_schools: school.deleted_status_id = 1 school.updated_date = dt.now() db.commit() # Crear los nuevos schools if isinstance(schools, list): for school_name in schools: if school_name and school_name.strip(): school_inputs = { "customer_id": id, "school_name": school_name.strip(), "school_address": None, "director_name": None, "community_school_password": None } school_class.store(school_inputs) return JSONResponse( status_code=status.HTTP_200_OK, content={ "status": 200, "message": "Customer and schools updated successfully", "data": result } )