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

# variables for all templates
# --------------------------------------------------------------------------------------------------------------------

matrix_finished = []

# ******************************************************************************************************************** #
# *** CLASS INSTRUCTIONS *** #
# ******************************************************************************************************************** #
class Instructions(Page):

    # only display instruction in round 1
    # ----------------------------------------------------------------------------------------------------------------
    def is_displayed(self):
        return self.subsession.round_number == 1

    # variables for template
    # ----------------------------------------------------------------------------------------------------------------
    def vars_for_template(self):
        return {
            'num_choices':  len(self.participant.vars['mpl_choices'])
        }


class Introduction_EBRT(Page):

    # only display instruction in round 1
    # ----------------------------------------------------------------------------------------------------------------
    def is_displayed(self):
        return self.subsession.round_number == 1

    # variables for template
    # ----------------------------------------------------------------------------------------------------------------
    def vars_for_template(self):
        return {
            'num_choices':  len(self.participant.vars['mpl_choices'])
        }


# ******************************************************************************************************************** #
# *** PAGE DECISION *** #
# ******************************************************************************************************************** #
class Decision(Page):

    # form model
    # ----------------------------------------------------------------------------------------------------------------
    form_model = 'player'

    # form fields
    # ----------------------------------------------------------------------------------------------------------------
    def get_form_fields(self):
        count = 1
        option_b = 0
        option_a= '50% chance gaining € 0, 50% chance of gaining € 10'
        for i in range(len(self.player.participant.vars['mpl_choices'])):
            self.player.participant.vars['mpl_choices'][i] = (count, 'choice_'+str(count), '5/10', '10/10', option_a, '€ '+str(option_b))
            count = count + 1
            option_b = option_b + 1

        # unzip list of form_fields from <mpl_choices> list
        form_fields = [list(t) for t in zip(*self.player.participant.vars['mpl_choices'])][1]


        # provide form field associated with pagination or full list
        if Constants.one_choice_per_page:
            page = self.subsession.round_number
            return [form_fields[page - 1]]
        else:
            return form_fields

    # variables for template
    # ----------------------------------------------------------------------------------------------------------------
    def vars_for_template(self):

        # specify info for progress bar
        total = len(self.participant.vars['mpl_choices'])
        page = self.subsession.round_number
        progress = page / total * 100

        print("participant vars mpl choices: ")
        print(self.player.participant.vars['mpl_choices'])

        # TODO changing the vars mpl_choices to fit into the new game
        count = 1
        option_b = 0
        option_a= '50% chance gaining € 0, 50% chance of gaining € 10'
        self.player.participant.vars['mpl_choices'] = [None] * 11
        for i in range(11):
            self.player.participant.vars['mpl_choices'][i] = (count, 'choice_'+str(count), '5/10', '10/10', option_a, '100% chance of € '+str(option_b))
            count = count + 1
            option_b = option_b + 1


        # unzip list of form_fields from <mpl_choices> list
        form_fields = [list(t) for t in zip(*self.player.participant.vars['mpl_choices'])][1]

        if Constants.one_choice_per_page:
            return {
                'page':      page,
                'total':     total,
                'progress':  progress,
                'choices':   [self.player.participant.vars['mpl_choices'][page - 1]]
            }
        else:
            return {
                'choices':   self.player.participant.vars['mpl_choices'],
                'half_pie': 10,
            }

    # set player's payoff
    # ----------------------------------------------------------------------------------------------------------------
    def before_next_page(self):

        # unzip indices and form fields from <mpl_choices> list
        round_number = self.subsession.round_number
        form_fields = [list(t) for t in zip(*self.participant.vars['mpl_choices'])][1]
        indices = [list(t) for t in zip(*self.participant.vars['mpl_choices'])][0]
        index = indices[round_number - 1]


        # if choices are displayed in tabular format
        # ------------------------------------------------------------------------------------------------------------
        if not Constants.one_choice_per_page:

            print("Choices were: ")
            # replace choices in <choices_made>
            index = 1
            draw = random.randint(1, 11)
            print("draw of risk: ", draw)
            self.participant.vars['switch'] = False

            for j, choice in zip(indices, form_fields):
                choice_i = getattr(self.player, choice)
                print(choice_i)
                self.participant.vars['mpl_choices_made'][j - 1] = choice_i
                if choice_i == 'B':
                    self.participant.vars['switch'] = True
                if index == draw:
                    if(choice_i == 'A') and self.participant.vars['switch'] == False:
                        coin_toss = [0,10]
                        self.participant.vars['risk_score'] = random.choice(coin_toss)
                        print("current risk payoff: ")
                        print(self.participant.vars['risk_score'])
                        print(index)
                        break
                    if(choice_i == 'A') and self.participant.vars['switch'] == True:
                        self.participant.vars['risk_score'] = draw -1
                        print("current risk payoff: ")
                        print(self.participant.vars['risk_score'])
                        print(index)
                        break
                    if(choice_i == 'B'):
                        self.participant.vars['risk_score'] = index-1
                        print("current risk payoff: ")
                        print(self.participant.vars['risk_score'])
                        print(index)
                        break

                index = index + 1


# ******************************************************************************************************************** #
# *** PAGE RESULTS *** #
# ******************************************************************************************************************** #
class Results(Page):

    # skip results until last page
    # ----------------------------------------------------------------------------------------------------------------
    def is_displayed(self):
        return True

    # variables for template
    # ----------------------------------------------------------------------------------------------------------------
    def vars_for_template(self):

        # unzip <mpl_choices> into list of lists
        choices = [list(t) for t in zip(*self.participant.vars['mpl_choices'])]
        indices = choices[0]

        if Constants.one_choice_per_page:
            return {
                'payoff':         self.player.payoff,
                'indices':        indices
            }
        else:
            return {
                'option_to_pay':  self.player.option_to_pay,
                'payoff':         self.player.payoff,
                'indices':        indices,
                'num_groups': len(self.subsession.get_groups()),
                'count': len(matrix_finished),

            }


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        matrix_finished.append(1)

class AllGroupsWaitPage(WaitPage):
    wait_for_all_groups = True

# ******************************************************************************************************************** #
# *** PAGE SEQUENCE *** #
# ******************************************************************************************************************** #
page_sequence = [Instructions,Decision, Results, AllGroupsWaitPage]