from otree.api import Currency as c, currency_range
from . import models
from ._builtin import Page, WaitPage
from .models import Constants
import random


def vars_for_all_templates(self):
    row = [[random.randint(0, 1) for i in range(self.player.matrix)] for n in range(self.player.matrix)]
    count = sum([a.count(0) for a in row])
    row2 = [[random.randint(0, 1) for i in range(4)] for n in range(4)]
    count2 = sum([a.count(0) for a in row2])
    return {
        "row": row,
        "count": count,
        "row2": row2,
        "count2": count2,
    }


class TaskOne(Page):
    form_model = models.Player
    form_fields = ["task_1", "correct_task_1"]

    timeout_seconds = 61


class TaskTwo(Page):
    form_model = models.Player
    form_fields = ["task_2", "correct_task_2"]

    timeout_seconds = 61


class TaskThree(Page):
    form_model = models.Player
    form_fields = ["task_3", "correct_task_3"]

    timeout_seconds = 61


class TaskFour(Page):
    form_model = models.Player
    form_fields = ["task_4", "correct_task_4"]

    timeout_seconds = 61


class TaskFive(Page):
    form_model = models.Player
    form_fields = ["task_5", "correct_task_5"]

    timeout_seconds = 61


class TaskSix(Page):
    form_model = models.Player
    form_fields = ["task_6", "correct_task_6"]

    timeout_seconds = 61


class TaskSeven(Page):
    form_model = models.Player
    form_fields = ["task_7", "correct_task_7"]

    timeout_seconds = 61


class TaskEight(Page):
    form_model = models.Player
    form_fields = ["task_8", "correct_task_8"]

    timeout_seconds = 61


class TaskNine(Page):
    form_model = models.Player
    form_fields = ["task_9", "correct_task_9"]

    timeout_seconds = 61


class TaskTen(Page):
    form_model = models.Player
    form_fields = ["task_10", "correct_task_10"]

    timeout_seconds = 61

    def before_next_page(self):
        return self.player.set_payoffs()


class Results(Page):
    def vars_for_template(self):
        return {
            'correct_counting': self.player.correct_counting,
        }


page_sequence = [
    TaskOne,
    TaskTwo,
    TaskThree,
    TaskFour,
    TaskFive,
    TaskSix,
    TaskSeven,
    TaskEight,
    TaskNine,
    TaskTen,
    Results
]