sneed

drama/classes/award.py

@@ -3,16 +3,18 @@
 from sqlalchemy import *
 from sqlalchemy.orm import relationship
 #from .mix_ins import *
-from drama.__main__ import Base
+from drama.__main__ import Base, app
 
 AWARDS = {
 	"ban": {
+		"kind": "ban",
 		"title": "1-Day Ban",
 		"description": "Ban the author for a day.",
 		"icon": "fas fa-gavel",
 		"color": "text-danger"
 	},
 	"shit": {
+		"kind": "shit",
 		"title": "Literal Shitpost",
 		"description": "Let OP know how much their post sucks ass.",
 		"icon": "fas fa-poop",
@@ -28,8 +30,8 @@ class AwardRelationship(Base):
 	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"))
+	submission_id = Column(Integer, ForeignKey("submissions.id"), default=None)
+	comment_id = Column(Integer, ForeignKey("comments.id"), default=None)
 	kind = Column(String(20))
 
 	user = relationship("User", primaryjoin="AwardRelationship.user_id==User.id", lazy="joined")
@@ -44,6 +46,18 @@ class AwardRelationship(Base):
 		lazy="joined"
 	)
 
+	@property
+	def given(self):
+		return bool(self.submission_id) or bool(self.comment_id)
+
 	@property
 	def type(self):
 		return AWARDS[self.kind]
+
+	@property
+	def title(self):
+		return self.type['title']
+
+	@property
+	def class_list(self):
+		return self.type['icon']+' '+self.type['color']