from otree.api import (
    models,
    widgets,
    BaseConstants,
    BaseSubsession,
    BaseGroup,
    BasePlayer,
    Currency as c,
    currency_range,
)
import sqlite3


author = 'Marta Kozakiewicz'

doc = """
Retrieve the IQ results.
"""


class Constants(BaseConstants):
    name_in_url = 'iq_results'
    players_per_group = None
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass
        

class Player(BasePlayer):
    uniqueID = models.StringField(label="Your unique ID:",initial="P0")
    TOT_EARN = models.FloatField(label="Gesamtverdienst:",initial=0.0)
    IQ = models.FloatField(label="Betrag, der im IQ-Test verdient wurde:",initial=0)
    TASK = models.IntegerField(label="Zur Zahlung gezogene Aufgabe:",initial=0)
    BOX = models.IntegerField(label="Sie haben die folgende Box erraten:",initial=0)
    RANK = models.IntegerField(label="Echter Rang:",initial=0)
    S = models.FloatField(label="Summe S:",initial=0.0)
    NUM = models.FloatField(label="Zahl gezogen:",initial=0.0)
    MONEY_TASK = models.FloatField(label="Verdientes Geld für die Aufgabe:",initial=0.0)
    TEST_DATE = models.StringField(label="Date of test:",initial='01/01/1900') 
    NEW_DATE = models.StringField(label="Date of test:",initial='') 
    
    correct_flag = models.BooleanField(label="flag",initial=0)
    date_flag = models.BooleanField(label="flag",initial=1)

def custom_export(players):
    # header row
    yield ['uniqueID', 'correct_flag']
    for p in players:
        yield [p.uniqueID, p.correct_flag]