119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
from datetime import datetime
|
|
from enum import Enum as Enumeration
|
|
from sqlalchemy import (
|
|
Enum,
|
|
Boolean,
|
|
Column,
|
|
DateTime,
|
|
ForeignKey,
|
|
Integer,
|
|
String,
|
|
JSON,
|
|
)
|
|
from sqlalchemy.orm import column_property, relationship
|
|
from services.db import Base, local_session
|
|
from orm.community import Community
|
|
from orm.reaction import Reaction
|
|
from orm.topic import Topic
|
|
from orm.author import Author
|
|
|
|
|
|
class ShoutTopic(Base):
|
|
__tablename__ = "shout_topic"
|
|
|
|
id = None # type: ignore
|
|
shout = Column(ForeignKey("shout.id"), primary_key=True, index=True)
|
|
topic = Column(ForeignKey("topic.id"), primary_key=True, index=True)
|
|
|
|
|
|
class ShoutReactionsFollower(Base):
|
|
__tablename__ = "shout_reactions_followers"
|
|
|
|
id = None # type: ignore
|
|
follower = Column(ForeignKey("author.id"), primary_key=True, index=True)
|
|
shout = Column(ForeignKey("shout.id"), primary_key=True, index=True)
|
|
auto = Column(Boolean, nullable=False, default=False)
|
|
createdAt = Column(
|
|
DateTime, nullable=False, default=datetime.now, comment="Created at"
|
|
)
|
|
deletedAt = Column(DateTime, nullable=True)
|
|
|
|
|
|
class ShoutAuthor(Base):
|
|
__tablename__ = "shout_author"
|
|
|
|
id = None # type: ignore
|
|
shout = Column(ForeignKey("shout.id"), primary_key=True, index=True)
|
|
author = Column(ForeignKey("author.id"), primary_key=True, index=True)
|
|
caption = Column(String, nullable=True, default="")
|
|
|
|
|
|
class ShoutCommunity(Base):
|
|
__tablename__ = "shout_community"
|
|
|
|
id = None # type: ignore
|
|
shout = Column(ForeignKey("shout.id"), primary_key=True, index=True)
|
|
community = Column(ForeignKey("community.id"), primary_key=True, index=True)
|
|
|
|
|
|
class ShoutVisibility(Enumeration):
|
|
AUTHORS = 0
|
|
COMMUNITY = 1
|
|
PUBLIC = 2
|
|
|
|
|
|
class Shout(Base):
|
|
__tablename__ = "shout"
|
|
|
|
createdAt = Column(DateTime, nullable=False, default=datetime.now)
|
|
updatedAt = Column(DateTime, nullable=True)
|
|
publishedAt = Column(DateTime, nullable=True)
|
|
deletedAt = Column(DateTime, nullable=True)
|
|
|
|
createdBy = Column(ForeignKey("author.id"), comment="Created By")
|
|
deletedBy = Column(ForeignKey("author.id"), nullable=True)
|
|
|
|
body = Column(String, nullable=False, comment="Body")
|
|
slug = Column(String, unique=True)
|
|
cover = Column(String, nullable=True, comment="Cover image url")
|
|
lead = Column(String, nullable=True)
|
|
description = Column(String, nullable=True)
|
|
title = Column(String, nullable=True)
|
|
subtitle = Column(String, nullable=True)
|
|
layout = Column(String, nullable=True)
|
|
media = Column(JSON, nullable=True)
|
|
|
|
authors = relationship(lambda: Author, secondary=ShoutAuthor.__tablename__)
|
|
topics = relationship(lambda: Topic, secondary=ShoutTopic.__tablename__)
|
|
communities = relationship(
|
|
lambda: Community, secondary=ShoutCommunity.__tablename__
|
|
)
|
|
reactions = relationship(lambda: Reaction)
|
|
|
|
viewsOld = Column(Integer, default=0)
|
|
viewsAckee = Column(Integer, default=0)
|
|
views = column_property(viewsOld + viewsAckee)
|
|
|
|
visibility = Column(Enum(ShoutVisibility), default=ShoutVisibility.AUTHORS)
|
|
|
|
# TODO: these field should be used or modified
|
|
lang = Column(String, nullable=False, default="ru", comment="Language")
|
|
mainTopic = Column(ForeignKey("topic.slug"), nullable=True)
|
|
versionOf = Column(ForeignKey("shout.id"), nullable=True)
|
|
oid = Column(String, nullable=True)
|
|
|
|
@staticmethod
|
|
def init_table():
|
|
with local_session() as session:
|
|
s = session.query(Shout).first()
|
|
if not s:
|
|
entry = {
|
|
"slug": "genesis-block",
|
|
"body": "",
|
|
"title": "Ничего",
|
|
"lang": "ru",
|
|
}
|
|
s = Shout.create(**entry)
|
|
session.add(s)
|
|
session.commit()
|