from otree.api import (
    models, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c
)
import random


"""
Sim for "Second Price Auction" game. Small groups of a configurable size are formed
to participate in a series of auctions. In each round, each group member will
enter a bid for a fictitious item. The person with the highest bid wins and earns
their "value" for the item - and they must pay a value equivalent to the second highest
bid in the group. Thus, the winner's payoff is equal to their value minus the second
highest bid. The payoff for all other players is 0.

In each round, the value of the fictitious item for each individual player will be equal to an independently and randomly assigned number ranging from 0 to 100.
"""


class Constants(BaseConstants):
    name_in_url = 'secondPrice'
    players_per_group = None
    num_rounds = 3
    instructions_template = 'secondPrice/instructions.html'


class Subsession(BaseSubsession):

    def creating_session(self):
        group_matrix = []
        players = self.get_players()
        ppg = self.session.config['players_per_group']
        for i in range(0, len(players), ppg):
            group_matrix.append(players[i:i+ppg])
        self.set_group_matrix(group_matrix)


class Group(BaseGroup):
    highest = models.CurrencyField()
    second = models.CurrencyField()
    winner = models.IntegerField()

    def set_winner(self):
        players = self.get_players()
        bids = [p.bid for p in players]
        bids.sort()
        self.highest = bids[-1]
        self.second = bids[-2]
        tied = [p.id_in_group for p in players if p.bid == self.highest]
        self.winner = random.choice(tied)
        winner = self.get_player_by_id(self.winner)
        winner.isWinner = True

    def set_value(self):
        players = self.get_players()
        for p in players:
            p.value = c(int(random.uniform(0, 100)))


class Player(BasePlayer):
    value = models.CurrencyField(min=0)
    bid = models.CurrencyField(min=0, label='Please enter your bid.')
    isWinner = models.BooleanField(initial=False)

    def value_max(self):
        if self.round_number == 1 or self.round_number == 3:
            return 99
        else:
            return 31

    def set_payoff(self):
        if self.isWinner:
            self.payoff = self.value - self.group.second
        else:
            self.payoff = c(0)