4 Commits

12 changed files with 90 additions and 42 deletions

View File

@@ -145,12 +145,13 @@ async def obtener_acuse(**kwargs):
organizacion = pedimento.get("organizacion", None) organizacion = pedimento.get("organizacion", None)
pedimento_id = pedimento.get("id", None) pedimento_id = pedimento.get("id", None)
rest_response = await acuse_rest_controller.post_document( rest_response = await acuse_rest_controller.post_or_update_document(
binary_content=pdf_bytes, binary_content=pdf_bytes,
organizacion=organizacion, organizacion=organizacion,
pedimento=pedimento_id, pedimento=pedimento_id,
file_name=_file_name, file_name=_file_name,
document_type=4 document_type=4,
identifier=idEdocument_efc,
) )
if rest_response is None: if rest_response is None:

View File

@@ -134,12 +134,13 @@ async def consume_ws_get_cove(**kwargs):
# Enviar documento # Enviar documento
_file_name = f"vu_COVE_{pedimento_app}_{cove}.xml" _file_name = f"vu_COVE_{pedimento_app}_{cove}.xml"
try: try:
document_response = await coves_rest_controller.post_document( document_response = await coves_rest_controller.post_or_update_document(
soap_response=soap_response, soap_response=soap_response,
organizacion=kwargs.get('pedimento').get('organizacion'), organizacion=kwargs.get('pedimento').get('organizacion'),
pedimento=kwargs.get('pedimento').get('id'), pedimento=kwargs.get('pedimento').get('id'),
file_name=_file_name, file_name=_file_name,
document_type=8, document_type=8,
identifier=cove,
) )
except Exception as e: except Exception as e:
logger.error(f"Error detectado en la respuesta SOAP: {str(e)}") logger.error(f"Error detectado en la respuesta SOAP: {str(e)}")
@@ -318,12 +319,14 @@ async def consume_ws_get_acuse_cove(**kwargs):
organizacion = pedimento.get("organizacion", None) organizacion = pedimento.get("organizacion", None)
pedimento_id = pedimento.get("id", None) pedimento_id = pedimento.get("id", None)
rest_response = await coves_rest_controller.post_document( cove_identifier = kwargs['cove'].get('cove', '')
rest_response = await coves_rest_controller.post_or_update_document(
binary_content=pdf_bytes, binary_content=pdf_bytes,
organizacion=organizacion, organizacion=organizacion,
pedimento=pedimento_id, pedimento=pedimento_id,
file_name=_file_name, file_name=_file_name,
document_type=7 document_type=7,
identifier=cove_identifier,
) )
acuse_status = await change_acuse_status( acuse_status = await change_acuse_status(

View File

@@ -3,6 +3,7 @@ import logging
from celery import Celery from celery import Celery
from celery_app import celery_app from celery_app import celery_app
from typing import Dict, Any from typing import Dict, Any
from fastapi import HTTPException as _HTTPException
from .services import consume_ws_get_cove, consume_ws_get_acuse_cove from .services import consume_ws_get_cove, consume_ws_get_acuse_cove
from api.api_v2.modules.tasks.services import update_task, register_task from api.api_v2.modules.tasks.services import update_task, register_task
@@ -83,7 +84,9 @@ def process_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str, Any]:
) )
except Exception as update_error: except Exception as update_error:
logger.error(f"No se pudo actualizar el estado de la tarea: {update_error}") logger.error(f"No se pudo actualizar el estado de la tarea: {update_error}")
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise
@@ -159,5 +162,7 @@ def process_acuse_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str,
) )
except Exception as update_error: except Exception as update_error:
logger.error(f"No se pudo actualizar el estado de la tarea: {update_error}") logger.error(f"No se pudo actualizar el estado de la tarea: {update_error}")
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise

View File

