from __future__ import division

# from otree.common import Currency as c, currency_range, safe_json

from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants
import logging
# from otreeutils.pages import ExtendedPage

logger = logging.getLogger(__name__)




# ******************************************************************************************************************** #
# *** CLASS INSTRUCTIONS *** #
# ******************************************************************************************************************** #
class Instructions(Page):
    # custom_name_in_url = 'e'
    timeout_seconds = 300
    # only display instruction in round 1
    def is_displayed(self):
        return self.subsession.round_number == 1

    # variables for use in template
    def vars_for_template(self):
        return {
            'num_rows': Constants.num_rows,
            'num_cols': Constants.num_cols,
            'num_boxes': Constants.num_rows * Constants.num_cols,
            'num_nobomb': Constants.num_rows * Constants.num_cols - 1,
            'box_value': Constants.box_value,
            'time_interval': Constants.time_interval,
            'num_rounds': Constants.num_rounds,
        }

    def before_next_page(self):
        self.player.my_memory_payoff = self.player.participant.vars['payoff_memory']
        if self.timeout_happened:
            self.player.timeout = 1 # participant is excluded from study if time runs out


# ******************************************************************************************************************** #
# *** CLASS BOMB RISK ELICITATION TASK *** #
# ******************************************************************************************************************** #
class Decision(Page):
    # custom_name_in_url = 'e'
    timeout_seconds = 300
    # form fields on player level
    form_model = 'player'
    form_fields = [
        'bomb',
        'boxes_collected',
        'bomb_row',
        'bomb_col',
    ]

    # BRET settings for Javascript application
    def vars_for_template(self):
        reset = self.participant.vars.get('reset', False)
        if reset:
            del self.participant.vars['reset']

        input = not Constants.devils_game if not Constants.dynamic else False

        otree_vars = {
            'reset': reset,
            'input': input,
            'random': Constants.random,
            'dynamic': Constants.dynamic,
            'num_rows': Constants.num_rows,
            'num_cols': Constants.num_cols,
            'feedback': Constants.feedback,
            'undoable': Constants.undoable,
            'box_width': Constants.box_width,
            'box_height': Constants.box_height,
            'time_interval': Constants.time_interval,
            'num_boxes': Constants.num_rows * Constants.num_cols,
            'num_nobomb': Constants.num_rows * Constants.num_cols - 1,
            'box_value': Constants.box_value,
            'num_rounds': Constants.num_rounds,
        }

        return {
            'otree_vars': otree_vars,
            'num_rows': Constants.num_rows,
            'num_cols': Constants.num_cols,
            'num_boxes': Constants.num_rows * Constants.num_cols,
            'num_nobomb': Constants.num_rows * Constants.num_cols - 1,
            'box_value': Constants.box_value,
            'time_interval': Constants.time_interval,
            'num_rounds': Constants.num_rounds,
        }

    def before_next_page(self):
        self.participant.vars['reset'] = True
        self.player.set_payoff()
        if self.timeout_happened:
            self.player.timeout = 1  # participant is excluded from study if time runs out


# ******************************************************************************************************************** #
# *** CLASS RESULTS *** #
# ******************************************************************************************************************** #
class Results(Page):
    # custom_name_in_url = 'e'
    timeout_seconds = 300
    # only display results after all rounds have been played
    def is_displayed(self):
        return self.subsession.round_number == Constants.num_rounds

    # variables for use in template
    def vars_for_template(self):
        total_payoff = sum([p.payoff for p in self.player.in_all_rounds()])
        self.participant.vars['bret_payoff'] = total_payoff
        self.participant.vars['my_bret_payoff'] = total_payoff

        return {
            'player_in_all_rounds': self.player.in_all_rounds(),
            'box_value': Constants.box_value,
            'boxes_total': Constants.num_rows * Constants.num_cols,
            'boxes_collected': self.player.boxes_collected,
            'bomb': self.player.bomb,
            'bomb_row': self.player.bomb_row,
            'bomb_col': self.player.bomb_col,
            'round_result': self.player.round_result,
            'round_to_pay': self.participant.vars['round_to_pay'],
            'payoff': self.player.payoff,
            'total_payoff': total_payoff,
        }
    def before_next_page(self):
        if self.timeout_happened:
            self.player.timeout = 1 # participant is excluded from study if time runs out

class TimeOutNew(Page): #shown if timeout
    # custom_name_in_url='time_out'
    def is_displayed(self):
        return self.player.timeout == 1
    # the participant is kicked out of the experiment if he the timer runs out


# ******************************************************************************************************************** #
# *** PAGE SEQUENCE *** #
# ******************************************************************************************************************** #
page_sequence = [
    Instructions,
    TimeOutNew,
    Decision,
    TimeOutNew,
    Results,
    TimeOutNew
]

# if Constants.instructions:
#     page_sequence.insert(0,Instructions)
#
# if Constants.results:
#     page_sequence.append(Results)