core/models.py - master - Ksq/fastapi_app

core/models.py
48 строк2 KB
KSQ-web
Добавление репозитория
20 апр 2025, 16:29
20 апр 2025, 16:29e640632
from sqlalchemy import String, Integer, ForeignKey, Text, DateTime
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
from sqlalchemy.ext.asyncio import AsyncAttrs
import datetime


class Base(AsyncAttrs, DeclarativeBase):
    __abstract__ = True


class Users(Base):
    __tablename__ = "users"
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    login: Mapped[str] = mapped_column(String(100), nullable=False)
    password: Mapped[str] = mapped_column(String(100))
    phone_number: Mapped[str] = mapped_column(String(15))
    email: Mapped[str] = mapped_column(String(50))
    car_number: Mapped[str] = mapped_column(String(15))
    group_id: Mapped[int] = mapped_column(Integer, nullable=False)
    avatar_url: Mapped[str] = mapped_column(String(255))
    orders = relationship("Orders", back_populates="staff")


class Orders(Base):
    __tablename__ = "orders"
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    car_number: Mapped[str] = mapped_column(String(15))
    start_address: Mapped[str] = mapped_column(String(100))
    start_latitude: Mapped[str] = mapped_column(String(20))
    start_longitude: Mapped[str] = mapped_column(String(20))
    order_author: Mapped[int] = mapped_column(Integer, nullable=False)
    finish_address: Mapped[str] = mapped_column(String(100))
    finish_latitude: Mapped[str] = mapped_column(String(20))
    finish_longitude: Mapped[str] = mapped_column(String(20))
    order_status: Mapped[int] = mapped_column(Integer, nullable=False)
    staff_author: Mapped[int] = mapped_column(Integer, ForeignKey('users.id'), nullable=False)
    reason: Mapped[str] = mapped_column(String(500))
    staff = relationship("Users", back_populates="orders")


class PMMessage(Base):
    __tablename__ = "pm_message"
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    send_author: Mapped[int] = mapped_column(Integer, nullable=False)
    from_message: Mapped[int] = mapped_column(Integer, nullable=False)
    text_message: Mapped[str] = mapped_column(Text, nullable=True)
    order_id: Mapped[int] = mapped_column(Integer, ForeignKey('orders.id'), nullable=False)
    message_time: Mapped[datetime.datetime] = mapped_column(DateTime, nullable=False)