sasalatart/triviapp

import pickle
from wtforms import Form, TextField, SelectField, HiddenField, validators
from db import r


class Question:
    def __init__(self, category, question, option_a, option_b, option_c, option_d, answer):
        self.category = category
        self.question = question
        self.option_a = option_a
        self.option_b = option_b
        self.option_c = option_c
        self.option_d = option_d
        self.answer = answer

    def save(self):
        self.id = int(r.get('question:next_id') or 1)
        r.sadd('question:ids', self.id)
        r.set('question:' + str(self.id), pickle.dumps(self))
        r.set('question:next_id', int(r.get('question:next_id') or 1) + 1)

    @staticmethod
    def find(id):
        if r.sismember('question:ids', str(id)):
            return pickle.loads(r.get('question:' + str(id)))
        else:
            return None

    @staticmethod
    def random_id():
        return r.srandmember('question:ids')

    @staticmethod
    def size():
        return r.scard('question:ids')


class NewQuestionForm(Form):
    category = SelectField('Category',
                           choices=[('animals', 'Animals'),
                                    ('arts', 'Arts'),
                                    ('computing', 'Computing'),
                                    ('entertainment', 'Entertainment'),
                                    ('geography', 'Geography'),
                                    ('government', 'Government'),
                                    ('history', 'History'),
                                    ('mathematics', 'Mathematics'),
                                    ('other', 'Other'),
                                    ('people', 'People'),
                                    ('phrases', 'Phrases'),
                                    ('politics', 'Politics'),
                                    ('science', 'Science'),
                                    ('words', 'Words')],
                           validators=[validators.Required()])

    question = TextField('Question',
                         validators=[validators.Required(),
                                     validators.Length(min=3, max=200)])
    option_a = TextField('Option A',
                         validators=[validators.Required(),
                                     validators.Length(min=1, max=200)])
    option_b = TextField('Option B',
                         validators=[validators.Required(),
                                     validators.Length(min=1, max=200)])
    option_c = TextField('Option C',
                         validators=[validators.Required(),
                                     validators.Length(min=1, max=200)])
    option_d = TextField('Option D',
                         validators=[validators.Required(),
                                     validators.Length(min=1, max=200)])
    answer = SelectField('Answer',
                         choices=[('a', 'A'),
                                  ('b', 'B'),
                                  ('c', 'C'),
                                  ('d', 'D')],
                         validators=[validators.Required()])


class AnswerForm(Form):
    answer = HiddenField('Answer',
                         validators=[validators.Required(),
                                     validators.AnyOf(['a', 'b', 'c', 'd'],
                                                      message=u'Invalid value, must be one of: %(values)s',
                                                      values_formatter=None)])