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


author = 'Your name here'

doc = """
Results are presented here
"""


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    Results_warnings = models.IntegerField()
    Results_time_spent= models.FloatField(blank=True)


def custom_export(players):
    yield ['Session','Prolific ID','Payoff']
    for p in players:
        p_vars = p.participant.vars
        if p.participant._index_in_pages == p.participant._max_page_index:
            try:
                prolific_id = p_vars['prolific_id']
                payoff = float(p.participant.payoff.to_real_world_currency(p.session))
                yield [p.session.code,prolific_id,payoff]
            except KeyError:
                pass
        else:
            pass