from otree.api import Currency as c, currency_range
from otree.api import *
from ._builtin import Page, WaitPage
from .models import *
from settings import SESSION_CONFIG_DEFAULTS

import random

class Contest1(Page):
    def is_displayed(self):
        players = [p for p in self.group.get_players()]

        return True

    def vars_for_template(self):

        return {
            'award1': 30000,
                  }

    form_model = 'player'
    form_fields = ['effort']

class ResultsWaitPage1(WaitPage):

    def after_all_players_arrive(self):
       
        players = [p for p in self.group.get_players() ]
        efforts = [int(p.effort) for p in players]
        ids = [p.id_in_group for p in players]
        total_effort1 = sum(efforts)
        if total_effort1==0:
            weight=[1/3, 1/3, 1/3]
            print(weight)
            winner = random.choices(ids, weights=weight)[0]
        else:
            weight = [j / total_effort1 for j in efforts]
            print(weight)
            winner = random.choices(ids, weights=weight)[0]
        for i in range(len(weight)):
            players[i].weight1 = round(weight[i],2)
            players[i].total_effort1 = total_effort1

        for p in players:
            if p.id_in_group == winner:
                p.is_winner1 = "Yes"
                p.expenditure_stage1 = p.marginal_cost * int(p.effort) ** 2
                p.round_payoff = Constants.endowment - p.marginal_cost * int(p.effort) ** 2+ Constants.award1
                p.expenditure_stage1_string = "Your expenditure for this round = your marginal cost("+str(p.marginal_cost)+")* your effort level^2(" + str(int(p.effort) ** 2) + ")= " + str(p.expenditure_stage1)
                p.income_info_string = "Your payoff for this round = 60000 - your expenditure for this round("+str(p.marginal_cost * int(p.effort) ** 2)+")+ award for the winner(30000)= "+ str(p.round_payoff)

                p.Monetary_payoff= round(float(p.round_payoff) * Constants.rate)  

                print(p.round_payoff)
                print(p.Monetary_payoff) 

            else:
                p.is_winner1 = "No"
                p.expenditure_stage1 = p.marginal_cost * int(p.effort) ** 2
                p.round_payoff = Constants.endowment - p.marginal_cost * int(p.effort) ** 2
                p.expenditure_stage1_string = "Your expenditure for this round = your marginal cost(" + str(
                    p.marginal_cost) + ")* your effort level^2(" + str(int(p.effort) ** 2) + ")= " + str(p.expenditure_stage1)
                p.income_info_string = "Your payoff for this round = 60000 - your expenditure for this round("+str(p.marginal_cost * int(p.effort) ** 2)+")= "+ str(p.round_payoff)

                p.Monetary_payoff= round(float(p.round_payoff) * Constants.rate)  

                print(p.round_payoff)
                print(p.Monetary_payoff) 


class Results1(Page):
    def is_displayed(self):

        return True

class beforeEnd(WaitPage):
    form_model = 'player'
 

    def after_all_players_arrive(self):
        import random
        players = [p for p in self.group.get_players() if p.round_number == Constants.NUM_ROUNDS]
        random_round =random.randint(1, Constants.NUM_ROUNDS)
        player_in_selected_round = [p.in_round(random_round) for p in players]

        print("players length:" ,len(player_in_selected_round))
        for p in player_in_selected_round:
            p.in_round(Constants.NUM_ROUNDS).random_round= random_round
            p.in_round(Constants.NUM_ROUNDS).selected_round = random_round
            p.in_round(Constants.NUM_ROUNDS).payoff = p.Monetary_payoff
            p.in_round(Constants.NUM_ROUNDS).selected_round_payoff = p.round_payoff
            p.in_round(Constants.NUM_ROUNDS).total_payoff = p.Monetary_payoff +SESSION_CONFIG_DEFAULTS['participation_fee'] 

class End(Page):
    def is_displayed(self):
        return self.player.round_number == Constants.NUM_ROUNDS


page_sequence = [
    Contest1,
    ResultsWaitPage1,
    Results1,
    beforeEnd,
    End,
]