| 123456789101112131415161718192021222324252627282930 |
- import os
- from pathlib import Path
- from area.area import AreaService
- # 导出供外部使用
- _DEFAULT_PATH = Path(__file__).with_name("city.json")
- _area_service = AreaService(str(_DEFAULT_PATH))
- def get_city(city_str):
- """可以通过市获取省份（重构版，底层使用 Trie 树解析）"""
- city_id = 0
- province_id = 0
- city = ""
- province = ""
-
- if not city_str:
- return city_id, province_id, city, province
- match = _area_service.search_area(city_str)
- if match:
- city_id = match.city_id
- province_id = match.province_id
- city = match.city
- province = match.province
- return city_id, province_id, city, province
- if __name__ == "__main__":
- city_id, province_id, city, province = get_city("玉林")
- print(city_id, province_id, city, province)
|
