database/models.py - main - facex/friends_target

database/models.py
50 строк2 KB
axlerya
fix: big fix vk
22 июл 2025, 12:57
22 июл 2025, 12:57e21475a
from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy import String, Integer, ForeignKey, BigInteger
from sqlalchemy.ext.declarative import declarative_base
from typing import List

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id: Mapped[int] = mapped_column(primary_key=True)
    tg_id: Mapped[int] = mapped_column(BigInteger, unique=True, nullable=True)
    vk_id: Mapped[int] = mapped_column(BigInteger, unique=True, nullable=True)
    vk_access_token: Mapped[str] = mapped_column(String, nullable=True)
    

    cabinets: Mapped[List["Cabinet"]] = relationship(back_populates="user", cascade="all, delete-orphan")


class Cabinet(Base):
    __tablename__ = 'cabinets'

    id: Mapped[int] = mapped_column(primary_key=True)
    ads_id: Mapped[str] = mapped_column(String, nullable=False, unique=True)
    name: Mapped[str] = mapped_column(String, nullable=False)

    tg_id: Mapped[int] = mapped_column(ForeignKey('users.tg_id'), nullable=False)
    user: Mapped["User"] = relationship(back_populates="cabinets")

    compains: Mapped[List["Compain"]] = relationship(back_populates="cabinet", cascade="all, delete-orphan")


class Compain(Base):
    __tablename__ = 'compains'

    id: Mapped[int] = mapped_column(primary_key=True)
    ads_id: Mapped[str] = mapped_column(ForeignKey('cabinets.ads_id'), nullable=False)
    ads_compain_id: Mapped[str] = mapped_column(String, nullable=False)
    name: Mapped[str] = mapped_column(String, nullable=False)

    cabinet: Mapped["Cabinet"] = relationship(back_populates="compains")

class ProcessingState(Base):
    __tablename__ = "processing_state"

    id: Mapped[int] = mapped_column(primary_key=True)
    tg_id: Mapped[int] = mapped_column(BigInteger, ForeignKey('users.tg_id'), unique=True)
    
    last_folder: Mapped[str] = mapped_column(String, nullable=True)
    last_photo: Mapped[str] = mapped_column(String, nullable=True)