open-law/app/models/interpretation.py

59 lines
1.6 KiB
Python

from flask_login import current_user
from app import db, models as m
from app.models.utils import BaseModel
class Interpretation(BaseModel):
__tablename__ = "interpretations"
text = db.Column(db.Text, unique=False, nullable=False)
plain_text = db.Column(db.Text, unique=False)
approved = db.Column(db.Boolean, default=False)
marked = db.Column(db.Boolean, default=False)
# Foreign keys
user_id = db.Column(db.ForeignKey("users.id"))
section_id = db.Column(db.ForeignKey("sections.id"))
# Relationships
user = db.relationship("User")
section = db.relationship("Section")
comments = db.relationship("Comment", viewonly=True, order_by="desc(Comment.id)")
votes = db.relationship("InterpretationVote", viewonly=True)
tags = db.relationship(
"Tag",
secondary="interpretation_tags",
back_populates="interpretations",
)
@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
@property
def active_comments(self):
return [comment for comment in self.comments if not comment.is_deleted]
@property
def book(self) -> m.Book:
return self.section.version.book
def __repr__(self):
return f"<Interpretation: {self.id}>"