mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
65 lines
2.5 KiB
Python
Executable File
65 lines
2.5 KiB
Python
Executable File
from nonebot import on_command
|
||
from .data_source import get_yiqing_data, get_city_and_province_list
|
||
from services.log import logger
|
||
from nonebot.adapters.onebot.v11 import MessageEvent, GroupMessageEvent, Message
|
||
from nonebot.params import CommandArg
|
||
from configs.config import NICKNAME
|
||
from .other_than import get_other_data
|
||
|
||
__zx_plugin_name__ = "疫情查询"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
全国疫情查询
|
||
指令：
|
||
疫情 中国/美国/英国...
|
||
疫情 [省份/城市]
|
||
* 当省份与城市重名时，可在后添加 "市" 或 "省" *
|
||
示例：疫情 吉林 <- [省]
|
||
示例：疫情 吉林市 <- [市]
|
||
""".strip()
|
||
__plugin_des__ = "实时疫情数据查询"
|
||
__plugin_cmd__ = ["疫情 [省份/城市]", "疫情 中国"]
|
||
__plugin_type__ = ("一些工具",)
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier & yzyyz1387"
|
||
__plugin_settings__ = {
|
||
"level": 5,
|
||
"default_status": True,
|
||
"limit_superuser": False,
|
||
"cmd": ["查询疫情", "疫情", "疫情查询"],
|
||
}
|
||
|
||
|
||
yiqing = on_command("疫情", aliases={"查询疫情", "疫情查询"}, priority=5, block=True)
|
||
|
||
|
||
@yiqing.handle()
|
||
async def _(event: MessageEvent, arg: Message = CommandArg()):
|
||
msg = arg.extract_plain_text().strip()
|
||
city_and_province_list = get_city_and_province_list()
|
||
if msg:
|
||
if msg in city_and_province_list or msg[:-1] in city_and_province_list:
|
||
result = await get_yiqing_data(msg)
|
||
if result:
|
||
await yiqing.send(result)
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 查询疫情: {msg}"
|
||
)
|
||
else:
|
||
await yiqing.send("查询失败!!!!", at_sender=True)
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 查询疫情失败"
|
||
)
|
||
else:
|
||
rely = await get_other_data(msg)
|
||
if rely:
|
||
await yiqing.send(rely)
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 查询疫情成功"
|
||
)
|
||
else:
|
||
await yiqing.send(f"{NICKNAME}没有查到{msg}的疫情查询...")
|