# 导入json
import json
# 导入request
import requests

# 创建折线图
from pyecharts.charts import Line
from pyecharts import options as opts


def creatAnalysisMap():
    # 创建折线图
    # line = Line("中国疫情发展趋势图 "+time_str, width=1200, height=600)
    # 请求中国疫情数据链接
    baseUrl = "https://view.inews.qq.com/g2/getOnsInfo?name=disease_other&callback=jQuery34105560710307785617_1592363143544&_=1592363143545"
    # 当前请求链接返回的数据
    chinaData = requests.get(baseUrl).text.replace('"{', '{').replace('}"}', '}}').replace('\\', '')
    # 处理数据
    chinaData = chinaData[chinaData.index("{"):-1]
    # 将当前的json字符串转换成python的字典
    baseData = json.loads(chinaData)
    # 保存疫情属性列表
    attrList = ["累计确诊", "现有疑似", "现有确诊", "现有重症", "境外输入"]
    # 日期列表
    dateList = []
    # 累计确诊列表
    confirmList = []
    # 现有疑似列表
    suspectList = []
    # 现有确诊列表
    nowConfirmList = []
    # 现有重症列表
    nowSevereList = []
    # 境外输入列表
    importedCaseList = []
    # 解析数据
    for item in baseData["data"]["chinaDayList"]:
        # 追加日期
        dateList.append(item["date"])
        # 追加累计确诊列表
        confirmList.append(item["confirm"])
        # 追加现有疑似列表
        suspectList.append(item["suspect"])
        # 追加现有确诊列表
        nowConfirmList.append(item["nowConfirm"])
        # 追加现有重症列表
        nowSevereList.append(item["nowSevere"])
        # 追加境外输入列表
        importedCaseList.append(item["importedCase"])
    # 添加各个属性直线图
    line = (
        Line()
        .add_xaxis(xaxis_data=dateList)
        .add_yaxis('累计确诊', y_axis=confirmList, is_symbol_show=False)
        .add_yaxis('现有疑似', y_axis=suspectList, is_symbol_show=False)
        .add_yaxis('现有确诊', y_axis=nowConfirmList, is_symbol_show=False)
        .add_yaxis('现有重症', y_axis=nowSevereList, is_symbol_show=False)
        .add_yaxis('境外输入', y_axis=importedCaseList, is_symbol_show=False)
        .set_global_opts(title_opts=opts.TitleOpts(title='疫情折线图'),
                         xaxis_opts=opts.AxisOpts(name='时间'),
                         yaxis_opts=opts.AxisOpts(name='人数'))
    )
    line.render(path="中国疫情发展趋势图.html")

creatAnalysisMap()