@@ -203,12 +203,13 @@ async def obtener_edoc(**kwargs):
# No guardaremos el archivo localmente por seguridad # No guardaremos el archivo localmente por seguridad
logger.debug(f"Procesando documento {numero_documento} para pedimento {pedimento_id}") logger.debug(f"Procesando documento {numero_documento} para pedimento {pedimento_id}")
rest_response = await edocs_rest_controller.post_document( rest_response = await edocs_rest_controller.post_or_update_document(
binary_content=pdf_bytes, binary_content=pdf_bytes,
organizacion=organizacion, organizacion=organizacion,
pedimento=pedimento_id, pedimento=pedimento_id,
file_name=_file_name, file_name=_file_name,
document_type=5 document_type=5,
identifier=numero_documento,
) )
print(f"rest_response >>>> {rest_response}") print(f"rest_response >>>> {rest_response}")

View File

@@ -1,10 +1,11 @@
import asyncio import asyncio
import logging import logging
import time import time
from celery_app import celery_app from celery_app import celery_app
from typing import Dict from typing import Dict
from fastapi import HTTPException as _HTTPException
from .services import obtener_edoc from .services import obtener_edoc
from api.api_v2.modules.tasks.services import register_task, update_task from api.api_v2.modules.tasks.services import register_task, update_task
# Logger para el módulo # Logger para el módulo
@@ -92,7 +93,9 @@ def process_edoc_download_request(self, edoc_data: Dict) -> Dict:
) )
except Exception as update_error: except Exception as update_error:
logger.error(f"Error actualizando estado de tarea: {update_error}") logger.error(f"Error actualizando estado de tarea: {update_error}")
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise
finally: finally:
# Cerrar el loop para liberar recursos # Cerrar el loop para liberar recursos

View File

@@ -59,12 +59,13 @@ async def consume_ws_get_partida(**kwargs):
file_name_request = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}_REQUEST.xml" file_name_request = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}_REQUEST.xml"
document_response = await partida_rest_controller.post_document( document_response = await partida_rest_controller.post_or_update_document(
soap_response=soap_xml, soap_response=soap_xml,
organizacion=kwargs.get('pedimento').get('organizacion'), organizacion=kwargs.get('pedimento').get('organizacion'),
pedimento=kwargs.get('pedimento').get('id'), pedimento=kwargs.get('pedimento').get('id'),
file_name=file_name_request, file_name=file_name_request,
document_type=17, # Tipo de documento para petición de partidas document_type=17,
identifier=f"_PT_{pedimento_app}_{partida.get('numero', '')}_",
) )
except Exception as e: except Exception as e:
logger.error(f"Error al enviar documento request: {e}") logger.error(f"Error al enviar documento request: {e}")
@@ -97,12 +98,13 @@ async def consume_ws_get_partida(**kwargs):
if soap_error(soap_response): if soap_error(soap_response):
error_file_name = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}_ERROR.xml" error_file_name = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}_ERROR.xml"
try: try:
document_response = await partida_rest_controller.post_document( document_response = await partida_rest_controller.post_or_update_document(
soap_response=soap_response, soap_response=soap_response,
organizacion=kwargs.get('pedimento').get('organizacion'), organizacion=kwargs.get('pedimento').get('organizacion'),
pedimento=kwargs.get('pedimento').get('id'), pedimento=kwargs.get('pedimento').get('id'),
file_name=error_file_name, file_name=error_file_name,
document_type=18, document_type=18,
identifier=f"_PT_{pedimento_app}_{partida.get('numero', '')}_",
) )
except Exception as e: except Exception as e:
logger.error(f"Error al guardar la respuesta de error: {e}") logger.error(f"Error al guardar la respuesta de error: {e}")
@@ -127,12 +129,13 @@ async def consume_ws_get_partida(**kwargs):
# Enviar documento # Enviar documento
_file_name = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}.xml" _file_name = f"vu_PT_{pedimento_app}_{partida.get('numero', '')}.xml"
try: try:
document_response = await partida_rest_controller.post_document( document_response = await partida_rest_controller.post_or_update_document(
soap_response=soap_response, soap_response=soap_response,
organizacion=kwargs.get('pedimento').get('organizacion'), organizacion=kwargs.get('pedimento').get('organizacion'),
pedimento=kwargs.get('pedimento').get('id'), pedimento=kwargs.get('pedimento').get('id'),
file_name=_file_name, file_name=_file_name,
document_type=1, document_type=1,
identifier=f"_PT_{pedimento_app}_{partida.get('numero', '')}_",
) )
except Exception as e: except Exception as e:
logger.error(f"Error al enviar documento: {e}") logger.error(f"Error al enviar documento: {e}")

