from sqlalchemy import *
from sqlalchemy.orm import relationship
from files.classes.base import CreatedBase
from files.helpers.lazy import lazy
from files.helpers.config.const import *

class Flag(CreatedBase):
	__tablename__ = "flags"

	id = Column(Integer, primary_key=True)
	post_id = Column(Integer, ForeignKey("submissions.id"), nullable=False)
	user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
	reason = Column(String)

	Index('flag_user_idx', user_id)

	user = relationship("User", primaryjoin = "Flag.user_id == User.id", uselist = False, viewonly=True)

	def __repr__(self):
		return f"<{self.__class__.__name__}(id={self.id})>"

	@lazy
	def realreason(self, v):
		return self.reason


class CommentFlag(CreatedBase):
	__tablename__ = "commentflags"

	id = Column(Integer, primary_key=True)
	comment_id = Column(Integer, ForeignKey("comments.id"), nullable=False)
	user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
	reason = Column(String)

	Index('cflag_user_idx', user_id)

	user = relationship("User", primaryjoin = "CommentFlag.user_id == User.id", uselist = False, viewonly=True)

	def __repr__(self):
		return f"<{self.__class__.__name__}(id={self.id})>"

	@lazy
	def realreason(self, v):
		return self.reason