sneed

drama/classes/domains.py

@@ -0,0 +1,46 @@
+from sqlalchemy import *
+from drama.__main__ import Base
+
+reasons = {
+	1: "URL shorteners are not allowed.",
+	3: "Piracy is not allowed.",
+	4: "Sites hosting digitally malicious content are not allowed.",
+	5: "Spam",
+	6: "Doxxing is not allowed.",
+	7: "Sexualizing minors is strictly prohibited."
+}
+
+
+class Domain(Base):
+
+	__tablename__ = "domains"
+	id = Column(Integer, primary_key=True)
+	domain = Column(String)
+	can_submit = Column(Boolean, default=True)
+	can_comment = Column(Boolean, default=True)
+	reason = Column(Integer, default=0)
+	show_thumbnail = Column(Boolean, default=False)
+	embed_function = Column(String(64), default=None)
+	embed_template = Column(String(32), default=None)
+
+	@property
+	def reason_text(self):
+		return reasons.get(self.reason)
+
+	@property
+	def permalink(self):
+		return f"/admin/domain/{self.domain}"
+	
+
+
+class BadLink(Base):
+
+	__tablename__ = "badlinks"
+	id = Column(Integer, primary_key=True)
+	reason = Column(Integer)
+	link = Column(String(512))
+	autoban = Column(Boolean, default=False)
+
+	@property
+	def reason_text(self):
+		return reasons.get(self.reason)