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


class General(Page):
    form_model = 'player'
    form_fields = ['gender',
                   'age',
                   'nationality',
                   'study']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Friends(Page):
    form_model = 'player'
    form_fields = ['friends']

    def is_displayed(self):
        return self.player.treatment > 0 and self.player.id_in_group != 4


class GuessG(Page):
    form_model = 'player'
    form_fields = ['guess_gender1', 'guess_gender2']

    def is_displayed(self):
        return self.player.treatment > 0 and self.player.id_in_group != 4


class GuessN(Page):
    form_model = 'player'
    form_fields = ['guess_nationality1', 'guess_nationality2']

    def is_displayed(self):
        return self.player.treatment > 0 and self.player.id_in_group != 4


class Comp(Page):
    form_model = 'player'
    form_fields = ['competetiveness']

    def is_displayed(self):
        return self.player.id_in_group != 4

    def before_next_page(self):
        if (self.player.orsee == 'female' and self.player.guess_gender1 == 2) or (self.player.orsee == 'male' and self.player.guess_gender1 == 1):
            correct = True
        else:
            correct = False
        if correct:
            self.player.payoff = 1
        else:
            self.player.payoff = 0

        if self.player.paying_part == 1:
            self.player.total = self.player.payoff + self.player.participant.vars['from_1']
        if self.player.paying_part == 2:
            self.player.total = self.player.payoff + self.player.participant.vars['from_2']

class Compe(Page):
    form_model = 'player'
    form_fields = ['competetition']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Risk(Page):
    form_model = 'player'
    form_fields = ['risk']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Time(Page):
    form_model = 'player'
    form_fields = ['time']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Intent(Page):
    form_model = 'player'
    form_fields = ['intentions']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Alt(Page):
    form_model = 'player'
    form_fields = ['altruism']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Bottle(Page):
    form_model = 'player'
    form_fields = ['bottle']

    def is_displayed(self):
        return self.player.id_in_group != 4


class Pun(Page):
    form_model = 'player'
    form_fields = ['punish']

    def is_displayed(self):
        return self.player.id_in_group != 4

class Strategy(Page):
    form_model = 'player'
    form_fields = ['strategy', 'elaborate2', 'elaborate3']

    def is_displayed(self):
        return self.player.treatment > 0 and self.player.id_in_group != 4


class Experiment(Page):
    form_model = 'player'
    form_fields = ['heard', 'content', 'aim', 'understand', 'time_pressure', 'enjoy', 'skills']

    def is_displayed(self):
        return self.player.id_in_group != 4


class End(Page):
    pass


page_sequence = [General,
                 GuessG,
                 GuessN,
                 Friends,
                 Comp,
                 Compe,
                 Risk,
                 Time,
                 Intent,
                 Alt,
                 Bottle,
                 Pun,
                 Strategy,
                 Experiment,
                 End]