CRUD de agentes aduanales funcional, edicion y creacion

backend/api/v1/modules/a76/customs_brokers/routes.py

@@ -1,35 +1,64 @@
 from typing import Dict, Any
-from api.v1.common.tenant_crud_routes import TenantCRUDRoutes
-from core.database import get_core_db
-from core.security import get_current_user, validate_access_to_resource
 from fastapi import APIRouter, Depends, HTTPException, Query
 from sqlalchemy.orm import Session
 
+from core.database import get_core_db
+from core.security import get_current_user, validate_access_to_resource
+
+from api.v1.common.tenant_crud_routes import TenantCRUDRoutes
 from . import dto, services
 
-# Create main router
 router = APIRouter()
 
-# Create CRUD routes for CustomsBroker using TenantCRUDRoutes
 customs_broker_crud = TenantCRUDRoutes(
     service=services.CustomsBrokerService,
     create_schema=dto.CustomsBrokerCreateDTO,
     update_schema=dto.CustomsBrokerUpdateDTO,
     response_schema=dto.CustomsBrokerResponseDTO,
-    prefix="/customs-brokers",  # No prefix since it's already in the parent router
+    prefix="/customs-brokers", 
     tags=[],
     resource_name="Customs Broker",
     id_name="broker_key",
     id_type=str,
-    enable_list=True,  # Enable list endpoint with pagination
+    enable_list=True,
 )
 
-# Include the CRUD routes
 router.include_router(customs_broker_crud.router)
 
 
-# Additional routes for child resources (CustomsBrokerVU and CustomsBrokerPersonnel)
-# These remain as manual routes since they have different patterns
+@router.patch(
+    "/customs-brokers/{broker_key}", 
+    response_model=dto.CustomsBrokerResponseDTO,
+)
+def update_customs_broker(
+    broker_key: str,
+    broker_data: dto.CustomsBrokerUpdateDTO,
+    company_id: int = Query(..., description="Company ID"),
+    db: Session = Depends(get_core_db),
+    current_user: Dict[str, Any] = Depends(get_current_user),
+):
+    """
+    Actualización parcial (PATCH).
+    """
+    tenant_id = validate_access_to_resource(db, company_id, current_user)
+    
+    broker = services.CustomsBrokerService.get_by_id(db, broker_key, tenant_id, company_id)
+    if not broker:
+        raise HTTPException(status_code=404, detail="Customs Broker not found")
+
+    updated_broker = services.CustomsBrokerService.update(
+        db=db,
+        broker_key=broker_key,
+        tenant_id=tenant_id,
+        broker_data=broker_data,  
+        company_id=company_id     
+    )
+    
+    if not updated_broker:
+        raise HTTPException(status_code=400, detail="Error updating Customs Broker")
+        
+    return updated_broker
+
 
 @router.put(
     "/customs-broker-vu/{broker_key}", 
@@ -44,7 +73,6 @@ def update_customs_broker_vu(
 ):
     tenant_id = validate_access_to_resource(db, company_id, current_user)
     
-    # Verify the broker exists and belongs to the tenant/company
     broker = services.CustomsBrokerService.get_by_id(db, broker_key, tenant_id, company_id)
     if not broker:
         raise HTTPException(status_code=404, detail="Customs Broker not found")
@@ -69,7 +97,6 @@ def update_customs_broker_personnel(
 ):
     tenant_id = validate_access_to_resource(db, company_id, current_user)
     
-    # Verify the broker exists and belongs to the tenant/company
     broker = services.CustomsBrokerService.get_by_id(db, broker_key, tenant_id, company_id)
     if not broker:
         raise HTTPException(status_code=404, detail="Customs Broker not found")
@@ -81,5 +108,4 @@ def update_customs_broker_personnel(
         raise HTTPException(
             status_code=404, detail="Customs Broker Personnel not found"
         )
-    return updated_personnel
-
+    return updated_personnel