diff --git a/plugins/yiqing/data_source.py b/plugins/yiqing/data_source.py index 9f0b6ef2..5248de41 100644 --- a/plugins/yiqing/data_source.py +++ b/plugins/yiqing/data_source.py @@ -25,17 +25,17 @@ async def get_yiqing_data(area: str): if area == '中国': province = area province_type = "" - elif area[-1] == '市' and area not in data.keys(): + elif area in data.keys() and area[-1] != '市': + province = area if area[-1] != '省' else area[:-1] + if len(data[province]) == 1: + province_type = "市" + city = "" + else: area = area[:-1] if area[-1] == '市' else area for p in data.keys(): if area in data[p]: province = p city = area - elif area in data.keys(): - province = area if area[-1] != '省' else area[:-1] - if len(data[province]) == 1: - province_type = "市" - city = "" if not province and not city: return "小真寻只支持国内的疫情查询喔..." async with aiohttp.ClientSession(headers=get_user_agent()) as session: