tuxbot-bot/cogs/utils/models/poll.py

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, BigInteger, JSON, ForeignKey, Boolean
from sqlalchemy.orm import relationship
from werkzeug.security import generate_password_hash, check_password_hash

Base = declarative_base()


class Poll(Base):
    __tablename__ = 'polls'

    id = Column(Integer, primary_key=True, autoincrement=True)
    channel_id = Column(BigInteger)
    message_id = Column(BigInteger)

    content = Column(JSON)
    is_anonymous = Column(Boolean)

    available_choices = Column(Integer)
    choice = relationship("Responses")


class Responses(Base):
    __tablename__ = 'responses'

    id = Column(Integer, primary_key=True, autoincrement=True)
    user = Column(BigInteger)

    poll_id = Column(Integer, ForeignKey('polls.id'))
    choice = Column(Integer)