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

class Consent(Page):
    form_model = 'player'
    form_fields = ['consent']

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'

class Welcome(Page):
    def vars_for_template(self):
        self.group.init_setting()

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

class Info(Page):
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'
    def before_next_page(self):
        self.player.tast()

class Summary(Page):

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'based on your choice'
        else:
            next_is = 'based on your tasting and choice'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

class passcode(Page):
    form_model = 'player'
    form_fields = ['passcode_2']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if values[f] !=self.session.config['passcode_second']]
        if errors:
            return 'Your password is incorrect'

class Survey_coffee1(Page):
    form_model = 'player'
    form_fields = ['coffee_like']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'

class Survey_coffee2(Page):
    form_model = 'player'
    form_fields = ['coffee_quality']

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'you selected based on the information provided'
        else:
            next_is = 'you selected based on your tasting'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'

class Survey_coffee3(Page):
    form_model = 'player'
    form_fields = ['coffee_sweetness']

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'you selected based on the information provided'
        else:
            next_is = 'you selected based on your tasting'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee4(Page):
    form_model = 'player'
    form_fields = ['coffee_flavor']

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'you selected based on the information provided'
        else:
            next_is = 'you selected based on your tasting'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee5(Page):
    form_model = 'player'
    form_fields = ['coffee_impression']

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'conveyed in the information provided'
        else:
            next_is = 'of its combined taste'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee6(Page):
    form_model = 'player'
    form_fields = ['coffee_recom']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'
        # return True

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee7(Page):
    form_model = 'player'
    form_fields = ['coffee_drink']

    def vars_for_template(self):
        if self.player.tasting_new == 0:
            next_is = 'Drink this coffee'
        else:
            next_is = 'Drink this coffee again'
        return {'tasting_new': self.player.tasting_new,
                'next_is': next_is
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee8(Page):
    form_model = 'player'
    form_fields = ['coffee_serve']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Survey_coffee9(Page):
    form_model = 'player'
    form_fields = ['coffee_buy']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class Payment(Page):
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'


class Payment2(Page):
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'


class SetPrice(Page):
    form_model = 'player'
    form_fields = ['W', 'lockin']

    def is_displayed(self):
        if self.player.participant.vars['consent'].lower() != 'consent':
            return False
        if self.player.lockin.lower() != 'lockin':
            return True
        else:
            return False

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'

    def before_next_page(self):
        self.player.set_payoff1()

    def vars_for_template(self):
        return {
                'round': ['first', 'second', 'third'][self.player.test_round],
                'round_n': ['1st', '2nd', '3rd'][self.player.test_round],
                }


class Res1(Page):
    def vars_for_template(self):
        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            next_info = 'Click the "Next" button to make the supplier another offer!'
        else:
            next_info = 'Click the "Next" button to continue with the experiment!'
        return {'is_reject': self.player.is_reject,
                'cost_bonus': self.player.cost_bonus,
                'next_info': next_info
                }

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'  or self.player.is_reject == 'rejected'

    def before_next_page(self):
        self.player.show_res1 = 0
        self.player.test_round += 1
        self.player.test_times = 0
        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            self.player.lockin = '-1'

class Res12(Page):
    def vars_for_template(self):
        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            next_info = 'Click the "Next" button to make the supplier another offer!'
        else:
            next_info = 'Click the "Next" button to continue with the experiment! '
        return {'is_reject': self.player.is_reject,
                'cost_bonus': self.player.cost_bonus,
                'next_info': next_info
                }

    def is_displayed(self):
        print(self.player.show_res1)
        return (self.player.participant.vars['consent'].lower() == 'consent' or self.player.is_reject == 'rejected') \
            and self.player.show_res2

    def before_next_page(self):
        self.player.test_round += 1
        self.player.test_times = 0

        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            self.player.lockin = '-1'


class Res123(Page):
    def vars_for_template(self):
        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            next_info = 'Click the "Next" button to finish the post-experimental survey!'
        else:
            next_info = 'Click the "Next" button to continue with the experiment! '
        return {'is_reject': self.player.is_reject,
                'cost_bonus': self.player.cost_bonus,
                'next_info': next_info
                }

    def is_displayed(self):
        return (self.player.participant.vars['consent'].lower() == 'consent' or self.player.is_reject == 'rejected') \
            and self.player.show_res3

    def before_next_page(self):
        self.player.test_round += 1
        self.player.test_times = 0

        if self.player.is_reject == 'rejected' and self.player.test_round<3:
            self.player.lockin = '-1'


class SetPrice2(Page):
    form_model = 'player'
    form_fields = ['R', 'lockin2']

    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'

    def is_displayed(self):
        if self.player.participant.vars['consent'].lower() != 'consent':
            return False
        if self.player.is_reject == 'accepted':
            if self.player.lockin2.lower() != 'lockin':
                return True
            else:
                return False
        else:
            return False

    def before_next_page(self):
        # self.player.price_check()
        self.player.set_payoff2()

    def vars_for_template(self):
        if self.player.test_times2 == 0:
            earn =  'XXX'
            cost_bonus = self.player.cost_bonus
            profit_bonus =  'XXX'
            total_bonus =  'XXX'
            market_demand =  'XXX'
            coffee_not_used =  'XXX'

        else:
            earn = self.player.earn
            cost_bonus = self.player.cost_bonus
            profit_bonus = self.player.profit_bonus
            total_bonus = self.player.total_bonus
            market_demand = self.player.market_demand
            coffee_not_used = self.player.coffee_not_used
        return {'earn': earn,
                'cost_bonus': cost_bonus,
                'profit_bonus': profit_bonus,
                'total_bonus': total_bonus,
                'market_demand':market_demand,
                'coffee_not_used':coffee_not_used
                }

class Res2(Page):
    def is_displayed(self):
        if self.player.participant.vars['consent'].lower() != 'consent':
            return False
        if self.player.is_reject == 'accepted':
            return True
        else:
            return False


class Survey(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'coffee']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'



    def error_message(self, values):
        errors = [1 for f in values if not values[f]]
        if errors:
            return 'you should select your answer'


class End(Page):
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'


class Survey1(Page):
    form_model = 'player'
    form_fields = ['coffee_howoften', 'coffee_where', 'coffee_where_string',
                   'door_unlocked', 'lend_money', 'lend_personal', 'lie_parents', 'lie_roommates',
                   'lie_acquaintances', 'lie_friends', 'lie_partner']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'
        # return True
    def before_next_page(self):
        self.player.set_payoff_final()
    def error_message(self, values):
        errors = [1 for f in values if 'string' not in f and not values[f]]
        if errors:
            return 'you should select your answer'


class Survey2(Page):
    form_model = 'player'
    form_fields = ['take_advantage', 'try_to_be_helpful', 'trust',
                   'count_on_strangers', 'deal_with_strangers', 'recycle',
                   'more_pay_fair_trade', 'frequency_not_buy', 'car', 'frequency_shoes']

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'
        # return True

    def error_message(self, values):
        errors = [1 for f in values if 'string' not in f and not values[f]]
        if errors:
            return 'you should select your answer'

class Survey3(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'gender_string','race','race_string',
                   'edu','major','major_string','parents','employment', 'employment_string',
                   'income','marital','children',
                   ]

    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'
        # return True

    def error_message(self, values):
        errors = [1 for f in values if 'string' not in f and not values[f]]
        if errors:
            return 'you should select your answer'


class Final(Page):
    def is_displayed(self):
        return self.player.participant.vars['consent'].lower() == 'consent'

    def vars_for_template(self):
        self.player.payoff_cem = round(float(self.player.participant.vars['payoff_cem']) * 0.02,2)
        self.player.payoff_trust = round(float(self.player.participant.vars['payoff_trust']) * 0.02,2)
        self.player.payoff_total = round((float(self.player.total_bonus) + float(self.player.payoff_cem) + float(self.player.payoff_trust)),2)

        return {'id': self.player.id_in_group,
                'payoff_trust':self.player.payoff_trust,
                'payoff_cem':self.player.payoff_cem,
                'payoff_all': self.player.payoff_total}


page_sequence = []


page_sequence += [
    Welcome,
                  Info,
                  Summary,
                  passcode,
                  Survey_coffee1,
                  Survey_coffee2,
                  Survey_coffee3,
                  Survey_coffee4,
                  Survey_coffee5,
                  Survey_coffee6,
                  Survey_coffee7,
                  Survey_coffee8,
                  Survey_coffee9,
                  Payment,
                  Payment2,
                  ]

page_sequence += [SetPrice] * 100
page_sequence += [Res1]
page_sequence += [SetPrice] * 100
page_sequence += [Res12]
page_sequence += [SetPrice] * 100
page_sequence += [Res123]

page_sequence += [SetPrice2] * 100
page_sequence += [Res2, Survey1, Survey2, Survey3, Final,]