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

class introduction (Page):
    def before_next_page(self):
        import time
        self.player.time_0 = time.strftime("%H:%M:%S", time.gmtime(time.time()))


class letter_page_1(Page):
    form_model = "player"
    form_fields = ["T1_working_man_dec", "T1_working_man_conf", "T2_working_woman_dec", "T2_working_woman_conf", "T3_retired_woman_dec","T3_retired_woman_conf",
                   "T4_enterpreneur_man_dec", "T4_enterpreneur_man_conf", "T5_enterpreneur_woman_dec", "T5_enterpreneur_woman_conf", "T6_unemployed_man_dec","T6_unemployed_man_conf",]
    def vars_for_template(self):
        try:
            if self.player.displayed_letter == 3:
                if self.player.displayed_letter == 3:
                    path_image = "blur_3.png"
                    letter_image = "dopis3"

        except TypeError:
            self.player.displayed_letter = 3
            path_image = "blur_3.png"
            letter_image = "dopis3"

        return dict(image_path=path_image.format(self.player), real_image = letter_image.format(self.player))

    def before_next_page(self):
        import time
        if self.player.T1_working_man_dec == "Nie":
            self.player.payoff += Constants.payment_per_correct_answer
        if self.player.T2_working_woman_dec == "Nie":
            self.player.payoff += Constants.payment_per_correct_answer
        if self.player.T3_retired_woman_dec == "Áno":
            self.player.payoff += Constants.payment_per_correct_answer
        if self.player.T4_enterpreneur_man_dec == "Áno":
            self.player.payoff += Constants.payment_per_correct_answer
        if self.player.T5_enterpreneur_woman_dec == "Áno":
            self.player.payoff += Constants.payment_per_correct_answer
        if self.player.T6_unemployed_man_dec == "Nie":
            self.player.payoff += Constants.payment_per_correct_answer
        self.player.time_1 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page1(Page):
    form_model = 'player'
    form_fields = ['A11', "A12"]

    def before_next_page(self):
        import time
        self.player.time_2 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page1_1 (Page):
    form_model = 'player'
    form_fields = ['A131', "A132", "A133", "A14"]

    def before_next_page(self):
        import time
        self.player.time_3 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class letter_page_2 (Page):
    form_model = 'player'
    def vars_for_template (self):
        if self.player.displayed_letter == 1:
            letter_image_2 = "dopis_1_oslo.png"
        elif self.player.displayed_letter == 2:
            letter_image_2 = "dopis_2_oslo.png"
        else:
            letter_image_2 = "dopis_3_oslo.png"
        return dict(real_image_oslo = letter_image_2.format(self.player))

    def before_next_page(self):
        import time
        self.player.time_4 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page2 (Page):
    form_model = 'player'
    form_fields = ['A15','A151','A16', "A17", "A18", "A19"]

    def before_next_page(self):
        import time
        self.player.time_5 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page3 (Page):
    form_model = 'player'
    form_fields = ['A21','A23', "A24", "A25"]

    def before_next_page(self):
        import time
        self.player.time_6 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page4 (Page):
    form_model = 'player'
    form_fields = ['A31','A32','A33','A34', "A35"]

    def before_next_page(self):
        import time
        self.player.time_7 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page4_1 (Page):
    form_model = 'player'
    form_fields = ["A351", "A36", "A37"]

    def before_next_page(self):
        import time
        self.player.time_8 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page5 (Page):
    form_model = 'player'
    form_fields = ["B31", "B32"]

    def before_next_page(self):
        import time
        self.player.time_9 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page6 (Page):
    form_model = 'player'
    form_fields = ["A41", "A42", "A43"]

    def before_next_page(self):
        import time
        self.player.time_10 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class Survey_page7 (Page):
    form_model = 'player'
    form_fields = ["A51","A52", "A53", "A54"]

    def before_next_page(self):
        import time
        self.player.time_11 = time.strftime("%H:%M:%S", time.gmtime(time.time()))

class results(Page):
    pass

page_sequence = [introduction, letter_page_1, Survey_page1, Survey_page1_1, letter_page_2, Survey_page2, Survey_page3, Survey_page4, Survey_page4_1, Survey_page5, Survey_page6, Survey_page7, results]