42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
"""Celery task: async report generation."""
|
||
|
|
import asyncio
|
||
|
|
import logging
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
from app.workers.celery_app import celery_app
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def _run_async(coro):
|
||
|
|
loop = asyncio.new_event_loop()
|
||
|
|
try:
|
||
|
|
return loop.run_until_complete(coro)
|
||
|
|
finally:
|
||
|
|
loop.close()
|
||
|
|
|
||
|
|
|
||
|
|
@celery_app.task(name="app.workers.report_tasks.generate_report_async", bind=True)
|
||
|
|
def generate_report_async(self, case_id_str: str, report_type: str):
|
||
|
|
_run_async(_run(case_id_str, report_type))
|
||
|
|
|
||
|
|
|
||
|
|
async def _run(case_id_str: str, report_type: str):
|
||
|
|
from app.core.database import async_session_factory
|
||
|
|
from app.models.report import ReportType
|
||
|
|
from app.schemas.report import ReportCreate
|
||
|
|
from app.services.report_service import generate_report
|
||
|
|
|
||
|
|
case_id = UUID(case_id_str)
|
||
|
|
body = ReportCreate(report_type=ReportType(report_type))
|
||
|
|
|
||
|
|
async with async_session_factory() as db:
|
||
|
|
try:
|
||
|
|
report = await generate_report(case_id, body, db)
|
||
|
|
await db.commit()
|
||
|
|
logger.info("Report generated for case %s: %s", case_id, report.file_path)
|
||
|
|
except Exception as e:
|
||
|
|
await db.rollback()
|
||
|
|
logger.error("Report generation failed: %s", e)
|
||
|
|
raise
|