from db import *
from blue_forecast import forecast_blue
from red_forecast import red_forecast
import json


def general_red_index(_real_red, _forecast_data):
    index_list = []
    data_list = list(_forecast_data.keys())
    for _red in data_list:
        if _red in _real_red:
            index_list.append(data_list.index(_red) + 1)
    index_group = [0, 0, 0]
    for index in index_list:
        if 1 <= index <= 11:
            index_group[0] += 1
        elif 11 < index <= 22:
            index_group[1] += 1
        elif 22 < index <= 33:
            index_group[2] += 1
        else:
            print(f'索引值存在错误{index_list}')
    return index_list, index_group


def general_blue_index(_real_blue, _forecast_data):
    return list(_forecast_data.keys()).index(_real_blue) + 1


def insert_index_group(_date_id, _type, _index_group):
    if _date_id not in db_index_get_last_date_id(_type):
        _data = {
            'dateId': _date_id,
            'type': _type,
            'index_group': _index_group
        }
        result = db_index_insert_index_group(_data)
        if result > 0:
            print(f'插入index_group成功 {_index_group}')
        else:
            print('插入index_group失败')


def update_real_data():
    new_history_date_id, old_history_date_id = db_history_get_last_open_id()
    blue_date_id = db_blue_get_last_one_open_id()
    red_date_id = db_red_get_last_one_open_id()
    all_data = db_history_get_all_data()
    # 如果 历史里面有新数据： 需要计算真实的index，然后预测新数据
    # 如果 历史记录中是老数据： 需要预测新数据
    if red_date_id == old_history_date_id:
        # 插入 RedForecastAll real_red, real_red_index, real_red_index_group
        real_red = json.loads(all_data[-1]['red'])
        forecast_red_data = db_red_get_data_by_date_id(red_date_id)
        for data in forecast_red_data:
            _id = data['id']
            history_index_list, history_index_group = general_red_index(real_red, json.loads(data['history']))
            last_index_list, last_index_group = general_red_index(real_red, json.loads(data['last']))
            update_data = {
                'real_red': json.dumps(real_red),
                'real_red_index_history': json.dumps(history_index_list),
                'real_red_index_group_history': json.dumps(history_index_group),
                'real_red_index_last': json.dumps(last_index_list),
                'real_red_index_group_last': json.dumps(last_index_group)
            }
            db_red_update(_id, red_date_id, update_data)
            # 插入 IndexGroup index_group
            insert_index_group(
                {'dateId': red_date_id, 'type': 'history', 'index_group': json.dumps(history_index_group)})
            insert_index_group({'dateId': red_date_id, 'type': 'last', 'index_group': json.dumps(last_index_group)})

    if blue_date_id['five'] == old_history_date_id:
        # 插入 real_blue, prv_index, post_index, history_index
        real_blue = all_data[-1]['blue']
        forecast_blue_data = db_blue_get_data_by_date_id(blue_date_id['five'], 5)
        for data in forecast_blue_data:
            _id = data['id']
            prv_index = general_blue_index(real_blue, json.loads(data['prv']))
            post_index = general_blue_index(real_blue, json.loads(data['post']))
            history_index = general_blue_index(real_blue, json.loads(data['history']))
            update_data = {
                'real_blue': real_blue,
                'prv_index': prv_index,
                'post_index': post_index,
                'history_index': history_index
            }
            db_blue_update(blue_date_id['five'], update_data, _id, 5)

    if blue_date_id['three'] == old_history_date_id:
        real_blue = all_data[-1]['blue']
        forecast_blue_data = db_blue_get_data_by_date_id(blue_date_id['three'], 3)
        for data in forecast_blue_data:
            _id = data['id']
            prv_index = general_blue_index(real_blue, json.loads(data['prv']))
            post_index = general_blue_index(real_blue, json.loads(data['post']))
            history_index = general_blue_index(real_blue, json.loads(data['history']))
            update_data = {
                'real_blue': real_blue,
                'prv_index': prv_index,
                'post_index': post_index,
                'history_index': history_index
            }
            db_blue_update(blue_date_id['three'], update_data, _id, 3)


# 插入预测数据
def general_forecast_data():
    forecast_blue()
    red_forecast()


if __name__ == "__main__":
    # update_real_data()
    # general_forecast_data()
    pass