Se agrego coves

This commit is contained in:
2025-08-06 14:28:34 -06:00
parent 543a3558ca
commit ea40a0c74e
3 changed files with 34 additions and 26 deletions

View File

@@ -253,25 +253,27 @@ class APIController:
"""
return await self._make_request_async('PUT', f'customs/edocuments/{edocument_id}/', data=data)
async def get_cer(self, id: str) -> Dict[str, Any]:
async def get_cer(self, id: str) -> bytes:
"""
Método para obtener un certificado específico desde la API.
Método para obtener un certificado específico desde la API (como binario).
Args:
id: UUID del certificado a consultar
Returns:
bytes: Contenido binario del certificado
"""
return await self._make_request_async('GET', f'vucem/vucem/{id}/download_cer/')
return await self._make_request_async('GET', f'vucem/vucem/{id}/download_cer/', return_bytes=True)
async def get_key(self, id: str) -> Dict[str, Any]:
async def get_key(self, id: str) -> bytes:
"""
Método para obtener una llave específica desde la API.
Método para obtener una llave específica desde la API (como binario).
Args:
id: UUID de la llave a consultar
Returns:
bytes: Contenido binario de la llave
"""
return await self._make_request_async('GET', f'vucem/vucem/{id}/download_key/')
async def _make_request_async(self, method: str, endpoint: str, data=None):
return await self._make_request_async('GET', f'vucem/vucem/{id}/download_key/', return_bytes=True)
async def _make_request_async(self, method: str, endpoint: str, data=None, return_bytes: bool = False):
"""
Método asíncrono para hacer peticiones a la API usando httpx.
"""
@@ -296,9 +298,11 @@ class APIController:
response.raise_for_status()
logger.info(f"Respuesta exitosa: {response.status_code}")
result = response.json() if response.content else {}
return result
if return_bytes:
return response.content
else:
result = response.json() if response.content else {}
return result
except httpx.TimeoutException as e:
logger.error(f"Timeout en petición a {url}: {e}")