import asyncio
import logging
from celery import Celery
from celery_app import celery_app
from typing import Dict, Any

from .services import consume_ws_get_cove, consume_ws_get_acuse_cove
from api.api_v2.modules.tasks.tasks import run_async_task

# Logger para el módulo
logger = logging.getLogger(__name__)


@celery_app.task(bind=True)
def process_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str, Any]:
    loop = asyncio.get_event_loop()
    cove_response = loop.run_until_complete(consume_ws_get_cove(**cove_request))

    return {"status": "processed", "data": cove_response}


@celery_app.task(bind=True)
def process_acuse_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str, Any]:
    loop = asyncio.get_event_loop()
    acuse_response = loop.run_until_complete(consume_ws_get_acuse_cove(**cove_request))

    return {"status": "processed", "data": acuse_response}