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

class MyWaitPage(WaitPage):
    group_by_arrival_time= True

    pass

class Instructions(Page):
    pass

class MyPage1(Page):
    form_model = 'group'
    form_fields = ['age_1',
                   'gender_1',
                   'ethnic_group_1',
                   'hair_color_1',
                   'hair_type_1',
                   'eye_color_1',
                   'height_1',
                   'body_size_1']

    def is_displayed(self):
        return self.player.id_in_group == 1

    pass


class MyPage2(Page):
    form_model = 'group'
    form_fields = ['age_2',
                   'gender_2',
                   'ethnic_group_2',
                   'hair_color_2',
                   'hair_type_2',
                   'eye_color_2',
                   'height_2',
                   'body_size_2']

    def is_displayed(self):
        return self.player.id_in_group == 2

    pass


class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):

        group = self.group
        p1 = group.get_player_by_id(1)
        p2 = group.get_player_by_id(2)
        p1.image = group.gender_2 + group.ethnic_group_2 + group.hair_type_2 + group.hair_color_2 + group.eye_color_2
        p2.image = group.gender_1 + group.ethnic_group_1 + group.hair_type_1 + group.hair_color_1 + group.eye_color_1


        pass


class Results(Page):
    def vars_for_template(self):
        return {
            'image_path': 'survey_trust/{}.png'.format(self.player.image)
        }
    pass


page_sequence = [
    MyWaitPage,
    Instructions,
    MyPage1,
    MyPage2,
    ResultsWaitPage,
    Results
]