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


class GroupWP(WaitPage):
    group_by_arrival_time = True
#   def after_all_players_arrive(self):


class Role(Page):
    def vars_for_template(self):
        return {'treatment': Constants.treatment}

    def before_next_page(self):
        self.group.set_conditions()

class MarketInformation(Page):
    def vars_for_template(self):
        return {'treatment': Constants.treatment}

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


class SellerInfoBuyerValuation(Page):
    def vars_for_template(self):
#        return {'treatment': Constants.treatment, 'valuation1': self.player.private_value}
        return {'treatment': Constants.treatment}

    def is_displayed(self) -> bool:
        return self.player.role() == 'seller' #and (Constants.treatment == 1 or Constants.treatment == 2)


class SellerDecisionPricingMechanism(Page):
    def vars_for_template(self):
        return {'treatment': Constants.treatment}

    form_model = 'group'
    form_fields = ['s_choice_PWYW_FP']

    def is_displayed(self) -> bool:
        return self.player.role() == 'seller'


class SellerDecisionPriceProdCostInfo(Page):
    def vars_for_template(self):
        return {'treatment': Constants.treatment}

    form_model = 'group'
    form_fields = ['s_decision_price_FP']

    def is_displayed(self) -> bool:
        return self.player.role() == 'seller' and self.group.s_choice_PWYW_FP == 1


class SellerDecisionsWP(WaitPage):

    def after_all_players_arrive(self):
       pass


class BuyerValuationBuyingDecision(Page):
    def vars_for_template(self):
        return {
            'treatment': Constants.treatment,
            's_choice': self.group.s_choice_PWYW_FP
        }

    form_model = 'player'
    form_fields = ['b_decision_buy']

    def is_displayed(self) -> bool:
        return self.player.role() == 'buyer'


class BuyerPriceDecision(Page):
    def vars_for_template(self):
        return {'treatment': Constants.treatment}

    form_model = 'player'
    form_fields = ['b_decision_price_PWYW']

    def is_displayed(self) -> bool:
        return self.player.role() == 'buyer' and self.group.s_choice_PWYW_FP == 0 and self.player.b_decision_buy == 1


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        self.group.set_payoffs()


class Results(Page):
    pass


class Questionnaire(Page):

    form_model = 'player'
    form_fields = ['survey_easy_difficult', 'survey_risk', 'strategy_seller', 'strategy_buyer', 'age', 'gender', 'feedback']

    pass

class Bye(Page):
    pass

page_sequence = [
    GroupWP,
    Role,
    MarketInformation,
    SellerInfoBuyerValuation,
    SellerDecisionPricingMechanism,
    SellerDecisionPriceProdCostInfo,
    SellerDecisionsWP,
    BuyerValuationBuyingDecision,
    BuyerPriceDecision,
    ResultsWaitPage,
    Results,
    Questionnaire,
    Bye
]