mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
80 lines
2.3 KiB
Python
Executable File
80 lines
2.3 KiB
Python
Executable File
from utils.message_builder import image
|
|
from configs.path_config import TEXT_PATH
|
|
from configs.config import NICKNAME
|
|
from typing import List
|
|
from nonebot import Driver
|
|
from utils.http_utils import AsyncHttpx
|
|
import ujson as json
|
|
import nonebot
|
|
|
|
driver: Driver = nonebot.get_driver()
|
|
|
|
china_city = TEXT_PATH / "china_city.json"
|
|
|
|
data = {}
|
|
|
|
|
|
async def get_weather_of_city(city: str) -> str:
|
|
"""
|
|
获取城市天气数据
|
|
:param city: 城市
|
|
"""
|
|
code = _check_exists_city(city)
|
|
if code == 999:
|
|
return "不要查一个省份的天气啊，很累人的！"
|
|
elif code == 998:
|
|
return f"{NICKNAME}没查到!!试试查火星的天气？"
|
|
else:
|
|
data_json = (
|
|
await AsyncHttpx.get(
|
|
f"https://v0.yiketianqi.com/api?unescape=1&version=v91&appid=43656176&appsecret=I42og6Lm&ext=&cityid=&city={city[:-1]}"
|
|
)
|
|
).json()
|
|
if wh := data_json.get('data'):
|
|
w_type = wh[0]["wea_day"]
|
|
w_max = wh[0]["tem1"]
|
|
w_min = wh[0]["tem2"]
|
|
fengli = wh[0]["win_speed"]
|
|
ganmao = wh[0]["narrative"]
|
|
fengxiang = ','.join(wh[0].get('win', []))
|
|
repass = f"{city}的天气是 {w_type} 天\n最高温度: {w_max}\n最低温度: {w_min}\n风力: {fengli} {fengxiang}\n{ganmao}"
|
|
return repass
|
|
else:
|
|
return data_json.get("errmsg") or "好像出错了？再试试？"
|
|
|
|
|
|
def _check_exists_city(city: str) -> int:
|
|
"""
|
|
检测城市是否存在合法
|
|
:param city: 城市名称
|
|
"""
|
|
global data
|
|
city = city if city[-1] != "市" else city[:-1]
|
|
for province in data.keys():
|
|
for city_ in data[province]:
|
|
if city_ == city:
|
|
return 200
|
|
for province in data.keys():
|
|
if city == province:
|
|
return 999
|
|
return 998
|
|
|
|
|
|
def get_city_list() -> List[str]:
|
|
"""
|
|
获取城市列表
|
|
"""
|
|
global data
|
|
if not data:
|
|
try:
|
|
with open(china_city, "r", encoding="utf8") as f:
|
|
data = json.load(f)
|
|
except FileNotFoundError:
|
|
data = {}
|
|
city_list = []
|
|
for p in data.keys():
|
|
for c in data[p]:
|
|
city_list.append(c)
|
|
city_list.append(p)
|
|
return city_list
|