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


class Welcome(Page):
    form_model = Player
    form_fields = ['mobile_device', 'time_start']


class Klausur(Page):
    form_model = Player
    form_fields = ['klausur_teilnahme', 'klausur_punkte',
                   'klausur_fair', 'klausur_leicht', 'klausur_uebungsblaetter', 'klausur_themen',
                   'time_klausur']


class Internet(Page):
    form_model = Player
    form_fields = ['internet_anschluss', 'mobildatenvolumen', 'internet_schnell', 'internet_probleme',
                   'time_internet']


class Conjoint_Pakete(Page):
    def vars_for_template(self):
        return {'randomnumber_vl_1': self.player.randomnumber_vl_1,
                'randomnumber_vl_2': self.player.randomnumber_vl_2,
                'randomnumber_tut_1': self.player.randomnumber_tut_1,
                'randomnumber_tut_2': self.player.randomnumber_tut_2,
                'randomnumber_ex_1': self.player.randomnumber_ex_1,
                'randomnumber_ex_2': self.player.randomnumber_ex_2
                }

    form_model = Player
    form_fields = ['choice_package_corona', 'choice_package_no_corona', 'time_conjoint']




class Wahlzettel_Ankuendigung(Page):
    form_model = Player
    form_fields = ['time_wahlzettel_ankuendigung']


class Politik(Page):
    def vars_for_template(self):
        return {'random_foreign_candidate': self.player.random_foreign_candidate,
                'random_beruf_cdu': self.player.random_beruf_cdu,
                'random_beruf_spd': self.player.random_beruf_spd,
                'random_liste_cdu': self.player.random_liste_cdu,
                'random_liste_spd': self.player.random_liste_spd,
                'random_liste_gruene': self.player.random_liste_gruene,
                'random_liste_fdp': self.player.random_liste_fdp,
                'random_liste_afd': self.player.random_liste_afd,
                'random_liste_linke': self.player.random_liste_linke,
                'random_lastname_cdu': self.player.random_lastname_cdu,
                'random_firstname_cdu': self.player.random_firstname_cdu,
                'random_lastname_spd': self.player.random_lastname_spd,
                'random_firstname_spd': self.player.random_firstname_spd,
                'random_lastname_gruene': self.player.random_lastname_gruene,
                'random_firstname_gruene': self.player.random_firstname_gruene,
                'random_lastname_fdp': self.player.random_lastname_fdp,
                'random_firstname_fdp': self.player.random_firstname_fdp,
                'random_lastname_afd': self.player.random_lastname_afd,
                'random_firstname_afd': self.player.random_firstname_afd,
                'random_lastname_linke': self.player.random_lastname_linke,
                'random_firstname_linke': self.player.random_firstname_linke,
                'random_lastname_foreign': self.player.random_lastname_foreign,
                'random_firstname_foreign': self.player.random_firstname_foreign
                }

    form_model = Player
    form_fields = ['vote_candidate', 'vote_party', 'candname_cdu', 'candname_spd', 'candname_gruene',
                   'candname_fdp', 'candname_afd', 'candname_linke','time_politik']

class SkalometerPartei(Page):
    form_model = Player
    form_fields = ['haltung_cdu', 'haltung_csu', 'haltung_spd', 'haltung_gruene',
                   'haltung_fdp', 'haltung_afd', 'haltung_linke', 'time_skalometer_partei']


class WahlOB(Page):
    def vars_for_template(self):
        return {'random_age_mayor1': self.player.random_age_mayor1,
                'random_age_mayor2': self.player.random_age_mayor2,
                'random_dr_mayor1': self.player.random_dr_mayor1,
                'random_dr_mayor2': self.player.random_dr_mayor2,
                'random_beruf_mayor1': self.player.random_beruf_mayor1,
                'random_beruf_mayor2': self.player.random_beruf_mayor2
                }

    form_model = Player
    form_fields = ['vote_mayor', 'time_vote_mayor']


class WahlBT(Page):

    def vars_for_template(self):
        return {'random_age_mp1': self.player.random_age_mp1,
                'random_age_mp2': self.player.random_age_mp2,
                'random_dr_mp1': self.player.random_dr_mp1,
                'random_dr_mp2': self.player.random_dr_mp2,
                'random_beruf_mp1': self.player.random_beruf_mp1,
                'random_beruf_mp2': self.player.random_beruf_mp2
                }

    form_model = Player
    form_fields = ['vote_mp', 'time_vote_mp']


class RandomNumber(Page):

    def vars_for_template(self):
        return {'randomnumberdue': self.player.randomnumberdue}

    form_model = Player
    form_fields = ['rand_num', 'time_randomNumber']


class Matrikelnummer(Page):
    form_model = Player
    form_fields = ['matrikelnummer','time_matrikelnr']


class End(Page):
    form_model = Player
    form_fields = ['time_end']


page_sequence = [Welcome,
                 Klausur,
                 Internet,
                 Conjoint_Pakete,
                 Wahlzettel_Ankuendigung,
                 Politik,
                 WahlOB,
                 WahlBT,
                 SkalometerPartei,
                 RandomNumber,
                 Matrikelnummer,
                 End]