View File

@@ -4,6 +4,7 @@ import time
from celery import Celery from celery import Celery
from celery_app import celery_app from celery_app import celery_app
from typing import Dict, Any from typing import Dict, Any
from fastapi import HTTPException as _HTTPException
from .services import consume_ws_get_partida from .services import consume_ws_get_partida
from api.api_v2.modules.tasks.services import register_task, update_task from api.api_v2.modules.tasks.services import register_task, update_task
@@ -89,6 +90,8 @@ def process_partida_request(self, partida_request: Dict[str, Any]) -> Dict[str,
) )
except Exception as update_error: except Exception as update_error:
logger.error(f"Error al actualizar estado de tarea: {update_error}") logger.error(f"Error al actualizar estado de tarea: {update_error}")
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise
finally: finally:
# Limpiar el event loop # Limpiar el event loop

View File

@@ -64,12 +64,12 @@ async def consume_ws_get_pedimento_completo(**kwargs) -> Dict[str, Any]:
file_name_request = f"VU_PC_{pedimento_data.get('pedimento_app', 'unknown')}_REQUEST.xml" file_name_request = f"VU_PC_{pedimento_data.get('pedimento_app', 'unknown')}_REQUEST.xml"
document_response = await pedimento_rest_controller.post_document( document_response = await pedimento_rest_controller.post_or_update_document(
soap_response=soap_xml, soap_response=soap_xml,
organizacion=pedimento_data.get('organizacion'), organizacion=pedimento_data.get('organizacion'),
pedimento=pedimento_data.get('id'), pedimento=pedimento_data.get('id'),
file_name=file_name_request, file_name=file_name_request,
document_type=13, document_type=13,
) )
except Exception as e: except Exception as e:
logger.error(f"Error al enviar documento request: {e}") logger.error(f"Error al enviar documento request: {e}")
@@ -90,12 +90,12 @@ async def consume_ws_get_pedimento_completo(**kwargs) -> Dict[str, Any]:
if soap_error(soap_response): if soap_error(soap_response):
logger.error(f"Error en respuesta SOAP: {soap_response.text if hasattr(soap_response, 'text') else 'Sin detalles'}") logger.error(f"Error en respuesta SOAP: {soap_response.text if hasattr(soap_response, 'text') else 'Sin detalles'}")
document_response = await pedimento_rest_controller.post_document( document_response = await pedimento_rest_controller.post_or_update_document(
soap_response=None, soap_response=soap_response,
organizacion=pedimento_data.get('organizacion'), organizacion=pedimento_data.get('organizacion'),
pedimento=pedimento_data.get('id'), pedimento=pedimento_data.get('id'),
file_name=f"vu_PC_{pedimento_data.get('pedimento_app', 'unknown')}_ERROR.xml", file_name=f"vu_PC_{pedimento_data.get('pedimento_app', 'unknown')}_ERROR.xml",
document_type=14, document_type=14,
) )
raise HTTPException(status_code=500, detail="Error en la respuesta del servicio SOAP") raise HTTPException(status_code=500, detail="Error en la respuesta del servicio SOAP")
@@ -111,12 +111,12 @@ async def consume_ws_get_pedimento_completo(**kwargs) -> Dict[str, Any]:
# Enviar documento # Enviar documento
try: try:
document_response = await pedimento_rest_controller.post_document( document_response = await pedimento_rest_controller.post_or_update_document(
soap_response=soap_response, soap_response=soap_response,
organizacion=pedimento_data.get('organizacion'), organizacion=pedimento_data.get('organizacion'),
pedimento=pedimento_data.get('id'), pedimento=pedimento_data.get('id'),
file_name=file_name, file_name=file_name,
document_type=2, document_type=2,
) )
except Exception as e: except Exception as e:
logger.error(f"Error al enviar documento: {e}") logger.error(f"Error al enviar documento: {e}")

