from typing import List

# TODO: Format labels on session creation
# TODO: Pass correct class according to session in session fields


class Labels:
    # PaymentQuestions1
    satisf_fixedwage = "Ich bin zufrieden mit der fixen Vergütung in Höhe von <b>{fixed_wage}</b>."
    fair_fixedwage = "Ich finde die fixe Vergütung in Höhe von <b>{fixed_wage}</b> fair."

    # TaskQuestions1_1
    task_motivation = "Ich <b>habe mich angestrengt</b>, um die Aufgaben in diesem Experiment zu lösen."
    task_fun = "Ich hatte <b>Spaß</b> an den Aufgaben."
    task_difficult = "Ich fand die Aufgaben <b>schwierig</b>."
    task_exhausting = "Ich empfand die Aufgaben als <b>ermüdend</b>."
    task_obligation = "Ich fühlte mich zur <b>Bearbeitung der Aufgaben verpflichtet</b>."
    task_pressure = "Ich fühlte mich unter <b>Druck gesetzt Leistung zu bringen</b>."
    task_stress = "Ich fühlte mich <b>gestresst</b>."
    task_boredom = "Ich fühlte mich <b>gelangweilt</b>."
    task_concentration = "Ich fand es <b>schwer mich zu konzentrieren</b>."
    # TaskQuestions1_2
    task_competition = "Ich wollte <b>besser sein</b> als die anderen <b>ExperimentteilnehmerInnen</b>."
    task_best = "Ich wollte <b>die/der Beste</b> sein."
    task_reciprocity = "Ich wollte mich <b>im Gegenzug für die Bezahlung</b> anstrengen."
    task_recognition = "Ich wollte von Anderen <b>Anerkennung erhalten</b>."
    task_diligent = "Ich wollte die Aufgaben <b>gewissenhaft</b> bearbeiten"
    task_fixedwage = "Ich war durch die fixe Zahlung in Höhe <b>von {fixed_wage} motiviert</b>."
    task_competition2 = "Ich wollte <b>nicht schlechter sein</b> als die anderen ExperimentteilnehmerInnen."
    task_notworst = "Ich wollte <b>nicht die/der Schlechteste</b> sein."
    # TaskQuestions2_2
    outsideoption_interesting = "Ich fand die Artikel aus dem TUM-Magazin „Faszination Forschung“ <b>interessant</b>."
    outsideoption1 = "Die Artikel aus dem TUM-Magazin „Faszination Forschung“ haben mich <b>davon abgehalten</b> die mathematischen Aufgaben zu lösen."
    outsideoption2 = "Ich wollte <b>keine</b> mathematischen Aufgaben <b>lösen</b>."
    outsideoption_fun = "Ich hatte <b>Spaß</b> die Artikel aus dem TUM-Magazin „Faszination Forschung“ zu lesen."

    # AttitudeQuestions1
    comp1 = "Ich arbeite gerne in Situationen, in denen <b>ich mit anderen im Wettbewerb</b> stehe."
    comp2 = "Es ist mir wichtig, bei einer Aufgabe <b>besser abzuschneiden als andere</b>."
    comp3 = "Ich glaube, dass <b>Gewinnen</b> sowohl bei der Arbeit als auch in Spielen wichtig ist."
    comp4 = "Ich strenge mich mehr an, wenn ich <b>im Wettbewerb mit Anderen</b> stehe."

    # AttitudeQuestions2
    rec1 = "Wenn mir jemand einen <b>Gefallen</b> tut, bin ich bereit, dies <b>zu erwidern</b>."
    rec2 = "Wenn mir <b>schweres Unrecht zuteil</b> wird, werde ich mich um jeden Preis bei der nächsten Gelegenheit dafür <b>rächen</b>."
    rec3 = "Wenn mich jemand <b>in eine schwierige Lage bringt</b>, werde ich das <b>Gleiche</b> mit ihm machen."
    rec4 = "Ich strenge mich besonders an, um <b>jemandem zu helfen</b>, der mir früher <b>schon mal geholfen</b> hat."
    rec5 = "Wenn mich jemand <b>beleidigt</b>, werde ich mich ihm gegenüber <b>auch beleidigend verhalten</b>."
    rec6 = "Ich bin bereit, Kosten auf mich zu nehmen, um <b>jemandem zu helfen</b>, der mir früher <b>einmal geholfen</b> hat."

    # AttitudeQuestions3
    risk_gen = "Sind Sie im Allgemeinen ein risikobereiter Mensch oder versuchen Sie, Risiken zu vermeiden?"

    # AttitudeQuestions4
    risk1 = "beim Autofahren?"
    risk2 = "bei Geldanlagen?"
    risk3 = "bei Freizeit und Sport?"
    risk4 = "bei Ihrer beruflichen Karriere?"
    risk5 = "bei Ihrer Gesundheit?"
    risk6 = "beim Vertrauen in fremde Menschen?"

    def __init__(self, **formatters) -> None:
        """
        Format all labels with the fixed wage of the treatment
        """
        self.satisf_fixedwage = self.satisf_fixedwage.format(**formatters)
        self.fair_fixedwage = self.fair_fixedwage.format(**formatters)
        self.task_fixedwage = self.task_fixedwage.format(**formatters)

    def get_labels(self, page) -> List[str]:
        """
        Return a tuple (field_name, field_label) of all field for the specified page
        """
        labels = [(f, getattr(self, f)) for f in self.get_form_fields(page)]
        return labels


