# 数据处理
import json

from pyecharts.charts import Line
from pyecharts.options import TitleOpts, LabelOpts

f_us = open("D:/program/美国.txt", 'r', encoding="UTF-8")
f_jp = open("D:/program/日本.txt", 'r', encoding="UTF-8")
f_yd = open("D:/program/印度.txt", 'r', encoding="UTF-8")
us_data = f_us.read()
jp_data = f_jp.read()
yd_data = f_yd.read()
# 去掉不符合JSON的开头
us_data = us_data.replace("jsonp_1629344292311_69436(", "")
jp_data = jp_data.replace("jsonp_1629350871167_29498(", "")
yd_data = yd_data.replace("jsonp_1629350745930_63180(", "")
# 去掉不符合JSON的结尾
us_data = us_data[:-2]  # 切片
jp_data = jp_data[:-2]
yd_data = yd_data[:-2]
# JSON转Python字典
us_dict = json.loads(us_data)
jp_dict = json.loads(jp_data)
yd_dict = json.loads(yd_data)
# print(type(us_dict))
# print(us_dict)
# print(jp_dict)
# print(yd_dict)
# 获取trend key
us_trend_data = us_dict['data'][0]['trend']
jp_trend_data = jp_dict['data'][0]['trend']
yd_trend_data = yd_dict['data'][0]['trend']
# print(us_trend_data)
# 获取日期数据，用于x轴，取2020年（到315下标结束）
us_x_data = us_trend_data['updateDate'][:314]
jp_x_data = jp_trend_data['updateDate'][:315]
yd_x_data = yd_trend_data['updateDate'][:269]
# us_x_data = us_x_data[:314]
# jp_x_data = jp_x_data[:315]
# yd_x_data = yd_x_data[:269]
# print(us_x_data)
# print(jp_x_data)
# print(yd_x_data)

# 获取确诊数据，用于y轴，取2020年（到315结束）
us_y_data = us_trend_data['list'][0]['data'][:314]
jp_y_data = jp_trend_data['list'][0]['data'][:315]
yd_y_data = yd_trend_data['list'][0]['data'][:269]
# us_y_data = us_y_data[:314]
# jp_y_data = jp_y_data[:315]
# yd_y_data = yd_y_data[:269]

# 生成图表
line = Line()  # 构建折线图对象
# 添加x轴数据
line.add_xaxis(us_x_data)
# line.add_xaxis(jp_x_data)
# line.add_xaxis(yd_x_data)
# 添加y轴数据
line.add_yaxis("美国确诊人口", us_y_data, label_opts=LabelOpts(is_show=False))  # 不显示数据（系统配置）
line.add_yaxis("日本确诊人口", jp_y_data)
line.add_yaxis("印度确诊人口", yd_y_data)

# 设置全局选项
line.set_global_opts(
    # 标题
    title_opts=TitleOpts(title="2020年美印日三国确诊人数对比折线图", pos_left="center", pos_bottom="1%")
)

# 调用render方法生成图标
line.render("折线图.html")

# 关闭文件
f_yd.close()
f_us.close()
f_jp.close()