core/orm/community.py

from sqlalchemy import Column, DateTime, ForeignKey, String, func

from base.orm import Base, local_session


class CommunityFollower(Base):
    __tablename__ = "community_followers"

    id = None
    follower: Column = Column(ForeignKey("user.id"), primary_key=True)
    community: Column = Column(ForeignKey("community.id"), primary_key=True)
    joinedAt = Column(
        DateTime(timezone=True), nullable=False, server_default=func.now(), comment="Created at"
    )
    # role = Column(ForeignKey(Role.id), nullable=False, comment="Role for member")


class Community(Base):
    __tablename__ = "community"

    name = Column(String, nullable=False, comment="Name")
    slug = Column(String, nullable=False, unique=True, comment="Slug")
    desc = Column(String, nullable=False, default="")
    pic = Column(String, nullable=False, default="")
    createdAt = Column(
        DateTime(timezone=True), nullable=False, server_default=func.now(), comment="Created at"
    )

    @staticmethod
    def init_table():
        with local_session() as session:
            d = session.query(Community).filter(Community.slug == "discours").first()
            if not d:
                d = Community.create(name="Дискурс", slug="discours")
                session.add(d)
                session.commit()
            Community.default_community = d
            print("[orm] default community id: %s" % d.id)