from otree.api import *
import random

doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'nol_6'
    PLAYERS_PER_GROUP = 6
    NUM_ROUNDS = 10


class Subsession(BaseSubsession):
    payoff_round = models.IntegerField(initial=0)


class Group(BaseGroup):
    total_claims_group = models.IntegerField(initial=0)
    collective_sanction = models.BooleanField(initial=False)
    success_probability = models.FloatField(initial=0.00)
    supervisor_message = models.LongStringField(initial=" ", blank=True)


class Player(BasePlayer):
    part_2_quiz_status = models.BooleanField(initial=False)
    part_2_first_roll_report = models.IntegerField(min=1, max=6,)
    part_2_second_roll_report = models.IntegerField(min=1, max=6,)
    part_2_third_roll_report = models.IntegerField(min=1, max=6,)
    part_1_quiz_status = models.BooleanField(initial=False)
    payoff_for_round = models.IntegerField(initial=0)
    part_1_puzzle_status = models.StringField(
        initial='Not Completed',)
    part_1_puzzle_quiz_status = models.BooleanField(initial=False)
    time_elapsed_to_solve = models.StringField(
        initial='00:00', blank=True)
    puzzle_payoff = models.IntegerField(initial=0)
    time_taken_puzzle = models.IntegerField(initial=0)
    group_color = models.StringField()
    sanction_on_player = models.BooleanField(initial=False)


# PAGES

class welcome(Page):
    @ staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1 and player.part_2_quiz_status == False


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


class part_1_puzzle_quiz(Page):
    form_model = "player"
    form_fields = ['part_1_puzzle_quiz_status']

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


class wait_before_puzzle(WaitPage):
    wait_for_all_groups = True

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


class part_1_puzzle_page(Page):
    form_model = 'player'
    form_fields = ['part_1_puzzle_status', 'time_elapsed_to_solve',]
    timeout_seconds = 60*10

    @staticmethod
    def vars_for_template(player: Player):
        group_id = player.group.id_in_subsession
        return {
            "group_id": group_id,
        }

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        if player.part_1_puzzle_status == "Done":
            time_temp = player.time_elapsed_to_solve.split(
                ":")
            time_for_player = 600 - \
                (int(time_temp[0])*60 + int(time_temp[1]))
            player.time_taken_puzzle = time_for_player

        elif player.part_1_puzzle_status == "Not Completed":
            player.time_taken_puzzle = 99999

        else:
            player.time_taken_puzzle = 99999

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


class wait_after_puzzle(WaitPage):
    wait_for_all_groups = True

    def after_all_players_arrive(subsession: Subsession):
        players_list = subsession.get_players()
        player_times = []
        if (players_list[0].round_number == 1):
            for i in range(0, len(players_list)):
                player_times.append(
                    players_list[i].field_maybe_none('time_taken_puzzle'))
            winner_player_time = player_times[0]
            winner_player = 0
            for i in range(0, len(player_times)):
                if (player_times[i] < winner_player_time):
                    winner_player_time = player_times[i]
                    winner_player = i

            if player_times[winner_player] > 0 and player_times[winner_player] < 99999:
                players_list[winner_player].puzzle_payoff = 3
                print(winner_player+1, "is the winner")

            elif player_times[winner_player] == 99999 or player_times[winner_player] == None or player_times[winner_player] == 0:
                players_list[winner_player].puzzle_payoff = 0

            else:
                print("There is some error with puzzle payoff!")
                players_list[winner_player].puzzle_payoff = 0

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


class part_2_welcome(Page):

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


class part_2_quiz_page(Page):
    form_model = "player"
    form_fields = ['part_2_quiz_status']

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


class part_2_roll_1_page(Page):
    form_model = "player"
    form_fields = ['part_2_first_roll_report', ]

    @staticmethod
    def vars_for_template(player: Player):
        return {
            "round": player.round_number,
        }

    @ staticmethod
    def is_displayed(player):
        return player.round_number <= 10


class part_2_roll_2_page(Page):
    form_model = "player"
    form_fields = ['part_2_second_roll_report', ]

    @staticmethod
    def vars_for_template(player: Player):
        return {
            "round": player.round_number,
        }

    @ staticmethod
    def is_displayed(player):
        return player.round_number <= 10


