import json
from prettytable import PrettyTable
import requests
import pandas as pd
def getData_zt2020():
url = 'https://news.qq.com/zt2020/page/feiyan.htm'
headers = {
'user-agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A372 Safari/604.1'
}
r = requests.get(url, headers)
if r.status_code == 200:
return r.text
else:
print("数据拉取失败")
return ""
def getData_g2():
url = 'https://view.inews.qq.com/g2/getOnsInfo?name=disease_h5'
headers = {
'user-agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A372 Safari/604.1'
}
req = requests.get(url, headers)
if req.status_code == 200:
data = json.loads(json.loads(req.text)['data'])
return data
else:
print("数据拉取失败")
return ""
def format_list_prettytable(title, province_list):
table = PrettyTable(title)
for province in province_list:
table.add_row(province)
table.border = True
return table
def get_china(data):
"""
lastUpdateTime ：最后更新时间
chinaTotal ：累计值
chinaAdd ：每日新增 和 chinaTotal字段一致
confirm 确诊
heal 治愈
dead 死亡
nowConfirm 现有确诊
suspect 疑似
nowSevere 重症
chinaDayList不提供了,搞得我没法做趋势图
:param data:
:return:
"""
# 新增
china_add = data['chinaAdd']
add_confirm = china_add['confirm'] # 新增确诊
add_dead = china_add["dead"] # 新增死亡
add_now_confirm = china_add["nowConfirm"] # 新增现有确诊
add_suspect = china_add["suspect"] # 疑似新增
add_now_severe = china_add["nowSevere"] # 新增重症
add_heal = china_add["heal"] # 治愈新增
add_noInfect = china_add["noInfectH5"] # 无症状新增
add_importedCase = china_add["importedCase"] # 境外输入新增
add_localConfirm = china_add["localConfirmH5"] # 本土新增确诊
#累计
china_total = data['chinaTotal'] # 累计
total_confirm = china_total['confirm'] # 累计确诊
total_dead = china_total['dead'] # 累计死亡
total_heal = china_total['heal'] # 累计治愈
total_suspect = china_total['suspect'] # 现有疑似
total_nowSevere = china_total['nowSevere'] # 现有重症
total_noInfect = china_total['noInfect'] # 现有无症状
total_importedCase = china_total['importedCase'] # 现有境外输入
total_localConfirm = china_total['localConfirmH5'] # 本土现有
title = ["add_localConfirm", "add_noInfect", "add_suspect", "add_confirm", "total_localConfirm", "total_noInfect", "total_suspect"]
china_list = []
china_list.append([add_localConfirm, add_noInfect, add_suspect, add_confirm, total_localConfirm, total_noInfect, total_suspect])
china_table = format_list_prettytable(title, china_list)
return china_table
def get_all_province(data):
area_data = data["areaTree"][0] # 中国数据
chil_data = area_data['children']
province_list = []
for province in chil_data:
name = province["name"]
today_confirm = province["today"]["confirm"] # 今日新增确诊病例数
today_wzz = province["today"]["wzz_add"] # 今日新增无症状
now_confirm = province["total"]["nowConfirm"] # 现有确诊病例数
total_confirm = province["total"]["confirm"] # 累计确诊病例数
total_wzz = province["total"]["wzz"] # 累计确诊病例数
total_confirm = province["total"]["confirm"] # 累计确诊病例数
total_heal = province["total"]["heal"] # 累计治愈
province_list.append([name, today_confirm, today_wzz, now_confirm, total_confirm, total_heal])
title = ["name", "today_confirm", "today_wzz", "now_confirm", "total_confirm", "total_heal"]
province_table = format_list_prettytable(title, province_list)
return province_table
def parse_jiangsu_province(data, key_province):
area_data = json.loads(data)["areaTree"]
data = area_data[0]['children']
city_list = []
for province in data:
name = province['name']
if name == key_province:
children_list = province['children']
for children in children_list:
city = children["name"]
today_new = children["today"]["confirm"]
now_confirm = children["total"]["nowConfirm"]
total_confirm = children["total"]["confirm"]
today_heal = children['total']['heal']
city_list.append([city, today_new, now_confirm, total_confirm, today_heal])
return city_list
def data_write():
pass
def data_plot():
pass
if __name__ == '__main__':
data = getData_g2()
china_data = get_china(data)
print(china_data)
province_data = get_all_province(data)
print(province_data)
