|
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" |