diff options
Diffstat (limited to 'ATRIbot/plugins/covid19.py')
| -rw-r--r-- | ATRIbot/plugins/covid19.py | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/ATRIbot/plugins/covid19.py b/ATRIbot/plugins/covid19.py new file mode 100644 index 0000000..decb7c4 --- /dev/null +++ b/ATRIbot/plugins/covid19.py @@ -0,0 +1,78 @@ +# -*- coding:utf-8 -*- +import json +import requests +from nonebot import on_command, CommandSession +from AyaBot.plugins.module import google_translate + + +# 国外版 数据更新非常及时 +# 必须挂木弟子 +# 成功把之前的屎优化了！OHHHHHHHHHHHH +# 如需使用国内版，请查看下面的链接，下载后与这个文件替换 +# https://github.com/Kyomotoi/covid19/tree/master/Domestic + + + +url = "https://covid-19-coronavirus-statistics.p.rapidapi.com/v1/total" + + +LIST = """截至: {lastChecked} +国家: {location} +已治愈: {recovered} +已经死亡: {deaths} +总感染数: {confirmed} +现存感染人数: {nowConfirmed} +最后检查时间: {lastReported}""" + + +@on_command('covid19', aliases=['疫情', '疫情查询', '疫情情况'], only_to_me=False) +async def covid19(session: CommandSession): + country = session.get('country', prompt='请键入需要查询的国家(例:中国)') + if country == '美国': + pass + else: + re_msg = google_translate.translate(country[:4999], to='en', source='zh-CN') + # if re_msg[0]!='': + # await session.send(re_msg[0]) + await session.send('开始搜寻...\n如返回Global则为国家名有问题') + try: + querystring = {"country":"cy"} + if country == '美国': + re_msg = 'US' + querystring["country"] = re_msg + else: + querystring["country"] = re_msg[0] + print(querystring) + headers = { + 'x-rapidapi-host': "covid-19-coronavirus-statistics.p.rapidapi.com", + 'x-rapidapi-key': "a852be0d03msh8bd4299fe71bfeep100861jsn185ea925449c" + } + response = requests.request("GET", url, headers=headers, params=querystring) + # print(response.text) + html = response.text + c19 = json.loads(html) + await session.send(LIST.format( + lastChecked=c19["data"]["lastReported"], + location=c19["data"]["location"], + recovered=c19["data"]["recovered"], + deaths=c19["data"]["deaths"], + confirmed=c19["data"]["confirmed"], + nowConfirmed=c19["data"]["confirmed"] - c19["data"]["deaths"] - c19["data"]["recovered"], + lastReported=c19["data"]["lastChecked"], + ) + ) + except: + print('error') + await session.send('搜索出问题了呢，重新试试?') + [email protected]_parser +async def _(session: CommandSession): + stripped_arg = session.current_arg_text.strip() + if session.is_first_run: + if stripped_arg: + session.state['country'] = stripped_arg + return + + if not stripped_arg: + session.pause('要查询的国家不能为空，请重新输入') + session.state[session.current_key] = stripped_arg |