48 lines
2.5 KiB
Python
48 lines
2.5 KiB
Python
"""Модели проектов и заявок."""
|
|
|
|
import uuid
|
|
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import Column, DateTime, Enum, Float, ForeignKey, Integer, String, Text, func
|
|
from sqlalchemy.dialects.postgresql import UUID as PG_UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class Project(Base):
|
|
__tablename__ = "projects"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
client_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
|
|
title: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
description: Mapped[str] = mapped_column(Text, nullable=False)
|
|
category: Mapped[str | None] = mapped_column(String(100))
|
|
required_skills: Mapped[list] = mapped_column("required_skills", postgresql.JSONB, default=list)
|
|
skill_embeddings: Mapped[list | None] = mapped_column("skill_embeddings") # pgvector
|
|
budget_min: Mapped[float | None] = mapped_column(Float(precision=10, scale=2))
|
|
budget_max: Mapped[float | None] = mapped_column(Float(precision=10, scale=2))
|
|
status: Mapped[str] = mapped_column(Enum("open", "in_progress", "completed", "cancelled"), default="open")
|
|
deadline: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), onupdate=func.now())
|
|
|
|
# Relationships
|
|
client = relationship("User", foreign_keys=[client_id])
|
|
proposals: Mapped[list["Proposal"]] = relationship(back_populates="project")
|
|
|
|
|
|
class Proposal(Base):
|
|
__tablename__ = "proposals"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
project_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), ForeignKey("projects.id"), nullable=False)
|
|
freelancer_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
|
|
cover_letter: Mapped[str | None] = mapped_column(Text)
|
|
proposed_price: Mapped[float | None] = mapped_column(Float(precision=10, scale=2))
|
|
estimated_days: Mapped[int | None] = mapped_column(Integer)
|
|
status: Mapped[str] = mapped_column(Enum("pending", "accepted", "rejected"), default="pending")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
project: Mapped["Project"] = relationship(back_populates="proposals")
|