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_onepage'
    players_per_group = None
    num_rounds = 18

    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,seed+17] # 设置每一轮的随机种子
    randomBottom = 0.01  # 股票随机波动不能小于此值 ，如果为 None 不设置下限

    trail_template = 'AppFlakyFish_onepage/intro_trail.html'
    forecast_template = 'AppFlakyFish_onepage/intro_forecast.html'
    sellonly_template = 'AppFlakyFish_onepage/intro_sell.html'

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.45:
                        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.65:
                        return 'up'
                    else:
                        return 'down'

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

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

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

            def change_amount(regular='regular_2'):
                # 根据规则判定涨跌数量
                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 = rng.randint(70, 120)#随机整数在100-150间
            stock_2_price = rng.randint(70, 120)
            stock_3_price = rng.randint(70, 120)
            stock_4_price = rng.randint(70, 120)
            stock_5_price = rng.randint(70, 120)
            stock_6_price = rng.randint(70, 120)
            # stock_1_price = round(rng.random()*50+100, 2) #随机在1-3间
            # stock_2_price = round(rng.random()*50+100, 2)
            # stock_3_price = round(rng.random()*50+100, 2)
            # stock_4_price = round(rng.random()*50+100, 2)
            # stock_5_price = round(rng.random()*50+100, 2)
            # stock_6_price = round(rng.random()*50+100, 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)
    final_money = models.FloatField()
    rmb = models.FloatField()

    round_number_4 = models.IntegerField()

    def round_number_4(self):
        self.round_number_4 = self.round_number - 4

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

    stock_1_bid_amount = models.IntegerField(label='你要买入多少资产A', min=0, initial=0)
    stock_2_bid_amount = models.IntegerField(label='你要买入多少资产B', min=0, initial=0)
    stock_3_bid_amount = models.IntegerField(label='你要买入多少资产C', min=0, initial=0)
    stock_4_bid_amount = models.IntegerField(label='你要买入多少资产D', min=0, initial=0)
    stock_5_bid_amount = models.IntegerField(label='你要买入多少资产E', min=0, initial=0)
    stock_6_bid_amount = models.IntegerField(label='你要买入多少资产F', min=0, initial=0)

    stock_1_sell_amount = models.IntegerField(label='你要卖出多少资产A', min=0, initial=0)
    stock_2_sell_amount = models.IntegerField(label='你要卖出多少资产B', min=0, initial=0)
    stock_3_sell_amount = models.IntegerField(label='你要卖出多少资产C', min=0, initial=0)
    stock_4_sell_amount = models.IntegerField(label='你要卖出多少资产D', min=0, initial=0)
    stock_5_sell_amount = models.IntegerField(label='你要卖出多少资产E', min=0, initial=0)
    stock_6_sell_amount = models.IntegerField(label='你要卖出多少资产F', 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_net_amount_str = models.StringField(initial='-')
    stock_2_net_amount_str = models.StringField(initial='-')
    stock_3_net_amount_str = models.StringField(initial='-')
    stock_4_net_amount_str = models.StringField(initial='-')
    stock_5_net_amount_str = models.StringField(initial='-')
    stock_6_net_amount_str = models.StringField(initial='-')

    # 股票预测
    Asset_A_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )
    Asset_B_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )
    Asset_C_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )
    Asset_D_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )
    Asset_E_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )
    Asset_F_forecast = models.IntegerField(
        choices=[
            [65, '优质资产 上涨概率65%'],
            [55, '次优资产 上涨概率55%'],
            [50, '普通资产 上涨概率50%'],
            [45, '次劣资产 上涨概率45%'],
            [35, '劣质资产 上涨概率35%'],
        ]
    )

    def update_stock_1_amount(self):
        if self.round_number == 1 or self.round_number == 5:
            self.stock_1_amount = 0
        else:
            self.stock_1_amount = self.in_round(self.round_number - 1).stock_1_amount

    def update_stock_2_amount(self):
        if self.round_number == 1 or self.round_number == 5:
            self.stock_2_amount = 0
        else:
            self.stock_2_amount = self.in_round(self.round_number - 1).stock_2_amount

    def update_stock_3_amount(self):
        if self.round_number == 1 or self.round_number == 5:
            self.stock_3_amount = 0
        else:
            self.stock_3_amount = self.in_round(self.round_number - 1).stock_3_amount

    def update_stock_4_amount(self):
        if self.round_number == 1 or self.round_number == 5:
            self.stock_4_amount = 0
        else:
            self.stock_4_amount = self.in_round(self.round_number - 1).stock_4_amount

    def update_stock_5_amount(self):
        if self.round_number == 1 or self.round_number == 5:
            self.stock_5_amount = 0
        else:
            self.stock_5_amount = self.in_round(self.round_number - 1).stock_5_amount

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

# 历史持有量
    def stock_1_net_amount_history(self):
        history = [g.stock_1_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history

    def stock_2_net_amount_history(self):
        history =  [g.stock_2_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history

    def stock_3_net_amount_history(self):
        history =  [g.stock_3_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history

    def stock_4_net_amount_history(self):
        history =  [g.stock_4_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history

    def stock_5_net_amount_history(self):
        history =  [g.stock_5_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history

    def stock_6_net_amount_history(self):
        history =  [g.stock_6_net_amount_str for g in self.in_all_rounds()]
        length_of_history = len(history)
        length_total = Constants.num_rounds
        addition = ['-' for i in range(length_total-length_of_history)] ## 用'-'补位
        all_round_history = history+addition
        dict_all_round_history = {}
        for i, item in enumerate(all_round_history):
            dict_all_round_history[i] = item
        return dict_all_round_history