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


author = 'Your name here'

doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'colors'
    players_per_group = 2
    num_rounds = 10

    match_payoff = c(10)

class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):

    order1 = models.IntegerField()

    order2 = models.IntegerField()

    def set_payoffs(self):
        p1 = self.get_player_by_id(1)
        p2 = self.get_player_by_id(2)

        if p1.color == p2.color:
            p1.payoff = p2.payoff = Constants.match_payoff
            p1.match = p2.match = True
        else:
            p1.payoff = p2.payoff = c(0)
            p1.match = p2.match = False


class Player(BasePlayer):

    color = models.StringField()

    match = models.BooleanField()