from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants

from otree.api import Currency as c, currency_range
from ._builtin import Page
from .models import Constants
import datetime, json, time
import random
import math
import numpy as np


class introduction_multitasking(Page):
    form_model = 'player'


class disclaimer(Page):
    form_model = 'player'


class general_instructions(Page):
    form_model = 'player'

    def before_next_page(self):
        self.participant.vars['sssss'] = time.time()



a = [
    [["1 2", "1 1", "2 1", "1 1", "2"],
     ["3", "2", "1 1", "1 1", "3"]],  # 5 by 5, max 3

    [["3","1 1","3","1 1 1","2"],
     ["2","2","1 2","2 1","1 2"]]

]


a_sol = [
            [[1, 0, 1, 1, 0],
             [1, 0, 0, 0, 1],
             [1, 1, 0, 0, 1],
             [0, 1, 0, 0, 1],
             [0, 0, 1, 1, 0]],

            [[0, 0, 1, 1, 1],
             [0, 1, 0, 1, 0],
             [1, 1, 1, 0, 0],
             [1, 0, 1, 0, 1],
             [0, 0, 0, 1, 1]]
        ]

green = 12
gray = 13


class practice_game(Page):
    form_model = 'player'
    timeout_seconds = 600
    form_fields = ['points', 'total_wrong', 'total_right', 'wrong_green', 'wrong_gray', 'right_green', 'right_gray',
                   'time_between_moves', 'board_history', 'solve_order', 'total_green', 'total_gray']

    def vars_for_template(player):
        for i in range(1, 4):
            if player.round_number == i:
                player.participant.vars['game_number'] = i
                player.participant.vars['row'] = a[i-1][0]
                player.participant.vars['column'] = a[i-1][1]
                player.participant.vars['solution_0'] = a_sol[i-1][0]
                player.participant.vars['solution_1'] = a_sol[i-1][1]
                player.participant.vars['solution_2'] = a_sol[i-1][2]
                player.participant.vars['solution_3'] = a_sol[i-1][3]
                player.participant.vars['solution_4'] = a_sol[i-1][4]
                player.participant.vars['total_green'] = green
                player.participant.vars['total_gray'] = gray

                return dict(
                    game_number=player.participant.vars['game_number'],
                    r=[player.participant.vars['row'][h] for h in range(0, 5)],
                    c=[player.participant.vars['column'][h] for h in range(0, 5)],
                    sol_0=[player.participant.vars['solution_0'][h] for h in range(0, 5)],
                    sol_1=[player.participant.vars['solution_1'][h] for h in range(0, 5)],
                    sol_2=[player.participant.vars['solution_2'][h] for h in range(0, 5)],
                    sol_3=[player.participant.vars['solution_3'][h] for h in range(0, 5)],
                    sol_4=[player.participant.vars['solution_4'][h] for h in range(0, 5)],
                    total_green=player.participant.vars['total_green'],
                    total_gray=player.participant.vars['total_gray'],
                )


    def js_vars(player):
        if player.round_number == 1:
            return {
                'total_green': player.participant.vars['total_green'],
                'total_gray': player.participant.vars['total_gray'],
            }

    def before_next_page(self):
        self.participant.vars['points'] = self.player.points
        self.player.time_spent = math.floor(time.time() - self.participant.vars['sssss'])


class results(Page):
    form_model = 'player'


class finish_practice(Page):
    form_model = 'player'


class instructions_earnings(Page):
    form_model = 'player'


page_sequence = [#introduction_multitasking,
                 #disclaimer,
                 general_instructions,
                 practice_game,
                 #results,
                 #finish_practice,
                 #instructions_earnings,
                 ]