You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
4.1 KiB
100 lines
4.1 KiB
from db import * |
|
import json |
|
|
|
|
|
def general_red_index(_real_red, _forecast_data): |
|
index_list = [] |
|
for _red in _forecast_data.keys(): |
|
if _red in _real_red: |
|
index_list.append(_forecast_data.keys().index(_red) + 1) |
|
index_group = [0, 0, 0] |
|
for index in index_group: |
|
if 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 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': real_red, |
|
'real_red_index_history': history_index_list, |
|
'real_red_index_group_history': history_index_group, |
|
'real_red_index_last': last_index_list, |
|
'real_red_index_group_last': last_index_group |
|
} |
|
db_red_update(_id, red_date_id, update_data) |
|
# 插入 IndexGroup index_group |
|
db_index_insert_index_group({'type': 'history', 'index_group': json.dumps(history_index_group)}) |
|
db_index_insert_index_group({'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: |
|
print(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'])) |
|
print(prv_index) |
|
print(post_index) |
|
print(history_index) |
|
update_data = { |
|
'real_blue': real_blue, |
|
'prv_index': prv_index, |
|
'post_index': post_index, |
|
'history_index': history_index |
|
} |
|
print(update_data) |
|
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(): |
|
pass |
|
|
|
|
|
if __name__ == "__main__": |
|
update_real_data() |
|
general_forecast_data()
|
|
|