class part_2_roll_3_page(Page):
    form_model = "player"
    form_fields = ['part_2_third_roll_report', ]

    @staticmethod
    def vars_for_template(player: Player):
        return {
            "round": player.round_number,
        }

    @ staticmethod
    def is_displayed(player):
        return player.round_number <= 10


class part_2_ResultsWaitPage(WaitPage):
    wait_for_all_groups = False

    @ 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_4 = players_list[3]
        player_5 = players_list[4]
        player_6 = players_list[5]
        player_1_total = player_1.part_2_first_roll_report + \
            player_1.part_2_second_roll_report + player_1.part_2_third_roll_report
        player_2_total = player_2.part_2_first_roll_report + \
            player_2.part_2_second_roll_report + player_2.part_2_third_roll_report
        player_3_total = player_3.part_2_first_roll_report + \
            player_3.part_2_second_roll_report + player_3.part_2_third_roll_report
        player_4_total = player_4.part_2_first_roll_report + \
            player_4.part_2_second_roll_report + player_4.part_2_third_roll_report
        player_5_total = player_5.part_2_first_roll_report + \
            player_5.part_2_second_roll_report + player_5.part_2_third_roll_report
        player_6_total = player_6.part_2_first_roll_report + \
            player_6.part_2_second_roll_report + player_6.part_2_third_roll_report

        group.total_claims_group = player_1_total + \
            player_2_total + player_3_total + player_4_total + player_5_total + player_6_total
        if group.total_claims_group >= 72:
            group.collective_sanction = True
            group.success_probability = round(random.uniform(0, 1), 4)
            if group.success_probability > 0.25:
                player_1.payoff_for_round = player_1_total
                player_2.payoff_for_round = player_2_total
                player_3.payoff_for_round = player_3_total
                player_4.payoff_for_round = player_4_total
                player_5.payoff_for_round = player_5_total
                player_6.payoff_for_round = player_6_total
            elif group.success_probability <= 0.25:
                player_1.payoff_for_round = 0
                player_2.payoff_for_round = 0
                player_3.payoff_for_round = 0
                player_4.payoff_for_round = 0
                player_5.payoff_for_round = 0
                player_6.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
            player_4.payoff_for_round = player_4_total
            player_5.payoff_for_round = player_5_total
            player_6.payoff_for_round = player_6_total

        if player_1.round_number == 10:
            if group.subsession.payoff_round == 0:
                group.subsession.payoff_round = random.randint(1, 10)
            else:
                pass

    @ staticmethod
    def is_displayed(player):
        return player.round_number <= 10


class part_2_results(Page):

    @staticmethod
    def vars_for_template(player: Player):
        player_payoff_round = player.payoff_for_round

        if (player.round_number == 10):

            part_1_payoff = player.in_round(1).puzzle_payoff
            player.participant.part_1_payoff = part_1_payoff
            payoff_round = player.participant.payoff_round = player.subsession.payoff_round
            if payoff_round > 0 and payoff_round <= 10:
                part_2_payoff = player.in_round(payoff_round).payoff_for_round
                player.participant.part_2_payoff = part_2_payoff
                player.payoff = part_1_payoff + part_2_payoff

            if (not (player.participant.part_1_payoff)) or (player.participant.part_1_payoff == None):
                player.participant.part_1_payoff = 0
            else:
                pass

            if not (player.participant.part_2_payoff) or (player.participant.part_2_payoff == None):
                player.participant.part_2_payoff = 0
            else:
                pass

        return {
            'player_payoff_round': player_payoff_round,
            'round': player.round_number,
            'group_total': player.group.total_claims_group,
        }

    @ staticmethod
    def is_displayed(player):
        return player.round_number <= 10


page_sequence = [welcome, part_1_puzzle_quiz,
                 wait_before_puzzle, part_1_puzzle_page, wait_after_puzzle, part_2_welcome, part_2_quiz_page, part_2_roll_1_page, part_2_roll_2_page, part_2_roll_3_page, part_2_ResultsWaitPage, part_2_results]