from tortoise import fields, models
from enum import Enum

class TaskStatus(str, Enum):
    PENDING = "pending"
    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"

class ImportTask(models.Model):
    id = fields.IntField(pk=True)
    task_type = fields.CharField(max_length=50)
    status = fields.CharEnumField(TaskStatus, default=TaskStatus.PENDING)
    details = fields.JSONField(null=True)
    created_at = fields.DatetimeField(auto_now_add=True)
    updated_at = fields.DatetimeField(auto_now=True)

    class Meta:
        table = "import_tasks"