project-callisto/callisto-core

"""
mocks are fake models, used in place of actual models when rendering forms
"""

from . import fields


class MockPage(object):
    pk = None
    id = None

    def __init__(self, data):
        try:
            self.section = data[0].get("section", 1)
        except BaseException:
            self.section = 1
        self.mock_questions = self._create_questions(data)

    def _create_questions(self, data):
        questions = []
        for question_data in data:
            question = MockQuestion(question_data)
            questions.append(question)
        return questions


class MockQuestion(object):
    def __init__(self, data):
        self.pk = self.id = data.get("id")
        self.text = data.get("question_text")
        self.descriptive_text = data.get("descriptive_text")
        self.section = data.get("section")
        self.position = data.get("position", 0)
        self.serialized = self.data = data
        self.choices = [
            MockChoice(choice_data) for choice_data in data.get("choices", [])
        ]

    @property
    def type(self):
        try:
            return self.data.get("type").lower()
        except BaseException:
            return ""

    @property
    def field_id(self):
        return "question_" + str(self.id)

    @property
    def choices_data_array(self):
        return [choice.data for choice in self.choices]

    @property
    def choices_pk_text_array(self):
        return [(choice.pk, choice.text) for choice in self.choices]

    def make_field(self):
        field_generator = getattr(
            fields.QuestionField,  # from the QuestionFields object
            self.type,  # get the field that corresponds to the question type
            fields.QuestionField.singlelinetext,  # otherwise get a singlelinetext field
        )
        return field_generator(self)


class MockChoice(object):
    def __init__(self, data):
        self.pk = self.id = data.get("pk")
        self.text = data.get("text")
        self.position = data.get("position", 0)
        self.data = data