from otree.api import *
c = cu

doc = 'Implementation of the p-mpl method.'
class C(BaseConstants):
    NAME_IN_URL = 'pmpl'
    PLAYERS_PER_GROUP = None
    NUM_ROUNDS = 30
    NUM_PRODUCTS = 30
    ENDOWMENT = 10
    MAX = 10
    PRODUCTS = ('Bare', 'Cheetos', 'CheezitOriginal', 'ChipsAhoy', 'Coke', 'CokeZero', 'Doritos', 'Flipz', 'Goldfish', 'Haribo', 'HersheysMilkChocolate', 'IceBreakersPeppermint', 'KindCaramelAlmond', 'KindDarkChocolate', 'Lays', 'LotusBiscoff', 'MilanoDarkChocolate', 'MilanoMilkChocolate', 'MNM', 'Oreo', 'PockyChocolate', 'PockyStrawberry', 'PopTartsFrostedCookies', 'PopTartsFrostedSmores', 'Pringles', 'PringlesOnion', 'Skittles', 'Snickers', 'Sprite', 'TwixBar')
    PRODUCTS_DESCRIPTIONS = ('Bare Fugi & Reds Apple Chips (3.4 oz)', 'Cheetos Crunchy (8.5 oz)', 'Cheezit Original (12.4 oz)', 'Chips Ahoy Real Chocolate Chip Cookies Original (13 oz)', 'Coke (12 fl oz)', 'Coke Zero Sugar (12 fl oz)', 'Doritos Nacho Cheese (9.25 oz)', 'Flipz Milk Chocolate Covered Pretzels (7.5 oz)', 'Goldfish Cheddar (6.6 oz)', 'Haribo Gummi Candy (8 oz)', "Hershey's Milk Chocolate (1.55 oz)", 'Ice Breaker Ice Cubes Peppermint Flavored (3.24 oz, 40 pieces)', 'KIND Caramel Almond & Sea Salt (1.4 oz)', 'KIND Dark Chocolate Nuts & Sea Salt (1.4 oz)', "Lay's Classic (8 oz)", 'Lotus Biscoff Cookies (8.8 oz)', 'Milano Cookies Double Dark Chocolate (7.5 oz)', 'Milano Cookies Milk Chocolate (6 oz)', "M&M's Milk Chocolate (3.14 oz)", 'OREO Chocolate Sandwich Cookies (14.3 oz)', 'Pocky Chocolate Cream (2.47 oz)', 'Pocky Strawberry Cream (2.47 oz)', 'Pop-Tarts Frosted Cookies and Crème (13.5 oz, 8 Pop-Tarts)', "Pop-Tarts Frosted S'mores (13.5 oz, 8 Pop-Tarts)", 'Pringles Original (5.2 oz)', 'Pringles Sour Cream & Onion (5.5 oz)', 'Skittles Original (2.17 oz)', 'Snickers (1.86 oz)', 'Sprite (12 fl oz)', 'Twix (1.79 oz)')
    EXAMPLE = 'Cheeseburger'
    MAX_QUESTION = 1000
class Subsession(BaseSubsession):
    pass
def creating_session(subsession: Subsession):
    session = subsession.session
    round_number = subsession.round_number
    players = subsession.get_players()
    if round_number == 1:
        import random
        for p in players:
            participant = p.participant
            participant.vars['mpl_products_order'] = random.sample(C.PRODUCTS, k=len(C.PRODUCTS))
            participant.pmpl_round_to_pay = random.randint(1, C.NUM_PRODUCTS)
            participant.pmpl_question_to_pay = random.randint(1, C.MAX_QUESTION)
            if participant.mpl_treatment == "pmpl":
                participant.mpl_round_to_pay = participant.pmpl_round_to_pay
                participant.mpl_question_to_pay = participant.pmpl_question_to_pay
    
    for p in players:
        p.product = p.participant.vars['mpl_products_order'][round_number - 1]
class Group(BaseGroup):
    pass
class Player(BasePlayer):
    decision_question = models.IntegerField()
    product = models.StringField(choices=C.PRODUCTS)
    decision = models.FloatField(max=C.MAX, min=0.01)
    quiz_1a = models.StringField(choices=[['True', 'True'], ['False', 'False']], widget=widgets.RadioSelect)
    quiz_1b = models.StringField(choices=[['X', 'Option X'], ['Y', 'Option Y']], widget=widgets.RadioSelect)
    quiz_1c = models.StringField(choices=[['True', 'True'], ['False', 'False']], widget=widgets.RadioSelect)
    quiz_2a = models.StringField(choices=[['True', 'True'], ['False', 'False']], widget=widgets.RadioSelect)
    quiz_2b = models.StringField(choices=[['X', 'Option X'], ['Y', 'Option Y']], widget=widgets.RadioSelect)
    quiz_2c = models.StringField(choices=[['True', 'True'], ['False', 'False']], widget=widgets.RadioSelect)
def set_payoff(player: Player):
    participant = player.participant
    player_to_pay = player.in_round(participant.pmpl_round_to_pay)
    participant.pmpl_decision = player_to_pay.decision
    participant.pmpl_decision_question = int(player_to_pay.decision * 100)
    participant.pmpl_product = player_to_pay.product
    
    if participant.pmpl_question_to_pay < participant.pmpl_decision_question:
        participant.pmpl_money = C.ENDOWMENT - round(0.01 * participant.pmpl_question_to_pay, 2)
    
    else:
        participant.pmpl_money = C.ENDOWMENT
    
    if participant.mpl_treatment == "pmpl":
        participant.mpl_decision = participant.pmpl_decision
        participant.mpl_decision_question = participant.pmpl_decision_question
        participant.mpl_money = participant.pmpl_money
        participant.mpl_product = participant.pmpl_product
    
class InstructionB(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1
class QuizB(Page):
    form_model = 'player'
    form_fields = ['quiz_1a', 'quiz_1b', 'quiz_1c', 'quiz_2a', 'quiz_2b', 'quiz_2c']
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1
class AnswerB(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1
class DecisionB(Page):
    form_model = 'player'
    form_fields = ['decision']
    @staticmethod
    def vars_for_template(player: Player):
        return dict(
            product_description=C.PRODUCTS_DESCRIPTIONS[C.PRODUCTS.index(player.product)],
            product_img=f'mpl/{player.product}.jpg'
        )
    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        if player.round_number == C.NUM_ROUNDS:
            set_payoff(player)
        
        player.decision = round(player.decision, 2)
page_sequence = [InstructionB, QuizB, AnswerB, DecisionB]