rDrama/drama/classes/domains.py

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)
	embed_function = Column(String(64))
	embed_template = Column(String(32))

	@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)