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


class BasePage(Page):
    def vars_for_template(self):
        v =  {
            # 'treatment': self.session.config['treatment'],
            'other_player': self.player.get_partner(),
            'num_rounds': Constants.interaction_length[0],
            'random_matching_reminder': (self.player.treatment == 'random') and (self.player.round_number>1),
            'p11': Constants.payoff_matrix[str(self.player.interaction_number)]['X']['X'],
            'p12': Constants.payoff_matrix[str(self.player.interaction_number)]['X']['Y'],
            'p21': Constants.payoff_matrix[str(self.player.interaction_number)]['Y']['X'],
            'p22': Constants.payoff_matrix[str(self.player.interaction_number)]['Y']['Y'],
        }
        print('player.round_number',self.player,self.player.round_number,self.player.treatment,v)
        v.update(self.extra_vars_for_template())
        return v

    def extra_vars_for_template(self):
        return {}


class BaseWaitPage(WaitPage):
    def vars_for_template(self):
        v = {
            # 'treatment': self.session.config['treatment'],
            'other_player': self.player.get_partner(),
            'num_rounds': Constants.interaction_length[0],
            'p11': Constants.payoff_matrix[str(self.player.interaction_number)]['X']['X'],
            'p12': Constants.payoff_matrix[str(self.player.interaction_number)]['X']['Y'],
            'p21': Constants.payoff_matrix[str(self.player.interaction_number)]['Y']['X'],
            'p22': Constants.payoff_matrix[str(self.player.interaction_number)]['Y']['Y'],
        }
        v.update(self.extra_vars_for_template())
        return v

    def extra_vars_for_template(self):
        return {}


class Introduction(BasePage):
    # timeout_seconds = 30

    def is_displayed(self):
        if self.player.interaction_number == 1:
            print('This is the start of new match')
        return self.player.round_in_interaction == 1
        # return self.round_number == 1


class Decision(BasePage):
    # timeout_seconds = 30
    form_model = 'player'
    form_fields = ['action']

    def before_next_page(self):
        if self.timeout_happened:
            self.player.action = random.choice(['X','Y'])


class DecisionWaitPage(BaseWaitPage):
    template_name = 'repeated_game_randpay/DecisionWaitPage.html'

    def after_all_players_arrive(self):
        # it only gets executed once
        self.group.interact()
        # print('players have interacted!')


class Results(BasePage):
    timeout_seconds = 5
    # def get_timeout_seconds(self):
    #     if self.player.treatment == 'reputation':
    #         return None
    #     else:
    #         return 30


# class ResultsWaitPage(BaseWaitPage):
class ResultsWaitPage(WaitPage):
    template_name = 'repeated_game_randpay/ResultsWaitPage.html'
    wait_for_all_groups = True

    def is_displayed(self):
        return self.player.treatment == 'reputation'


class InteractionResults(BasePage):
    timeout_seconds = 15

    def is_displayed(self):
        return self.player.round_in_interaction == Constants.interaction_length[self.player.interaction_number-1]

    def extra_vars_for_template(self):
        paying_rounds = [p.round_in_interaction for p in self.player.in_all_rounds()
                         if p.interaction_number == self.player.interaction_number and p.paying_round == 1]
        return {
            'paying_round': paying_rounds[0],
        }


class InteractionWaitPage(BaseWaitPage):
    template_name = 'repeated_game_randpay/InteractionWaitPage.html'
    wait_for_all_groups = True

    def is_displayed(self):
        return self.player.round_in_interaction == Constants.interaction_length[self.player.interaction_number-1]

    def extra_vars_for_template(self):
        paying_rounds = [p.round_in_interaction for p in self.player.in_all_rounds()
                         if p.interaction_number == self.player.interaction_number and p.paying_round == 1]
        return {
            'paying_round': paying_rounds[0],
        }

page_sequence = [
    Introduction,
    Decision,
    DecisionWaitPage,
    Results,
    ResultsWaitPage,
    InteractionResults,
    InteractionWaitPage,
]