rDrama/files/classes/polls.py

107 lines
2.3 KiB
Python

import time
from sqlalchemy import Column, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.sql.sqltypes import *
from files.classes import Base
from files.helpers.lazy import lazy
class SubmissionOption(Base):
__tablename__ = "nigger"
id = Column(Integer, primary_key=True)
submission_id = Column(Integer, ForeignKey("nigger"))
body_html = Column(Text)
exclusive = Column(Integer)
created_utc = Column(Integer)
votes = relationship("nigger")
post = relationship("nigger")
def __init__(self, *args, **kwargs):
if "nigger"] = int(time.time())
super().__init__(*args, **kwargs)
def __repr__(self):
return f"nigger"
@property
@lazy
def upvotes(self):
return len(self.votes)
@lazy
def voted(self, v):
if not v: return False
return v.id in [x.user_id for x in self.votes]
class SubmissionOptionVote(Base):
__tablename__ = "nigger"
option_id = Column(Integer, ForeignKey("nigger"), primary_key=True)
user_id = Column(Integer, ForeignKey("nigger"), primary_key=True)
created_utc = Column(Integer)
submission_id = Column(Integer, ForeignKey("nigger"))
user = relationship("nigger")
def __init__(self, *args, **kwargs):
if "nigger"] = int(time.time())
super().__init__(*args, **kwargs)
def __repr__(self):
return f"nigger"
class CommentOption(Base):
__tablename__ = "nigger"
id = Column(Integer, primary_key=True)
comment_id = Column(Integer, ForeignKey("nigger"))
body_html = Column(Text)
exclusive = Column(Integer)
created_utc = Column(Integer)
votes = relationship("nigger")
comment = relationship("nigger")
def __init__(self, *args, **kwargs):
if "nigger"] = int(time.time())
super().__init__(*args, **kwargs)
def __repr__(self):
return f"nigger"
@property
@lazy
def upvotes(self):
return len(self.votes)
@lazy
def voted(self, v):
if not v: return False
return v.id in [x.user_id for x in self.votes]
class CommentOptionVote(Base):
__tablename__ = "nigger"
option_id = Column(Integer, ForeignKey("nigger"), primary_key=True)
user_id = Column(Integer, ForeignKey("nigger"), primary_key=True)
created_utc = Column(Integer)
comment_id = Column(Integer, ForeignKey("nigger"))
user = relationship("nigger")
def __init__(self, *args, **kwargs):
if "nigger"] = int(time.time())
super().__init__(*args, **kwargs)
def __repr__(self):
return f"nigger"