mirror of https://github.com/logos-co/open-law.git
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from flask_login import current_user
|
|
|
|
from app import db
|
|
from app.models.utils import BaseModel
|
|
|
|
|
|
class Comment(BaseModel):
|
|
__tablename__ = "comments"
|
|
|
|
# need to redeclare id to use it in the parent relationship
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
text = db.Column(db.Text, unique=False, nullable=False)
|
|
approved = db.Column(db.Boolean, default=False)
|
|
marked = db.Column(db.Boolean, default=False)
|
|
edited = db.Column(db.Boolean, default=False)
|
|
|
|
# Foreign keys
|
|
user_id = db.Column(db.ForeignKey("users.id"))
|
|
parent_id = db.Column(db.ForeignKey("comments.id"))
|
|
interpretation_id = db.Column(db.ForeignKey("interpretations.id"))
|
|
|
|
# Relationships
|
|
user = db.relationship("User")
|
|
children = db.relationship(
|
|
"Comment", backref=db.backref("parent", remote_side=[id]), viewonly=True
|
|
)
|
|
interpretation = db.relationship("Interpretation")
|
|
votes = db.relationship("CommentVote")
|
|
tags = db.relationship(
|
|
"Tag",
|
|
secondary="comment_tags",
|
|
back_populates="comments",
|
|
)
|
|
|
|
@property
|
|
def vote_count(self):
|
|
count = 0
|
|
|
|
for vote in self.votes:
|
|
if vote.positive:
|
|
count += 1
|
|
continue
|
|
count -= 1
|
|
|
|
return count
|
|
|
|
@property
|
|
def current_user_vote(self):
|
|
for vote in self.votes:
|
|
if vote.user_id == current_user.id:
|
|
return vote.positive
|
|
return None
|
|
|
|
def __repr__(self):
|
|
return f"<{self.id}: {self.text[:20]}>"
|