SmartUp Developer
74 lines
3.9 KiB
Python
74 lines
3.9 KiB
Python
from datetime import datetime, timezone
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import Boolean, DateTime, ForeignKey, Index, Integer, String, Text, text
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class Website(Base):
|
|
__tablename__ = "websites"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
site_type: Mapped[str] = mapped_column(String(32), default="sub2api")
|
|
base_url: Mapped[str] = mapped_column(String(512), nullable=False)
|
|
api_prefix: Mapped[str] = mapped_column(String(128), default="/api/v1/admin")
|
|
auth_type: Mapped[str] = mapped_column(String(32), default="api_key")
|
|
auth_config_json: Mapped[str] = mapped_column(Text, default="{}")
|
|
groups_endpoint: Mapped[str] = mapped_column(String(256), default="/groups")
|
|
group_update_endpoint: Mapped[str] = mapped_column(String(256), default="/groups/{id}")
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
auto_sync_enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
timeout_seconds: Mapped[int] = mapped_column(Integer, default=30)
|
|
last_status: Mapped[str] = mapped_column(String(32), default="unknown")
|
|
last_checked_at: Mapped[Optional[datetime]] = mapped_column(DateTime, nullable=True)
|
|
last_error: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc)
|
|
)
|
|
|
|
|
|
class WebsiteGroupBinding(Base):
|
|
__tablename__ = "website_group_bindings"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
|
|
website_id: Mapped[int] = mapped_column(Integer, ForeignKey("websites.id", ondelete="CASCADE"), index=True)
|
|
target_group_id: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
target_group_name: Mapped[str] = mapped_column(String(255), default="")
|
|
source_groups_json: Mapped[str] = mapped_column(Text, default="[]")
|
|
percent: Mapped[str] = mapped_column(String(32), default="0")
|
|
algorithm: Mapped[str] = mapped_column(String(64), default="max_plus_percent")
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc)
|
|
)
|
|
|
|
|
|
class WebsiteSyncLog(Base):
|
|
__tablename__ = "website_sync_logs"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
|
|
website_id: Mapped[int] = mapped_column(Integer, ForeignKey("websites.id", ondelete="CASCADE"), index=True)
|
|
binding_id: Mapped[Optional[int]] = mapped_column(
|
|
Integer, ForeignKey("website_group_bindings.id", ondelete="SET NULL"), nullable=True, index=True
|
|
)
|
|
target_group_id: Mapped[str] = mapped_column(String(255), default="")
|
|
target_group_name: Mapped[str] = mapped_column(String(255), default="")
|
|
algorithm: Mapped[str] = mapped_column(String(64), default="")
|
|
percent: Mapped[str] = mapped_column(String(32), default="0")
|
|
source_rates_json: Mapped[str] = mapped_column(Text, default="[]")
|
|
old_rate: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
|
new_rate: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
|
status: Mapped[str] = mapped_column(String(16), nullable=False)
|
|
message: Mapped[str] = mapped_column(Text, default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.now(timezone.utc), index=True)
|
|
|
|
__table_args__ = (
|
|
Index("ix_sync_website_created", "website_id", text("created_at DESC")),
|
|
Index("ix_sync_binding_created", "binding_id", text("created_at DESC")),
|
|
)
|