View File

@@ -1,7 +1,8 @@
from celery_app import celery_app from celery_app import celery_app
from .services import put_pedimento_data from .services import put_pedimento_data
import asyncio import asyncio
import logging import logging
from fastapi import HTTPException as _HTTPException
from ..tasks.services import register_task, update_task from ..tasks.services import register_task, update_task
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -84,8 +85,10 @@ def process_pedimento_completo_request(self, pedimento_data: dict):
) )
except Exception as update_error: except Exception as update_error:
logger.error(f"Error actualizando estado de tarea: {update_error}") logger.error(f"Error actualizando estado de tarea: {update_error}")
# Re-lanzar la excepción para que Celery la marque como fallida # Re-lanzar la excepción para que Celery la marque como fallida
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise
finally: finally:
loop.close() loop.close()

View File

@@ -87,8 +87,33 @@ async def obtener_remesa(**kwargs) -> Dict[str, Any]:
) )
# Generar nombre de archivo # Generar nombre de archivo
file_name = f"vu_RM_{pedimento_data.get('pedimento_app', 'unknown')}.xml" file_name = f"vu_RM_{pedimento_data.get('pedimento_app', 'unknown')}.xml"
# "No hay información" NO es un error — VUCEM confirma que el pedimento no tiene remesas.
# No se crea documento de error; se corrige el flag en el pedimento.
_SIN_REMESAS = '<ns2:mensaje>No hay información para la búsqueda solicitada</ns2:mensaje>'
if hasattr(soap_response, 'text') and _SIN_REMESAS in soap_response.text:
pedimento_id = pedimento_data.get('id')
logger.info(
f"Pedimento {pedimento_data.get('pedimento_app')} no tiene remesas "
f"(confirmado por VUCEM). Corrigiendo flag remesas=False."
)
try:
await remesa_rest_controller._make_request_async(
'PATCH',
f'customs/pedimentos/{pedimento_id}/',
data={'remesas': False},
)
except Exception as patch_err:
logger.warning(f"No se pudo actualizar remesas=False para {pedimento_id}: {patch_err}")
return create_service_response(
message="Pedimento sin remesas confirmado por VUCEM",
data={"remesas": False},
metadata={"pedimento_app": pedimento_data.get('pedimento_app')},
)
if soap_error(soap_response): if soap_error(soap_response):
file_name = f"vu_RM_{pedimento_data.get('pedimento_app', 'unknown')}_ERROR.xml" file_name = f"vu_RM_{pedimento_data.get('pedimento_app', 'unknown')}_ERROR.xml"
document_response = await remesa_rest_controller.post_or_update_document( document_response = await remesa_rest_controller.post_or_update_document(
soap_response=soap_response, soap_response=soap_response,
@@ -97,38 +122,31 @@ async def obtener_remesa(**kwargs) -> Dict[str, Any]:
file_name=file_name, file_name=file_name,
document_type=16, document_type=16,
) )
# Aquí necesitamos extraer el mensaje de error real
error_message = "Error en la respuesta del servicio SOAP" error_message = "Error en la respuesta del servicio SOAP"
# Intentar extraer mensaje de error del XML de respuesta
if hasattr(soap_response, 'text') and soap_response.text: if hasattr(soap_response, 'text') and soap_response.text:
try: try:
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
root = ET.fromstring(soap_response.text) root = ET.fromstring(soap_response.text)
# Buscar mensajes de error comunes en respuestas SOAP de VUCEM
# Esto puede variar según el servicio, pero comúnmente buscan:
for fault in root.findall('.//{http://schemas.xmlsoap.org/soap/envelope/}Fault'): for fault in root.findall('.//{http://schemas.xmlsoap.org/soap/envelope/}Fault'):
faultcode = fault.find('.//faultcode')
faultstring = fault.find('.//faultstring') faultstring = fault.find('.//faultstring')
if faultstring is not None and faultstring.text: if faultstring is not None and faultstring.text:
error_message = faultstring.text error_message = faultstring.text
break break
# También podría estar en una estructura de error específica de VUCEM
for error in root.findall('.//{http://www.ventanillaunica.gob.mx/common/ws/oxml/respuesta}error'): for error in root.findall('.//{http://www.ventanillaunica.gob.mx/common/ws/oxml/respuesta}error'):
msg = error.find('.//{http://www.ventanillaunica.gob.mx/common/ws/oxml/respuesta}message') msg = error.find('.//{http://www.ventanillaunica.gob.mx/common/ws/oxml/respuesta}message')
if msg is not None and msg.text: if msg is not None and msg.text:
error_message = msg.text error_message = msg.text
break break
except Exception as parse_error: except Exception as parse_error:
logger.error(f"Error al parsear respuesta SOAP para extraer mensaje: {parse_error}") logger.error(f"Error al parsear respuesta SOAP para extraer mensaje: {parse_error}")
# Lanzar excepción con el mensaje de error real
raise HTTPException( raise HTTPException(
status_code=500, status_code=500,
detail=f"Error en la respuesta del servicio SOAP: {error_message}" detail=f"Error en la respuesta del servicio SOAP: {error_message}"
) )
# Enviar documento # Enviar documento

