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

# 分别打开三哥国家文件
import json

f_us = open("D:/美国.txt", "r", encoding="UTF-8")
f_jp = open("D:/日本.txt", "r", encoding="UTF-8")
f_in = open("D:/印度.txt", "r", encoding="UTF-8")

# 去掉不属于于文件的开头结尾
us_data = f_us.read()  # 美国
us_data = us_data.replace("jsonp_1629344292311_69436(", "")
us_data = us_data[:-2]

jp_data = f_jp.read()  # 日本
jp_data = jp_data.replace("jsonp_1629350871167_29498(", "")
jp_data = jp_data[:-2]

in_data = f_in.read()  # 印度
in_data = in_data.replace("jsonp_1629350745930_63180(", "")
in_data = in_data[:-2]

# 将json转换为python字典
us_dict = json.loads(us_data)
jp_dict = json.loads(jp_data)
in_dict = json.loads(in_data)

# 获取trend key
us_trend_data = us_dict["data"][0]["trend"]
jp_trend_data = jp_dict["data"][0]["trend"]
in_trend_data = in_dict["data"][0]["trend"]

# 获取日期数据,用于x轴，取2020年结束，
us_x_data = us_trend_data["updateDate"][:314]
jp_x_data = jp_trend_data["updateDate"][:314]
in_x_data = in_trend_data["updateDate"][:314]

# 获取确认数据，用于y轴
us_y_data = us_trend_data["list"][0]["data"][:314]
jp_y_data = jp_trend_data["list"][0]["data"][:314]
in_y_data = in_trend_data["list"][0]["data"][:314]

# 生产图表
Line = Line()  # 构建折线图对象
Line.add_xaxis(us_x_data)  # x轴是公用的，所以使用一个国家的数据就可以
Line.add_yaxis("美国确诊人数", us_y_data, label_opts=LabelOpts(is_show=False))  # label = 图标数字是否显示
Line.add_yaxis("日本确诊人数", jp_y_data, label_opts=LabelOpts(is_show=False))
Line.add_yaxis("印度确诊人数", in_y_data, label_opts=LabelOpts(is_show=False))

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

)
# 调用render方法，生成图表
Line.render()
f_us.close()
f_in.close()
f_jp.close()