43 lines
2.3 KiB
Python
43 lines
2.3 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from decimal import Decimal
|
|
from sqlalchemy import String, DateTime, Enum as SAEnum, ForeignKey, Text, Integer, Numeric
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from app.database import Base
|
|
import enum
|
|
|
|
|
|
class OrderStatus(str, enum.Enum):
|
|
draft = "draft"
|
|
submitted = "submitted"
|
|
processing = "processing"
|
|
completed = "completed"
|
|
rejected = "rejected"
|
|
|
|
|
|
class Order(Base):
|
|
__tablename__ = "orders"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
order_number: Mapped[str] = mapped_column(String(50), unique=True, nullable=False, index=True)
|
|
template_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("templates.id"), nullable=True)
|
|
status: Mapped[OrderStatus] = mapped_column(SAEnum(OrderStatus), default=OrderStatus.draft, nullable=False)
|
|
created_by: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
|
|
source_excel: Mapped[str] = mapped_column(String(1000), nullable=True)
|
|
notes: Mapped[str] = mapped_column(Text, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
|
|
submitted_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
processing_started_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
completed_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
rejected_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
estimated_price: Mapped[Decimal | None] = mapped_column(Numeric(12, 2), nullable=True)
|
|
|
|
template: Mapped["Template"] = relationship("Template", back_populates="orders")
|
|
created_by_user: Mapped["User"] = relationship("User", back_populates="orders", foreign_keys=[created_by])
|
|
items: Mapped[list["OrderItem"]] = relationship("OrderItem", back_populates="order", cascade="all, delete-orphan")
|
|
lines: Mapped[list["OrderLine"]] = relationship(
|
|
"OrderLine", back_populates="order", cascade="all, delete-orphan"
|
|
)
|