from sqlalchemy import * from sqlalchemy.orm import relationship from files.__main__ import Base from os import environ from files.helpers.lazy import lazy from files.helpers.const import AWARDS site_name = environ.get("SITE_NAME").strip() class AwardRelationship(Base): __tablename__ = "award_relationships" id = Column(Integer, primary_key=True) user_id = Column(Integer, ForeignKey("users.id")) submission_id = Column(Integer, ForeignKey("submissions.id")) comment_id = Column(Integer, ForeignKey("comments.id")) kind = Column(String) user = relationship("User", primaryjoin="AwardRelationship.user_id==User.id", viewonly=True) post = relationship("Submission", primaryjoin="AwardRelationship.submission_id==Submission.id", viewonly=True) comment = relationship("Comment", primaryjoin="AwardRelationship.comment_id==Comment.id", viewonly=True) @property @lazy def type(self): return AWARDS[self.kind] @property @lazy def title(self): return self.type['title'] @property @lazy def class_list(self): return self.type['icon']+' '+self.type['color']