import numpy as np
import os
from otree.api import *
import pandas as pd




doc = """
Experiment 2 for Social Inference project
"""


# OTREE CLASSES --------------------------------------------------------------------------------------------------------
class C(BaseConstants):
    NAME_IN_URL = 'SI_Exp2_Payment'
    PLAYERS_PER_GROUP = None
    NUM_ROUNDS = 1


# I think I may need 4 subsessions: 1) instructions + test, 2) MAB, 3) What do you think this was about? 4) Demographics + payment questionnaire
class Subsession(BaseSubsession):
    pass

class Group(BaseGroup):
    pass

class Player(BasePlayer):
    PaymentUsername = models.StringField(blank=True)
    VenmoPhone = models.StringField(blank=True)
    reward = models.FloatField()

    def store_round_data(self):
        self.round_data_field = self.round_number  # Storing round number as an example

    def before_next_page(self):
        self.store_round_data()

    def increment_counter(self):
        self.counter += 1
        return self.counter

# PAGES (ASK ABOUT TOP CHOICES: SELECT & INDICATE) ---------------------------------------------------------------------


class PaymentInfo(Page):
    template_name = 'SI_Exp2_Payment/PaymentInfo.html'

    form_model = 'player'
    form_fields = ['reward', 'PaymentUsername', 'VenmoPhone']

    #print(player.participant.vars)

    def vars_for_template(player: Player):
        # PLACERHOLDER
        reward = 1 #player.participant.vars['payout']
        return {'reward': reward}


class LastPage(Page):
    pass

# EXECUTE---------------------------------------------------------------------------------------------------------------

page_sequence = [PaymentInfo, LastPage]