force push

drama/classes/award.py

@@ -0,0 +1,49 @@
+#import requests
+#from os import environ
+from sqlalchemy import *
+from sqlalchemy.orm import relationship
+#from .mix_ins import *
+from drama.__main__ import Base, app
+
+AWARDS = {
+	"ban": {
+		"title": "1-Day Ban",
+		"description": "Ban the author for a day.",
+		"icon": "fas fa-gavel",
+		"color": "text-danger"
+	},
+	"shit": {
+		"title": "Literal Shitpost",
+		"description": "Let OP know how much their post sucks ass.",
+		"icon": "fas fa-poop",
+		"color": "text-black-50"
+	}
+}
+
+
+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"), 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")
+	post = relationship(
+		"Submission",
+		primaryjoin="AwardRelationship.submission_id==Submission.id",
+		lazy="joined"
+	)
+	comment = relationship(
+		"Comment",
+		primaryjoin="AwardRelationship.comment_id==Comment.id",
+		lazy="joined"
+	)
+
+	@property
+	def type(self):
+		return AWARDS[self.kind]