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


# METHOD: =================================================================================== #
# DEFINE VARIABLES USED IN ALL TEMPLATES ==================================================== #
# =========================================================================================== #
def vars_for_all_templates(self):
    return {
        'nr_courses': Constants.nr_courses,
        'players_per_group': Constants.players_per_group,
        'capacities': Constants.capacities,
        'indices': [j for j in range(1, Constants.nr_courses + 1)],
        'valuations': self.participant.vars['valuations'],
        'valuations_others': zip(self.participant.vars['other_types_names'],
                                 self.participant.vars['valuations_others']),
        's_len': Constants.s_len,
        'endowment': Constants.endowment
    }


class Instructions(Page):
    pass


class InstructionsFramed(Page):
    pass


class Decision(Page):

    form_model = 'player'

    # METHOD: =================================================================================== #
    # RETRIEVE FORM FIELDS FROM MODELS.PY ======================================================= #
    # =========================================================================================== #
    def get_form_fields(self):
        form_fields = [list(i) for i in zip(*self.participant.vars['form_fields_plus_index'])][1]

        return form_fields

    # METHOD: =================================================================================== #
    # CREATE VARIABLES TO DISPLAY ON DECISION.HTML ============================================== #
    # =========================================================================================== #
    def vars_for_template(self):
        form_fields = [list(i) for i in zip(*self.participant.vars['form_fields_plus_index'])][1]

        return {
                'form_fields': form_fields,
                }

    # METHOD: =================================================================================== #
    # BEFORE NEXT PAGE: WRITE BACK PLAYER BIDS TO PARTICIPANT VARS AND PREPARE TIE-BREAKER ====== #
    # =========================================================================================== #
    def before_next_page(self):
        # CREATE INDICES FOR MOST IMPORTANT VARS ================================================ #
        indices = [list(i) for i in zip(*self.participant.vars['form_fields_plus_index'])][0]
        form_fields = [list(i) for i in zip(*self.participant.vars['form_fields_plus_index'])][1]

        # DYNAMICALLY WRITE BACK PLAYER BIDS TO A LIST OF BIDS  ================================= #
        for n, bid in zip(indices, form_fields):
            bid_i = getattr(self.player, bid)
            self.participant.vars['player_bids'][n - 1] = int(bid_i)

        # PREPARE BIDS FOR TIE-BREAKING AND FOR THE ALLOCATION ================================== #
        self.player.prepare_decisions()

    # METHOD: =================================================================================== #
    # CONTROL THE SUM OF BIDS: SUM OF BIDS MUST NOT EXCEED THE PLAYERS' ENDOWMENT =============== #
    # =========================================================================================== #
    def error_message(self, values):
        form_fields = [list(i) for i in zip(*self.participant.vars['form_fields_plus_index'])][1]
        sum_of_bids = sum([values[i] for i in form_fields])
        if not Constants.enforce_binding:
            if sum_of_bids > Constants.endowment:
                return 'The sum of your bids must not exceed %s!' % Constants.endowment
        else:
            if sum_of_bids != Constants.endowment:
                return 'The sum of your bids must be exactly %s!' % Constants.endowment


class ResultsWaitPage(WaitPage):

    # METHOD: =================================================================================== #
    # AFTER ALL PLAYERS HAVE SUBMITTED BIDS: RUN UMBS MECHANISM AND SET PLAYERS' PAYOFFS ======== #
    # =========================================================================================== #
    def after_all_players_arrive(self):
        self.group.get_allocation()
        self.group.set_payoffs()

        # CREATE INDICES FOR MOST IMPORTANT VARS ================================================ #
        indices = [list(i) for i in zip(*self.session.vars['form_fields_clearing_bids_plus_index'])][0]
        form_fields_clearing_bids = [list(i) for i in zip(*self.session.vars['form_fields_clearing_bids_plus_index'])][1]

        # DYNAMICALLY WRITE BACK CLEARING BIDS TO MODEL FIELDS ================================== #
        for n, bid in zip(indices, form_fields_clearing_bids):
            clearing_bid_i = self.session.vars['clearing_bids'][n - 1]
            setattr(self.group, bid, clearing_bid_i)


class Results(Page):

    # METHOD: =================================================================================== #
    # CREATE VARIABLES TO DISPLAY ON RESULTS.HTML =============================================== #
    # =========================================================================================== #
    def vars_for_template(self):
        player_bids = [i for i in self.participant.vars['player_bids']]
        successful = [i for i in self.participant.vars['successful']]
        clearing_bids = [i for i in self.session.vars['clearing_bids']]
        payoff = self.player.payoff

        return {
                'player_bids': player_bids,
                'successful': successful,
                'clearing_bids': clearing_bids,
                'payoff': payoff
                }


class Thanks(Page):
    pass


page_sequence = [
    Decision,
    ResultsWaitPage,
]

if Constants.application_framing:
    if Constants.instructions:
        page_sequence.insert(0, InstructionsFramed)

    if Constants.results:
        page_sequence.insert(-1, Results)

else:
    if Constants.instructions:
        page_sequence.insert(0, Instructions)

    if Constants.results:
        page_sequence.insert(-1, Results)