drafts-orm-struct

orm/draft.py

@@ -1,22 +1,31 @@
 from datetime import datetime
 
-from sqlalchemy import Column, ForeignKey, DateTime, String
+from sqlalchemy import Boolean, Column, ForeignKey, DateTime, String
 from sqlalchemy.orm import relationship
 from base.orm import Base
 from orm.user import User
+from orm.topic import Topic
+
+
+class DraftTopic(Base):
+    __tablename__ = "draft_topic"
+
+    id = None  # type: ignore
+    collab = Column(ForeignKey("draft_collab.id"), primary_key=True)
+    topic = Column(ForeignKey("topic.id"), primary_key=True)
 
 
 class DraftAuthor(Base):
-    __tablename__ = "collab_author"
+    __tablename__ = "draft_author"
 
     id = None  # type: ignore
-    collab = Column(ForeignKey("collab.id"), primary_key=True)
+    collab = Column(ForeignKey("draft_collab.id"), primary_key=True)
     author = Column(ForeignKey("user.id"), primary_key=True)
-    # accepted = Column(Boolean, default=False)
+    accepted = Column(Boolean, default=False)
 
 
 class DraftCollab(Base):
-    __tablename__ = "draftcollab"
+    __tablename__ = "draft_collab"
 
     slug = Column(String, nullable=True, comment="Slug")
     title = Column(String, nullable=True, comment="Title")
@@ -25,8 +34,7 @@ class DraftCollab(Base):
     body = Column(String, nullable=True, comment="Body")
     cover = Column(String, nullable=True, comment="Cover")
     authors = relationship(lambda: User, secondary=DraftAuthor.__tablename__)
-    topics = relationship(lambda: Topic, secondary=ShoutTopic.__tablename__)
-    invites = relationship(lambda: User, secondary=CollabInvited.__tablename__)
+    topics = relationship(lambda: Topic, secondary=DraftTopic.__tablename__)
     createdAt = Column(DateTime, default=datetime.now, comment="Created At")
     updatedAt = Column(DateTime, default=datetime.now, comment="Updated At")
     chat = Column(String, unique=True, nullable=True)