from otree.api import *

from .models import Player


# ==========================================================
# Ending Page
# ==========================================================
def get_language(player):
    return player.session.config.get('language', 'en') == 'en'

class Ending_Page(Page):

    def vars_for_template(player):
        return dict(
            en=get_language(player),
        )


# ==========================================================
# Time Preference
# ==========================================================

class Time_Preference(Page):
    form_model = 'player'

    form_fields = [
        'time_preference_1',
        'time_preference_2',
        'time_preference_3',
        'time_preference_4',
        'time_preference_5',
        'time_preference_6',
        'time_preference_7',
        'time_preference_8',
        'time_preference_9',
        'time_preference_10',
    ]

    def vars_for_template(player):
        return dict(
            en=get_language(player),
        )


# ==========================================================
# Payment Information Consent
# ==========================================================

class Payment_Info_Consent(Page):
    form_model = 'player'
    form_fields = [
        'payment_info_consent',
    ]

    def vars_for_template(player):
        return dict(
            en=get_language(player),
        )



# ==========================================================
# Payment Information
# ==========================================================

class Payment_Info(Page):
    form_model = 'player'

    form_fields = [
        'payment_method_1',
        'payment_method_2',
        'ewallet_number',
        'bank_name',
        'bank_acc_num',
        'email_address',
    ]

    def is_displayed(self):
        return self.player.payment_info_consent

    def vars_for_template(player):
        return dict(
            en=get_language(player),
        )


# ==========================================================
# Contact Page
# ==========================================================

class Contact_Page(Page):

    def vars_for_template(player):
        return dict(
            en=get_language(player),
        )


# ==========================================================
# Page Sequence
# ==========================================================

page_sequence = [
    Ending_Page,
    Time_Preference,
    Payment_Info_Consent,
    Payment_Info,
    Contact_Page,
]