init commit

app/db/__init__.py

@@ -0,0 +1 @@
+

app/db/base.py

@@ -0,0 +1,3 @@
+from sqlalchemy.orm import declarative_base
+
+Base = declarative_base()

app/db/models.py

@@ -0,0 +1,56 @@
+from datetime import datetime
+from sqlalchemy import Column, Integer, BigInteger, String, DateTime, ForeignKey, Text, Boolean
+from sqlalchemy.orm import relationship
+from app.db.base import Base
+
+class User(Base):
+    __tablename__ = "users"
+    id = Column(Integer, primary_key=True)
+    tg_id = Column(BigInteger, unique=True, nullable=False, index=True)
+    name = Column(String(255))
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+
+class Chat(Base):
+    __tablename__ = "chats"
+    id = Column(Integer, primary_key=True)
+    chat_id = Column(BigInteger, unique=True, nullable=False, index=True)
+    type = Column(String(32))          # "group" | "supergroup" | "channel"
+    title = Column(String(255))
+    owner_user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
+    can_post = Column(Boolean, default=False, nullable=False)
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+
+    owner = relationship("User")
+
+class Draft(Base):
+    __tablename__ = "drafts"
+    id = Column(Integer, primary_key=True)
+    user_id = Column(Integer, ForeignKey("users.id"), index=True)
+    text = Column(Text, nullable=True)
+    status = Column(String(16), default="editing", index=True)  # editing | ready | sent
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+
+    user = relationship("User")
+    media = relationship("DraftMedia", cascade="all, delete-orphan", back_populates="draft")
+
+class DraftMedia(Base):
+    __tablename__ = "draft_media"
+    id = Column(Integer, primary_key=True)
+    draft_id = Column(Integer, ForeignKey("drafts.id"), index=True)
+    kind = Column(String(16))  # photo | video | animation
+    file_id = Column(String(255))
+    """Ordering inside album."""
+    order = Column(Integer, default=0)
+
+    draft = relationship("Draft", back_populates="media")
+
+class Delivery(Base):
+    __tablename__ = "deliveries"
+    id = Column(Integer, primary_key=True)
+    draft_id = Column(Integer, ForeignKey("drafts.id"), index=True)
+    chat_id = Column(BigInteger, index=True)
+    status = Column(String(16), default="new", index=True)  # new | sent | failed
+    error = Column(Text, nullable=True)
+    message_ids = Column(Text, nullable=True)   # CSV for album parts
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)

app/db/session.py

@@ -0,0 +1,10 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from app.config import load_config
+
+_config = load_config()
+engine = create_engine(_config.database_url, pool_pre_ping=True, future=True)
+SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False, future=True)
+
+def get_session():
+    return SessionLocal()