evo-sync/web/models/connections.py

from sqlalchemy import (
    Boolean, Column, DateTime, ForeignKey, Index, Integer,
    Numeric, String, Text, UniqueConstraint, func,
)

from web.database import Base


class EvotorConnection(Base):
    __tablename__ = "evotor_connections"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=True)
    evotor_user_id = Column(String(255), nullable=True)
    access_token = Column(Text, nullable=False)
    api_token = Column(String(255), nullable=True)  # token we return to Evotor in webhook responses
    store_id = Column(String(255), nullable=True)
    store_name = Column(String(255), nullable=True)
    refresh_token = Column(Text, nullable=True)
    token_expires_at = Column(DateTime, nullable=True)
    is_online = Column(Boolean, nullable=False, default=False)
    last_checked_at = Column(DateTime, nullable=True)
    connected_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())

    __table_args__ = (
        UniqueConstraint("user_id", name="ix_evotor_connections_user_id"),
        UniqueConstraint("evotor_user_id", name="ix_evotor_connections_evotor_user_id"),
    )


class VkConnection(Base):
    __tablename__ = "vk_connections"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    access_token = Column(Text, nullable=False)
    vk_user_id = Column(String(50), nullable=True)
    first_name = Column(String(255), nullable=True)
    last_name = Column(String(255), nullable=True)
    is_online = Column(Boolean, nullable=False, default=False)
    last_checked_at = Column(DateTime, nullable=True)
    connected_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())

    __table_args__ = (
        UniqueConstraint("user_id", name="ix_vk_connections_user_id"),
    )


class SyncConfig(Base):
    __tablename__ = "sync_configs"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    is_enabled = Column(Boolean, nullable=False, default=False)
    confirmed_at = Column(DateTime, nullable=True)
    created_at = Column(DateTime, nullable=False, server_default=func.now())
    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())

    __table_args__ = (
        UniqueConstraint("user_id", name="ix_sync_configs_user_id"),
    )


class SyncFilter(Base):
    __tablename__ = "sync_filters"

    id = Column(Integer, primary_key=True, autoincrement=True)
    sync_config_id = Column(Integer, ForeignKey("sync_configs.id", ondelete="CASCADE"), nullable=False)
    entity_type = Column(String(20), nullable=False)
    entity_id = Column(String(255), nullable=False)
    entity_name = Column(String(255), nullable=True)
    filter_mode = Column(String(10), nullable=False)
    parent_entity_id = Column(String(255), nullable=True)
    created_at = Column(DateTime, nullable=False, server_default=func.now())

    __table_args__ = (
        UniqueConstraint("sync_config_id", "entity_type", "entity_id",
                         name="uq_sync_filters_config_type_entity"),
    )


class VkCachedAlbum(Base):
    __tablename__ = "vk_cached_albums"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    vk_group_id = Column(String(50), nullable=False)
    album_id = Column(String(50), nullable=False)
    title = Column(String(255), nullable=False)
    count = Column(Integer, nullable=True)
    fetched_at = Column(DateTime, nullable=False)

    __table_args__ = (
        UniqueConstraint("user_id", "vk_group_id", "album_id", name="uq_vk_cached_albums"),
        Index("ix_vk_cached_albums_user_group", "user_id", "vk_group_id"),
    )


class VkCachedProduct(Base):
    __tablename__ = "vk_cached_products"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    vk_group_id = Column(String(50), nullable=False)
    vk_product_id = Column(String(50), nullable=False)
    album_id = Column(String(50), nullable=True)
    name = Column(String(255), nullable=False)
    description = Column(Text, nullable=True)
    price = Column(Numeric(12, 2), nullable=True)
    availability = Column(Integer, nullable=True)
    thumb_url = Column(String(1024), nullable=True)
    fetched_at = Column(DateTime, nullable=False)

    __table_args__ = (
        UniqueConstraint("user_id", "vk_group_id", "vk_product_id", name="uq_vk_cached_products"),
        Index("ix_vk_cached_products_user_group_album", "user_id", "vk_group_id", "album_id"),
    )


class CachedStore(Base):
    __tablename__ = "cached_stores"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    evotor_id = Column(String(255), nullable=False)
    name = Column(String(255), nullable=False)
    address = Column(String(500), nullable=True)
    fetched_at = Column(DateTime, nullable=False)

    __table_args__ = (
        UniqueConstraint("user_id", "evotor_id", name="uq_cached_stores_user_evotor"),
        Index("ix_cached_stores_user_id", "user_id"),
    )


class CachedGroup(Base):
    __tablename__ = "cached_groups"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    evotor_id = Column(String(255), nullable=False)
    store_evotor_id = Column(String(255), nullable=False)
    name = Column(String(255), nullable=False)
    fetched_at = Column(DateTime, nullable=False)

    __table_args__ = (
        UniqueConstraint("user_id", "evotor_id", name="uq_cached_groups_user_evotor"),
        Index("ix_cached_groups_user_store", "user_id", "store_evotor_id"),
    )


class CachedProduct(Base):
    __tablename__ = "cached_products"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    evotor_id = Column(String(255), nullable=False)
    store_evotor_id = Column(String(255), nullable=False)
    group_evotor_id = Column(String(255), nullable=True)
    name = Column(String(255), nullable=False)
    price = Column(Numeric(12, 2), nullable=True)
    quantity = Column(Numeric(12, 3), nullable=True)
    measure_name = Column(String(20), nullable=True)
    article_number = Column(String(100), nullable=True)
    allow_to_sell = Column(Boolean, nullable=True)
    fetched_at = Column(DateTime, nullable=False)
    synced_at = Column(DateTime, nullable=True)

    __table_args__ = (
        UniqueConstraint("user_id", "evotor_id", name="uq_cached_products_user_evotor"),
        Index("ix_cached_products_user_store_group", "user_id", "store_evotor_id", "group_evotor_id"),
    )