project/xhs/area_info/city_name_to_id.py

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

def get_city_2(city_str):
    """兼容旧版本的调用"""
    return get_city(city_str)

if __name__ == "__main__":
    city_id, province_id, city, province = get_city("玉林")
    print(city_id, province_id, city, province)