from otree.api import *
import random


doc = """
This app is to be used with the Groups of 3 Participants.
"""


class C(BaseConstants):
    NAME_IN_URL = 'nol_3'
    PLAYERS_PER_GROUP = 3
    NUM_ROUNDS = 3


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    total_claims_group = models.IntegerField(initial=0)
    collective_sanction = models.BooleanField(initial=False)
    success_probability = models.FloatField(initial=0.00)
    payoff_round_part_1 = models.IntegerField(initial=0)


class Player(BasePlayer):
    first_roll_report = models.IntegerField(min=1, max=6,)
    second_roll_report = models.IntegerField(min=1, max=6,)
    third_roll_report = models.IntegerField(min=1, max=6,)
    payoff_for_round = models.IntegerField(initial=0)
    total_claims_participant = models.IntegerField(initial=0)
    quiz_status = models.BooleanField(initial=False)


# PAGES
class welcome(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1


class quiz_page(Page):

    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        player.quiz_status = True


class part_1_welcome(Page):
    @staticmethod
    def is_displayed(player):
        return player.round_number == 1


class round_1_page(Page):
    form_model = "player"
    form_fields = ['first_roll_report', ]


class round_2_page(Page):
    form_model = "player"
    form_fields = ['second_roll_report', ]


class round_3_page(Page):
    form_model = "player"
    form_fields = ['third_roll_report', ]


class group_behavior(Page):
    pass


class part_2_welcome(Page):
    pass


class ResultsWaitPage(WaitPage):
    @staticmethod
    def after_all_players_arrive(group: Group):
        players_list = group.get_players()
        player_1 = players_list[0]
        player_2 = players_list[1]
        player_3 = players_list[2]
        player_1_total = player_1.first_roll_report + \
            player_1.second_roll_report + player_1.third_roll_report
        player_2_total = player_2.first_roll_report + \
            player_2.second_roll_report + player_2.third_roll_report
        player_3_total = player_3.first_roll_report + \
            player_3.second_roll_report + player_3.third_roll_report

        group.total_claims_group = player_1_total + player_2_total + player_3_total
        if group.total_claims_group >= 36:
            group.collective_sanction = True
            group.success_probability = round(random.uniform(0, 1), 4)
            if group.success_probability > 0.6:
                player_1.payoff_for_round = player_1_total
                player_2.payoff_for_round = player_2_total
                player_3.payoff_for_round = player_3_total
            elif group.success_probability <= 0.6:
                player_1.payoff_for_round = 0
                player_2.payoff_for_round = 0
                player_3.payoff_for_round = 0

        else:
            group.collective_sanction = False
            player_1.payoff_for_round = player_1_total
            player_2.payoff_for_round = player_2_total
            player_3.payoff_for_round = player_3_total
        if group.round_number == 3:
            group.payoff_round_part_1 = random.randint(1, 3)


class Results(Page):

    @staticmethod
    def vars_for_template(player: Player):
        player_payoff_round = player.payoff_for_round
        player.total_claims_participant = player.total_claims_participant + player_payoff_round
        player.zeroth_roll_report = player.in_round(1).zeroth_roll_report
        if player.round_number == 3:
            payoff_temp = player.group.payoff_round_part_1
            player.payoff = player.in_round(
                payoff_temp).total_claims_participant
            player.participant.part_0_payoff = 0
            player.participant.part_1_payoff = player.payoff
            player.participant.part_1_payoff_round = player.group.payoff_round_part_1
        return {
            'player_payoff_round': player_payoff_round,
        }


page_sequence = [welcome, quiz_page, part_1_welcome,
                 round_1_page, round_2_page, round_3_page, ResultsWaitPage, Results, ]