This commit is contained in:
2025-09-10 10:13:38 +08:00
parent aced5dd6be
commit 6a32f010d4

View File

@@ -78,15 +78,19 @@ def load_yunnan_area_data():
def query_area_info(query_str):
"""
查询行政区划信息(优化版)
查询行政区划信息
:param query_str: 简称或全称
:return: 包含full_name和area_code的字典未找到返回None
"""
# 修复:显式加载数据并初始化映射字典
abbrev_map, fullname_map = load_jiancheng_data()
yunnan_area_map = load_yunnan_area_data()
# 1. 先查简称映射表
if query_str in abbrev_map:
return abbrev_map[query_str]
# 2. 直接查询完整行政区划数据(新增)
# 2. 再查完整行政区划数据
if query_str in yunnan_area_map:
return yunnan_area_map[query_str]
@@ -94,7 +98,7 @@ def query_area_info(query_str):
if query_str in fullname_map:
return fullname_map[query_str]
# 4. 尝试模糊匹配(新增)
# 4. 尝试模糊匹配
for name, info in yunnan_area_map.items():
if query_str in name or name in query_str:
return info