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 = 'AppFlakyFish'
    players_per_group = None
    num_rounds = 17

    import datetime as dt
    hour = dt.datetime.now().strftime('%H')
    minute = dt.datetime.now().strftime('%M')
    second = dt.datetime.now().strftime('%S')
    seed = float(hour)+float(minute)+float(second)

    # randomSeed = [1.1,2.2,3.3,4.4,5.5,6.6,7.7,8.8,9.8,10.8,118,1234,133,143,153,163,173] # 设置每一轮的随机种子
    randomSeed = [seed,seed+1,seed+2,seed+3,seed+4,seed+5,seed+6,seed+7,seed+8,seed+9,seed+10,seed+11,seed+12,seed+13,seed+14,seed+15,seed+16] # 设置每一轮的随机种子
    randomBottom = 0.01  # 股票随机波动不能小于此值 ，如果为 None 不设置下限

class Subsession(BaseSubsession):

    pass

class Group(BaseGroup):
    stock_1_price = models.FloatField(min=0)
    stock_2_price = models.FloatField(min=0)
    stock_3_price = models.FloatField(min=0)
    stock_4_price = models.FloatField(min=0)
    stock_5_price = models.FloatField(min=0)
    stock_6_price = models.FloatField(min=0)

    def update_stock_price(self):
        import random
        roundNow = self.round_number
        rng = random.Random(Constants.randomSeed[roundNow-1])

        def generate_random(stock_name):
            # 根据规则生成每轮的随机变化量
            # input :{ stock_name:string变量，指定股票的名称如'stock_1'}
            # output:{ generated_amount:c变量，股票变更的值}
            def is_up_or_down(stock_name):
                # 根据规则判定涨或跌
                # input :{ stock_name:string变量，指定股票的名称如'stock_1'}
                # output:{ up_or_down:string变量，股票变更的方向，如'up'}
                if   stock_name == 'stock_1_price':
                    random_number = rng.random()
                    if random_number <= 0.65:
                        return 'up'
                    else:
                        return 'down'

                elif stock_name == 'stock_2_price':
                    random_number = rng.random()
                    if random_number <= 0.55:
                        return 'up'
                    else:
                        return 'down'

                elif stock_name == 'stock_3_price':
                    random_number = rng.random()
                    if random_number <= 0.50:
                        return 'up'
                    else:
                        return 'down'

                elif stock_name == 'stock_4_price':
                    random_number = rng.random()
                    if random_number <= 0.50:
                        return 'up'
                    else:
                        return 'down'

                elif stock_name == 'stock_5_price':
                    random_number = rng.random()
                    if random_number <= 0.45:
                        return 'up'
                    else:
                        return 'down'

                elif stock_name == 'stock_6_price':
                    random_number = rng.random()
                    if random_number <= 0.35:
                        return 'up'
                    else:
                        return 'down'

                else:
                    import os
                    print('无此股票')
                    os.exit(0)

            def change_amount(regular='regular_1'):
                # 根据规则判定涨跌数量
                def regular_1():
                    # 生成0-5的随机浮点数
                    amount = rng.random()*0.5
                    amount = round(amount, 2)
                    return amount

                def regular_2():
                    # 生成1,3,5的随机整数
                    values = [1.0, 3.0, 5.0]
                    amount = rng.choice(values)
                    return amount

                if regular == 'regular_1':
                    return regular_1()
                else:
                    return regular_2()

            direction = is_up_or_down(stock_name)
            amount = change_amount()
            if direction == 'up':
                return amount
            else:
                return -amount

        if self.round_number == 1:
            ## 设置股票初始价格
            stock_1_price = round(rng.random()*2+1, 2) #随机在1-3间
            stock_2_price = round(rng.random()*2+1, 2)
            stock_3_price = round(rng.random()*2+1, 2)
            stock_4_price = round(rng.random()*2+1, 2)
            stock_5_price = round(rng.random()*2+1, 2)
            stock_6_price = round(rng.random()*2+1, 2)
        else:
            stock_1_price = self.in_round(self.round_number - 1).stock_1_price
            newPrice = stock_1_price + generate_random(stock_name='stock_1_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_1_price = newPrice

            stock_2_price = self.in_round(self.round_number - 1).stock_2_price
            newPrice =  stock_2_price + generate_random(stock_name='stock_2_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_2_price = newPrice

            stock_3_price = self.in_round(self.round_number - 1).stock_3_price
            newPrice = stock_3_price + generate_random(stock_name='stock_3_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_3_price = newPrice

            stock_4_price = self.in_round(self.round_number - 1).stock_4_price
            newPrice = stock_4_price + generate_random(stock_name='stock_4_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_4_price = newPrice

            stock_5_price = self.in_round(self.round_number - 1).stock_5_price
            newPrice = stock_5_price + generate_random(stock_name='stock_5_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_5_price = newPrice


            stock_6_price = self.in_round(self.round_number - 1).stock_6_price
            newPrice = stock_6_price + generate_random(stock_name='stock_6_price')
            if Constants.randomBottom is not None:
                if newPrice<Constants.randomBottom:
                    newPrice = Constants.randomBottom
            stock_6_price = newPrice

        self.stock_1_price = stock_1_price
        self.stock_2_price = stock_2_price
        self.stock_3_price = stock_3_price
        self.stock_4_price = stock_4_price
        self.stock_5_price = stock_5_price
        self.stock_6_price = stock_6_price

    def stock_1_price_history(self):
        return [round(g.stock_1_price, 2) for g in self.in_previous_rounds()]

    def stock_2_price_history(self):
        return [round(g.stock_2_price, 2) for g in self.in_previous_rounds()]

    def stock_3_price_history(self):
        return [round(g.stock_3_price, 2) for g in self.in_previous_rounds()]

    def stock_4_price_history(self):
        return [round(g.stock_4_price, 2) for g in self.in_previous_rounds()]

    def stock_5_price_history(self):
        return [round(g.stock_5_price, 2) for g in self.in_previous_rounds()]

    def stock_6_price_history(self):
        return [round(g.stock_6_price, 2) for g in self.in_previous_rounds()]


class Player(BasePlayer):

    money = models.FloatField(min=0)

    stock_1_amount = models.IntegerField(min=0, initial=0)
    stock_2_amount = models.IntegerField(min=0, initial=0)
    stock_3_amount = models.IntegerField(min=0, initial=0)
    stock_4_amount = models.IntegerField(min=0, initial=0)
    stock_5_amount = models.IntegerField(min=0, initial=0)
    stock_6_amount = models.IntegerField(min=0, initial=0)

    stock_1_bid_amount = models.IntegerField(label='你要购买多少股票1', min=0, initial=0)
    stock_2_bid_amount = models.IntegerField(label='你要购买多少股票2', min=0, initial=0)
    stock_3_bid_amount = models.IntegerField(label='你要购买多少股票3', min=0, initial=0)
    stock_4_bid_amount = models.IntegerField(label='你要购买多少股票4', min=0, initial=0)
    stock_5_bid_amount = models.IntegerField(label='你要购买多少股票5', min=0, initial=0)
    stock_6_bid_amount = models.IntegerField(label='你要购买多少股票6', min=0, initial=0)

    stock_1_sell_amount = models.IntegerField(label='你要出售多少股票1', min=0, initial=0)
    stock_2_sell_amount = models.IntegerField(label='你要出售多少股票2', min=0, initial=0)
    stock_3_sell_amount = models.IntegerField(label='你要出售多少股票3', min=0, initial=0)
    stock_4_sell_amount = models.IntegerField(label='你要出售多少股票4', min=0, initial=0)
    stock_5_sell_amount = models.IntegerField(label='你要出售多少股票5', min=0, initial=0)
    stock_6_sell_amount = models.IntegerField(label='你要出售多少股票6', min=0, initial=0)

    stock_1_net_amount = models.IntegerField(initial=0)
    stock_2_net_amount = models.IntegerField(initial=0)
    stock_3_net_amount = models.IntegerField(initial=0)
    stock_4_net_amount = models.IntegerField(initial=0)
    stock_5_net_amount = models.IntegerField(initial=0)
    stock_6_net_amount = models.IntegerField(initial=0)

    stock_1_cost = models.FloatField()
    stock_2_cost = models.FloatField()
    stock_3_cost = models.FloatField()
    stock_4_cost = models.FloatField()
    stock_5_cost = models.FloatField()
    stock_6_cost = models.FloatField()

    stock_1_profit = models.FloatField(initial=0)
    stock_2_profit = models.FloatField(initial=0)
    stock_3_profit = models.FloatField(initial=0)
    stock_4_profit = models.FloatField(initial=0)
    stock_5_profit = models.FloatField(initial=0)
    stock_6_profit = models.FloatField(initial=0)

    final_money = models.FloatField()

    def update_stock_1_amount(self):
        if self.round_number == 5 and self.stock_1_amount is None:
            self.stock_1_amount = 0
        elif self.round_number > 5:
            self.stock_1_amount = self.in_round(self.round_number - 1).stock_1_amount

    def update_stock_2_amount(self):
        if self.round_number == 5 and self.stock_2_amount is None:
            self.stock_2_amount = 0
        elif self.round_number > 5:
            self.stock_2_amount = self.in_round(self.round_number - 1).stock_2_amount

    def update_stock_3_amount(self):
        if self.round_number == 5 and self.stock_3_amount is None:
            self.stock_3_amount = 0
        elif self.round_number > 5:
            self.stock_3_amount = self.in_round(self.round_number - 1).stock_3_amount

    def update_stock_4_amount(self):
        if self.round_number == 5 and self.stock_4_amount is None:
            self.stock_4_amount = 0
        elif self.round_number > 5:
            self.stock_4_amount = self.in_round(self.round_number - 1).stock_4_amount

    def update_stock_5_amount(self):
        if self.round_number == 5 and self.stock_5_amount is None:
            self.stock_5_amount = 0
        elif self.round_number > 5:
            self.stock_5_amount = self.in_round(self.round_number - 1).stock_5_amount

    def update_stock_6_amount(self):
        if self.round_number == 5 and self.stock_6_amount is None:
            self.stock_6_amount = 0
        elif self.round_number > 5:
            self.stock_6_amount = self.in_round(self.round_number - 1).stock_6_amount

# 更新钱
    def update_money(self):
        if self.round_number == 1:
            money = 10000.0
        else:
            money = self.in_round(self.round_number - 1).money
        self.money = money

# 更新每只股票成本
    def update_stock_1_cost(self):
        if self.round_number == 1:
            stock_1_cost = 0.0
        else:
            stock_1_cost = self.in_round(self.round_number - 1).stock_1_cost
        self.stock_1_cost = stock_1_cost

    def update_stock_2_cost(self):
        if self.round_number == 1:
            stock_2_cost = 0.0
        else:
            stock_2_cost = self.in_round(self.round_number - 1).stock_2_cost
        self.stock_2_cost = stock_2_cost

    def update_stock_3_cost(self):
        if self.round_number == 1:
            stock_3_cost = 0.0
        else:
            stock_3_cost = self.in_round(self.round_number - 1).stock_3_cost
        self.stock_3_cost = stock_3_cost

    def update_stock_4_cost(self):
        if self.round_number == 1:
            stock_4_cost = 0.0
        else:
            stock_4_cost = self.in_round(self.round_number - 1).stock_4_cost
        self.stock_4_cost = stock_4_cost

    def update_stock_5_cost(self):
        if self.round_number == 1:
            stock_5_cost = 0.0
        else:
            stock_5_cost = self.in_round(self.round_number - 1).stock_5_cost
        self.stock_5_cost = stock_5_cost

    def update_stock_6_cost(self):
        if self.round_number == 1:
            stock_6_cost = 0.0
        else:
            stock_6_cost = self.in_round(self.round_number - 1).stock_6_cost
        self.stock_6_cost = stock_6_cost

# 历史持有量
    def stock_1_net_amount_history(self):
        return [g.stock_1_net_amount for g in self.in_all_rounds()]

    def stock_2_net_amount_history(self):
        return [g.stock_2_net_amount for g in self.in_all_rounds()]

    def stock_3_net_amount_history(self):
        return [g.stock_3_net_amount for g in self.in_all_rounds()]

    def stock_4_net_amount_history(self):
        return [g.stock_4_net_amount for g in self.in_all_rounds()]

    def stock_5_net_amount_history(self):
        return [g.stock_5_net_amount for g in self.in_all_rounds()]

    def stock_6_net_amount_history(self):
        return [g.stock_6_net_amount for g in self.in_all_rounds()]