class FixedWageLabels(Labels):
    def __init__(self, **formatters) -> None:
        """
        Format all labels with the fixed wage of the treatment
        """
        super().__init__(**formatters)

    @staticmethod
    def get_form_fields(page) -> List[str]:
        """
        Return the form fields for the specificed page
        """
        page_name = page.__name__
        if page_name == "PaymentQuestions1":
            return ["satisf_fixedwage", "fair_fixedwage"]
        if page_name == "TaskQuestions1_1":
            return [
                "task_motivation",
                "task_fun",
                "task_difficult",
                "task_exhausting",
                "task_obligation",
                "task_pressure",
                "task_stress",
                "task_boredom",
                "task_concentration",
            ]
        if page_name == "TaskQuestions1_2":
            return [
                "task_competition",
                "task_best",
                "task_reciprocity",
                "task_recognition",
                "task_diligent",
                "task_fixedwage",
                "task_competition2",
                "task_notworst",
            ]
        if page_name == "TaskQuestions2_2":
            return ["outsideoption_interesting", "outsideoption1", "outsideoption2", "outsideoption_fun"]
        if page_name == "AttitudeQuestions1":
            return ["comp1", "comp2", "comp3", "comp4"]
        if page_name == "AttitudeQuestions2":
            return ["rec1", "rec2", "rec3", "rec4", "rec5", "rec6"]
        if page_name == "AttitudeQuestions3":
            return ["risk_gen"]
        if page_name == "AttitudeQuestions4":
            return ["risk1", "risk2", "risk3", "risk4", "risk5", "risk6"]


