from otree.api import *
import random, pandas as pd
import csv

doc = """
Basic Newsvendor Game
Author: Anugna Reddy Gondi
Updated: 10-01-21
"""


class Constants(BaseConstants):
    name_in_url = 'nv_game'
    players_per_group = None

    # Number of rounds (change the value below)
    num_rounds = 10

    # selling price (change the value below)
    price = 12

    # cost hm-high margin lm-low margin (change the value below)
    hm_cost = 3
    lm_cost = 9

    # demand uniformly distributed[0, 300] (change the values below)
    demand_min = 0
    demand_max = 300
    # demand_realz = pd.read_csv('demand.csv').to_dict()['demand'] #to have the same series for everybody


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    order_qty = models.IntegerField(min=0, label="Order Quantity:")
    test_order_qty = models.IntegerField(min=0, label="Test Order Quantity:")
    # demand realz
    demand_realization = models.IntegerField(min=Constants.demand_min, max=Constants.demand_max)
    # item_cost = models.IntegerField()


# functions
def creating_session(subsession):
    if subsession.round_number == 1:
        for player in subsession.get_players():
            participant = player.participant
            if subsession.session.config['margin_type'] == "high":
                participant.item_cost = Constants.hm_cost
            else:
                participant.item_cost = Constants.lm_cost


def set_payoffs(player):
    # dem_round = Constants.demand_realz[player.round_number - 1]
    # player.payoff = min(player.order_qty, Constants.demand_realz[
    player.payoff = min(player.order_qty,
                        player.demand_realization) * Constants.price - player.order_qty * player.participant.item_cost


# def deam(player):
#     for i in list(range(0,301)):
#         player.potential_profits[i]=min(player.test_order_qty, i) * Constants.price - player.test_order_qty * player.participant.item_cost
#     return player.potential_profits


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

    @staticmethod
    def vars_for_template(player):
        return {'round': Constants.num_rounds,
                'price': Constants.price,
                'cost': player.participant.item_cost,
                'demand_min': Constants.demand_min,
                'demand_max': Constants.demand_max}


class RoundData(Page):
    form_model = 'player'
    form_fields = ['test_order_qty', 'order_qty', 'demand_realization']

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

    @staticmethod
    def vars_for_template(player):
        return {'round': player.round_number,
                'price': Constants.price,
                'cost': player.participant.item_cost,
                'demand_min': Constants.demand_min,
                'demand_max': Constants.demand_max}

    @staticmethod
    def js_vars(player):
        all_poss_dems = list(range(Constants.demand_min, Constants.demand_max + 1))
        return dict(
            payoff=player.payoff,
            all_poss_dems=all_poss_dems,
        )

    @staticmethod
    def before_next_page(player, timeout_happened):
        set_payoffs(player)


# class ResultsWaitPage(WaitPage):
# after_all_players_arrive = 'set_payoffs'


class Results(Page):
    @staticmethod
    def vars_for_template(player):
        return {'round': player.round_number,
                'price': Constants.price,
                'cost': player.participant.item_cost,
                'demand_realized_round': player.demand_realization,
                'qty_sold': min(player.order_qty, player.demand_realization)
                }


page_sequence = [FirstPage, RoundData, Results]