init: new structure + fix lint errors
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import DateTime, Float, ForeignKey, JSON, String, Text
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
|
||||
from cpv3.db.base import Base, BaseModelMixin
|
||||
|
||||
|
||||
class Job(Base, BaseModelMixin):
|
||||
__tablename__ = "jobs"
|
||||
|
||||
broker_id: Mapped[str] = mapped_column(String(255))
|
||||
|
||||
user_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("users.id", ondelete="RESTRICT"), nullable=True, index=True
|
||||
)
|
||||
project_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True),
|
||||
ForeignKey("projects.id", ondelete="RESTRICT"),
|
||||
nullable=True,
|
||||
index=True,
|
||||
)
|
||||
|
||||
input_data: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
||||
output_data: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
||||
|
||||
status: Mapped[str] = mapped_column(String(16), default="PENDING")
|
||||
job_type: Mapped[str] = mapped_column(String(32), default="PENDING")
|
||||
|
||||
project_pct: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
current_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
|
||||
started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
finished_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
|
||||
class JobEvent(Base, BaseModelMixin):
|
||||
__tablename__ = "job_events"
|
||||
|
||||
job_id: Mapped[uuid.UUID] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("jobs.id", ondelete="CASCADE"), index=True
|
||||
)
|
||||
event_type: Mapped[str] = mapped_column(String(64))
|
||||
payload: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
||||
Reference in New Issue
Block a user