class PieceRateLabels(Labels):
    # PaymentQuestions1
    satisf_piecerate = (
        "Ich bin zufrieden mit der Auszahlung pro richtig gelöster Aufgabe in Höhe von <b>{piece_rate}</b>."
    )
    fair_piecerate = "Ich finde die Auszahlung pro richtig gelöster Aufgabe in Höhe von <b>{piece_rate}</b> fair."

    # TaskQuestions1_2
    task_piecerate = (
        "Ich war durch die Auszahlung in Höhe <b>von {piece_rate} pro richtig gelöster Aufgabe motiviert</b>. "
    )

    def __init__(self, **formatters) -> None:
        """
        Format all labels with the fixed wage and piece rate of the treatment
        """
        super().__init__(**formatters)
        self.satisf_piecerate = self.satisf_piecerate.format(**formatters)
        self.fair_piecerate = self.fair_piecerate.format(**formatters)
        self.task_piecerate = self.task_piecerate.format(**formatters)

    @staticmethod
    def get_form_fields(page) -> List[str]:
        """
        Return the form fields for the specificed page
        """
        page_name = page.__name__
        if page_name == "PaymentQuestions1":
            return ["satisf_fixedwage", "satisf_piecerate", "fair_fixedwage", "fair_piecerate"]
        if page_name == "TaskQuestions1_1":
            return [
                "task_motivation",
                "task_fun",
                "task_difficult",
                "task_exhausting",
                "task_obligation",
                "task_pressure",
                "task_stress",
                "task_boredom",
                "task_concentration",
            ]
        if page_name == "TaskQuestions1_2":
            return [
                "task_competition",
                "task_best",
                "task_reciprocity",
                "task_recognition",
                "task_diligent",
                "task_fixedwage",
                "task_piecerate",
                "task_competition2",
                "task_notworst",
            ]
        if page_name == "TaskQuestions2_2":
            return ["outsideoption_interesting", "outsideoption1", "outsideoption2", "outsideoption_fun"]
        if page_name == "AttitudeQuestions1":
            return ["comp1", "comp2", "comp3", "comp4"]
        if page_name == "AttitudeQuestions2":
            return ["rec1", "rec2", "rec3", "rec4", "rec5", "rec6"]
        if page_name == "AttitudeQuestions3":
            return ["risk_gen"]
        if page_name == "AttitudeQuestions4":
            return ["risk1", "risk2", "risk3", "risk4", "risk5", "risk6"]


class TournamentLabels(Labels):
    # PaymentQuestions1
    satisf_prize = "Ich bin mit den zu erreichenden Preisen in Höhe von <b>{prize1}</b>, <b>{prize2}</b>, <b>{prize3}</b> bzw. <b>{prize4}</b> zufrieden."
    fair_prize = "Ich finde die zu erreichenden Preise in Höhe von <b>{prize1}</b>, <b>{prize2}</b>, <b>{prize3}</b> bzw. <b>{prize4}</b> fair."

    # TaskQuestions1_2
    task_prize = "Ich war durch die Preise in Höhe von <b>{prize1}</b>, <b>{prize2}</b>, <b>{prize3}</b> bzw. <b>{prize4} motiviert</b>."

    def __init__(self, **formatters) -> None:
        """
        Format all labels with the fixed wage and tournament prizes of the treatment
        """
        super().__init__(**formatters)
        self.satisf_prize = self.satisf_prize.format(**formatters)
        self.fair_prize = self.fair_prize.format(**formatters)
        self.task_prize = self.task_prize.format(**formatters)

    @staticmethod
    def get_form_fields(page) -> List[str]:
        """
        Return the form fields for the specificed page
        """
        page_name = page.__name__
        if page_name == "PaymentQuestions1":
            return ["satisf_fixedwage", "satisf_prize", "fair_fixedwage", "fair_prize"]
        if page_name == "TaskQuestions1_1":
            return [
                "task_motivation",
                "task_fun",
                "task_difficult",
                "task_exhausting",
                "task_obligation",
                "task_pressure",
                "task_stress",
                "task_boredom",
                "task_concentration",
            ]
        if page_name == "TaskQuestions1_2":
            return [
                "task_competition",
                "task_best",
                "task_reciprocity",
                "task_recognition",
                "task_diligent",
                "task_fixedwage",
                "task_prize",
                "task_competition2",
                "task_notworst",
            ]
        if page_name == "TaskQuestions2_2":
            return ["outsideoption_interesting", "outsideoption1", "outsideoption2", "outsideoption_fun"]
        if page_name == "AttitudeQuestions1":
            return ["comp1", "comp2", "comp3", "comp4"]
        if page_name == "AttitudeQuestions2":
            return ["rec1", "rec2", "rec3", "rec4", "rec5", "rec6"]
        if page_name == "AttitudeQuestions3":
            return ["risk_gen"]
        if page_name == "AttitudeQuestions4":
            return ["risk1", "risk2", "risk3", "risk4", "risk5", "risk6"]