mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import enum
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
from uuid import UUID, uuid4
|
|
|
|
from sqlalchemy import Column, Enum
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from sqlmodel import SQLModel, Field, Relationship
|
|
|
|
|
|
class SourceEnum(enum.Enum): # TODO: move to domain?
|
|
PORTFOLIO = "portfolio_id"
|
|
|
|
|
|
class Task(SQLModel, table=True):
|
|
__tablename__ = "tasks"
|
|
|
|
id: UUID = Field(
|
|
default_factory=uuid4,
|
|
primary_key=True,
|
|
index=True,
|
|
)
|
|
task_source: str
|
|
job_started: Optional[datetime] = None
|
|
job_completed: Optional[datetime] = None
|
|
status: str = Field(default="In Progress")
|
|
service: Optional[str] = None
|
|
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
# source: Mapped[Optional[SourceEnum]] = mapped_column(Enum(SourceEnum)) <- SQLAlchemy not SQLModel
|
|
|
|
source: Optional[SourceEnum] = Field(
|
|
default=None,
|
|
sa_column=Column(
|
|
Enum(
|
|
SourceEnum,
|
|
name="source",
|
|
values_callable=lambda e: [m.value for m in e],
|
|
),
|
|
nullable=True,
|
|
),
|
|
)
|
|
source_id: Optional[str] = None
|
|
|
|
sub_tasks: list["SubTask"] = Relationship(back_populates="task")
|
|
|
|
|
|
class SubTask(SQLModel, table=True):
|
|
__tablename__ = "sub_task"
|
|
|
|
id: UUID = Field(
|
|
default_factory=uuid4,
|
|
primary_key=True,
|
|
index=True,
|
|
)
|
|
|
|
task_id: UUID = Field(foreign_key="tasks.id")
|
|
job_started: Optional[datetime] = None
|
|
job_completed: Optional[datetime] = None
|
|
status: str = Field(default="In Progress")
|
|
inputs: Optional[str] = None
|
|
outputs: Optional[str] = None
|
|
cloud_logs_url: Optional[str] = None
|
|
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
task: Optional["Task"] = Relationship(back_populates="sub_tasks")
|