1 files changed, 62 insertions, 0 deletions
diff --git a/server/szurubooru/model/comment.py b/server/szurubooru/model/comment.py
new file mode 100644
index 0000000..55c1596
--- /dev/null
+++ b/server/szurubooru/model/comment.py
@@ -0,0 +1,62 @@
+from sqlalchemy import Column, Integer, DateTime, UnicodeText, ForeignKey
+from sqlalchemy.orm import relationship, backref
+from sqlalchemy.sql.expression import func
+from szurubooru.db import get_session
+from szurubooru.model.base import Base
+
+
+class CommentScore(Base):
+    __tablename__ = 'comment_score'
+
+    comment_id = Column(
+        'comment_id',
+        Integer,
+        ForeignKey('comment.id'),
+        nullable=False,
+        primary_key=True)
+    user_id = Column(
+        'user_id',
+        Integer,
+        ForeignKey('user.id'),
+        nullable=False,
+        primary_key=True,
+        index=True)
+    time = Column('time', DateTime, nullable=False)
+    score = Column('score', Integer, nullable=False)
+
+    comment = relationship('Comment')
+    user = relationship(
+        'User',
+        backref=backref('comment_scores', cascade='all, delete-orphan'))
+
+
+class Comment(Base):
+    __tablename__ = 'comment'
+
+    comment_id = Column('id', Integer, primary_key=True)
+    post_id = Column(
+        'post_id', Integer, ForeignKey('post.id'), nullable=False, index=True)
+    user_id = Column(
+        'user_id', Integer, ForeignKey('user.id'), nullable=True, index=True)
+    version = Column('version', Integer, default=1, nullable=False)
+    creation_time = Column('creation_time', DateTime, nullable=False)
+    last_edit_time = Column('last_edit_time', DateTime)
+    text = Column('text', UnicodeText, default=None)
+
+    user = relationship('User')
+    post = relationship('Post')
+    scores = relationship(
+        'CommentScore', cascade='all, delete-orphan', lazy='joined')
+
+    @property
+    def score(self) -> int:
+        return (
+            get_session()
+            .query(func.sum(CommentScore.score))
+            .filter(CommentScore.comment_id == self.comment_id)
+            .one()[0] or 0)
+
+    __mapper_args__ = {
+        'version_id_col': version,
+        'version_id_generator': False,
+    }