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


class At_Att_a(Page):
    form_model = "player"
    form_fields = ["most_important_topic", "attitude_immigration_2",
                   "homosexuality", "affirmative_action", "punish_criminals"]
    def is_displayed(self):
        # Get the color value from the player object
        return self.participant.vars["color"] == "treatment_immigration"

class At_Att_b(Page):
    form_model = "player"
    form_fields = ["most_important_topic", "attitude_redistribution_2",
                   "social_security", "abortion", "environmental_protection"]
    def is_displayed(self):
        return self.participant.vars["color"] == "treatment_redistribution"

class At_Att_c(Page):
    form_model = "player"
    form_fields = ["most_important_topic", "attitude_environment_2",
                   "eu_unification", "tradition", "eu_membership"]
    def is_displayed(self):
        return self.participant.vars["color"] == "treatment_environment"



class Au_PoDem(Page):
    form_model = "player"
    form_fields = ["religion", "county", "born_in_germany", "parents_born_in_germany", 
                   "party_membership", "interest_politics", "employment",
                    "survey_biased", "final_comments", "comment_information" ]


#class Av_SB(Page):
 #   form_model = "player"
  #  form_fields = [ "info_citizens_credible", "info_mps_credible","survey_biased",]


class Aw_By(Page):
    form_model = "player"



page_sequence = [  
    At_Att_a, At_Att_b, At_Att_c,
    Au_PoDem,
    #Av_SB 
    Aw_By
    ]