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


class MyPage(Page):
    form_model = 'player'
    form_fields = ['bid']


class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        pass


class Results(Page):
    def vars_for_template(self):
        group = self.group 
        player = self.player
        p1 = group.get_player_by_id(1)
        p2 = group.get_player_by_id(2)
        p3 = group.get_player_by_id(3)
        p4 = group.get_player_by_id(4)
        bids_list = [p1.bid, p2.bid, p3.bid, p4.bid]
        group.price = max(bids_list)
        group.winner = 1 + np.argmax(bids_list)
        print('player', group.winner, 'wins at price', group.price)
        if player.id_in_group == group.winner : 
            player.profit = player.budget - group.price
        else : 
            player.profit = 0
        return{
            'price': group.price,
            'profit': player.profit,
            'winner': group.winner,
        }


page_sequence = [
    MyPage,
    ResultsWaitPage,
    Results
]