import requests
import xlwt
import json

# 创建工作簿对象
workbook = xlwt.Workbook(encoding='utf-8-sig')
# 创建工作表对象
worksheet = workbook.add_sheet('天气数据')

# 设置单元格样式
style = xlwt.easyxf('font: bold True;')

# 定义列标题
headers = ['date', 'city', 'country', 'weather', 'max_temperature', 'min_temperature', 'Wind', 'Wind_level']
# 写入列标题
for i, header in enumerate(headers):
    worksheet.write(0, i, header, style)

# 请求天气数据
url = 'https://weather.cma.cn/api/map/weather/1?t=1675244104697'
res = requests.get(url)
text = res.json()

# 遍历城市数据并写入Excel文件
for i, city in enumerate(text['data']['city']):
    # 写入数据到Excel文件
    row = [
        text['data']['date'],
        city[1],  # 城市
        city[2],  # 国家
        city[7],  # 天气
        city[6],  # 最高温
        city[11],  # 最低温
        city[9],  # 风向
        city[10]  # 风级
    ]
    for j, value in enumerate(row):
        worksheet.write(i+1, j, value)
    print(f'[*] {city[1]}的天气情况已下载!')

# 保存Excel文件
workbook.save('中国气象局天气.xls')