mirror of https://github.com/logos-co/open-law.git
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from datetime import datetime
|
|
|
|
from flask_login import current_user
|
|
|
|
from app import db
|
|
from app.models.utils import BaseModel
|
|
|
|
|
|
class Interpretation(BaseModel):
|
|
__tablename__ = "interpretations"
|
|
|
|
label = db.Column(db.String(256), unique=False, nullable=False)
|
|
text = db.Column(db.Text, unique=False, nullable=False)
|
|
marked = db.Column(db.Boolean, default=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.now)
|
|
|
|
# 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)
|
|
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]
|
|
|
|
def __repr__(self):
|
|
return f"<{self.id}: {self.label}>"
|