mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
correct SQLModel typing
This commit is contained in:
parent
08cf930f91
commit
78ff040711
1 changed files with 14 additions and 2 deletions
|
|
@ -3,7 +3,7 @@ from typing import Optional
|
|||
from datetime import datetime
|
||||
from uuid import UUID, uuid4
|
||||
|
||||
from sqlalchemy import Enum
|
||||
from sqlalchemy import Column, Enum
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
from sqlmodel import SQLModel, Field, Relationship
|
||||
|
||||
|
|
@ -27,7 +27,19 @@ class Task(SQLModel, table=True):
|
|||
service: Optional[str] = None
|
||||
updated_at: datetime = Field(default_factory=datetime.utcnow)
|
||||
|
||||
source: Mapped[Optional[SourceEnum]] = mapped_column(Enum(SourceEnum))
|
||||
# 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")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue