| from datetime import datetime | |
| from tortoise.transactions import in_transaction | |
| from App.routers.tasks.models import ImportTask | |
| from .models import UTTFund, UTTFundData | |
| from .service import fetch_all_utt_data, parse_utt_api_row | |
| async def run_utt_import_task(task_id: int): | |
| try: | |
| await ImportTask.filter(id=task_id).update(status="running") | |
| raw_data = await fetch_all_utt_data() | |
| if not raw_data: | |
| await ImportTask.filter(id=task_id).update(status="failed", details={"error": "No data"}) | |
| return | |
| for row in raw_data: | |
| symbol = row["internal_name"] | |
| name = row["scheme_name"] | |
| fund, _ = await UTTFund.get_or_create(symbol=symbol, defaults={"name": name}) | |
| parsed = parse_utt_api_row(row) | |
| exists = await UTTFundData.exists(fund=fund, date=parsed["date"]) | |
| if not exists: | |
| await UTTFundData.create(fund=fund, **parsed) | |
| await ImportTask.filter(id=task_id).update(status="completed") | |
| except Exception as e: | |
| await ImportTask.filter(id=task_id).update(status="failed", details={"error": str(e)}) |