core/orm/author.py

from datetime import datetime
from sqlalchemy import JSON as JSONType
from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String
from sqlalchemy.orm import relationship
from services.db import Base, local_session


class AuthorRating(Base):
    __tablename__ = "author_rating"

    id = None  # type: ignore
    rater = Column(ForeignKey("author.id"), primary_key=True, index=True)
    author = Column(ForeignKey("author.id"), primary_key=True, index=True)
    value = Column(Integer)

    @staticmethod
    def init_table():
        pass


class AuthorFollower(Base):
    __tablename__ = "author_follower"

    id = None  # type: ignore
    follower = Column(ForeignKey("author.id"), primary_key=True, index=True)
    author = Column(ForeignKey("author.id"), primary_key=True, index=True)
    createdAt = Column(DateTime, nullable=False, default=datetime.now)
    auto = Column(Boolean, nullable=False, default=False)


class Author(Base):
    __tablename__ = "author"

    user = Column(Integer, nullable=False)  # unbounded link with authorizer's User type
    bio = Column(String, nullable=True, comment="Bio")  # status description
    about = Column(String, nullable=True, comment="About")  # long and formatted
    userpic = Column(String, nullable=True, comment="Userpic")
    name = Column(String, nullable=True, comment="Display name")
    slug = Column(String, unique=True, comment="Author's slug")
    muted = Column(Boolean, default=False)
    createdAt = Column(DateTime, nullable=False, default=datetime.now)
    lastSeen = Column(DateTime, nullable=False, default=datetime.now)  # Td se 0e
    deletedAt = Column(DateTime, nullable=True, comment="Deleted at")
    links = Column(JSONType, nullable=True, comment="Links")
    ratings = relationship(AuthorRating, foreign_keys=AuthorRating.author)

    @staticmethod
    def init_table():
        with local_session() as session:
            default = session.query(Author).filter(Author.slug == "anonymous").first()
            if not default:
                default_dict = {
                    "user": 0,
                    "name": "Аноним",
                    "slug": "anonymous",
                }
                default = Author.create(**default_dict)
                session.add(default)
                discours_dict = {
                    "user": 1,
                    "name": "Дискурс",
                    "slug": "discours",
                }
                discours = Author.create(**discours_dict)
                session.add(discours)
                session.commit()
            Author.default_author = default