from otree.api import *

doc = """
Instructions for games
"""


class C(BaseConstants):
    NAME_IN_URL = 'instructions'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1



class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    # cooperate = models.BooleanField(
    #     choices=[[True, 'A'], [False, 'B']],
    #     doc="""This player's decision""",
    #     widget=widgets.RadioSelect,
    # )
    # input_button = models.CharField()
    number = models.IntegerField()
    comprehension = models.IntegerField(label="")
    comprehension2 = models.IntegerField(label="")
    decision_time = models.IntegerField()

    # def record_decision_time(self):
    #     self.decision_time = int(time.time()) - self.participant.vars['start_time']  # records DT
    #     self.participant.vars['start_time'] = int(time.time())  # updates start time

# FUNCTIONS
def set_payoffs(group: Group):
    for p in group.get_players():
        set_payoff(p)


def other_player(player: Player):
    return player.get_others_in_group()[0]


# def set_payoff(player: Player):
#     payoff_matrix = {
#         (False, True): C.PAYOFF_A,
#         (True, True): C.PAYOFF_B,
#         (False, False): C.PAYOFF_A,
#         (True, False): C.PAYOFF_C,
#     }
#     other = other_player(player)
#     player.payoff = payoff_matrix[(player.cooperate, other.cooperate)]


# PAGES
class Game_Introduction(Page):
    pass

class Game_Introduction2(Page):
    pass

class Comprehension1(Page):
    def is_displayed(self):
        return self.round_number == 1
    form_model = 'player'
    form_fields = ['comprehension']

    @staticmethod
    def error_message(player: Player, values):
        solutions = dict(comprehension=66)

        if values != solutions:
            return "Por favor, inténtalo de nuevo."

class Comprehension2(Page):
    form_model = 'player'
    form_fields = ['comprehension2']

    @staticmethod
    def error_message(player: Player, values):
        solutions = dict(comprehension2=105)

        if values != solutions:
            return "Por favor, inténtalo de nuevo."

class Game_Introduction3(Page):
    # def before_next_page(player, timeout_happened):
    #      player.participant.vars['start_time'] = int(time.time())
    pass

class brain(Page):
    pass

class recap(Page):
    pass

class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs


class Results(Page):
    @staticmethod
    def vars_for_template(player: Player):
        opponent = other_player(player)
        return dict(
            opponent=opponent,
            same_choice=player.cooperate == opponent.cooperate,
            my_decision=player.field_display('cooperate'),
            opponent_decision=opponent.field_display('cooperate'),
        )

class Probabilities(Page):
    def is_displayed(self):
        return self.round_number != C.NUM_ROUNDS
    form_model = 'player'
    form_fields = ["number"]



page_sequence = [Game_Introduction, Game_Introduction2, Comprehension1, Comprehension2, Game_Introduction3, brain, recap, Probabilities] #CombinedResults]