from sqlalchemy import *
from flask import g
from drama.__main__ import Base


class Image(Base):
	__tablename__ = "images"
	id = Column(BigInteger, primary_key=True)
	state = Column(String(8))
	number = Column(Integer)
	text = Column(String(64))
	deletehash = Column(String(64))
	
	@property
	def path(self):
		return f"/assets/images/platy.jpg"



def random_image():
	n=g.db.query(Image).count()
	return g.db.query(Image).order_by(Image.id.asc()).first()



class BadPic(Base):

	#Class for tracking fuzzy hashes of banned csam images

	__tablename__="badpics"
	id = Column(BigInteger, primary_key=True)
	description=Column(String(255), default=None)
	phash=Column(String(64))
	ban_reason=Column(String(64))
	ban_time=Column(Integer)