View File

@@ -2,6 +2,7 @@ from celery_app import celery_app
from .services import post_remesa_data from .services import post_remesa_data
import asyncio import asyncio
import logging import logging
from fastapi import HTTPException as _HTTPException
from ..tasks.services import register_task, update_task from ..tasks.services import register_task, update_task
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -98,6 +99,8 @@ def process_remesa_request(self, remesa_request: dict) -> dict:
logger.error(f"Error actualizando estado de tarea: {update_error}") logger.error(f"Error actualizando estado de tarea: {update_error}")
# Re-lanzar la excepción para que Celery la marque como fallida # Re-lanzar la excepción para que Celery la marque como fallida
if isinstance(e, _HTTPException):
raise RuntimeError(f"HTTP {e.status_code}: {e.detail}") from None
raise raise
finally: finally:

View File

@@ -58,6 +58,8 @@ class APIRESTController:
response = await client.post(url, json=data, headers=self.headers) response = await client.post(url, json=data, headers=self.headers)
elif method.upper() == 'PUT': elif method.upper() == 'PUT':
response = await client.put(url, json=data, headers=self.headers) response = await client.put(url, json=data, headers=self.headers)
elif method.upper() == 'PATCH':
response = await client.patch(url, json=data, headers=self.headers)
elif method.upper() == 'DELETE': elif method.upper() == 'DELETE':
response = await client.delete(url, headers=self.headers) response = await client.delete(url, headers=self.headers)
else: else:
@@ -95,7 +97,7 @@ class APIRESTController:
self, soap_response=None, organizacion: str = None, self, soap_response=None, organizacion: str = None,
pedimento: str = None, file_name: str = None, pedimento: str = None, file_name: str = None,
document_type: int = None, fuente: int = 2, document_type: int = None, fuente: int = 2,
identifier: str = None, identifier: str = None, binary_content: bytes = None,
) -> Dict[str, Any]: ) -> Dict[str, Any]:
""" """
Guarda un documento VU (request o error). Si ya existe uno del mismo Guarda un documento VU (request o error). Si ya existe uno del mismo
@@ -124,6 +126,7 @@ class APIRESTController:
return await self.post_document( return await self.post_document(
soap_response=soap_response, soap_response=soap_response,
binary_content=binary_content,
organizacion=organizacion, organizacion=organizacion,
pedimento=pedimento, pedimento=pedimento,
file_name=file_name, file_name=file_name,
@@ -553,6 +556,8 @@ class APIController:
print(f"response >>>> {response}") print(f"response >>>> {response}")
elif method.upper() == 'PUT': elif method.upper() == 'PUT':
response = await client.put(url, json=data, headers=self.headers) response = await client.put(url, json=data, headers=self.headers)
elif method.upper() == 'PATCH':
response = await client.patch(url, json=data, headers=self.headers)
elif method.upper() == 'DELETE': elif method.upper() == 'DELETE':
response = await client.delete(url, headers=self.headers) response = await client.delete(url, headers=self.headers)
else: else: