#!/usr/bin/python3.7
# -*- coding: utf-8 -*-
# @Time : 2019/5/1 23:26
# @Email : jtyoui@qq.com
# @Software: PyCharm
from jtyoui.decorators import deprecationWarning
import jtyoui
import os
import json
import re
re_id_card = """
(?P<province>[^省]+省|.+自治区)
(?P<city>[^自治州]+自治州|[^市]+市|[^盟]+盟|[^地区]+地区|.+区划)
(?P<county>[^市]+市|[^县]+县|[^旗]+旗|[^区]+区)?
(?P<town>[^区]+区|[^镇]+镇)?
(?P<village>.*)
""".replace('\n', '')
_Address = {}
_TREE = None
[文档]def load_address_file(file_address_path):
"""加载地址文件数据
:param file_address_path: 加载地址文件的路径，没有地址文件默认自动下载。
:return: 地址文件数据，类型字典
"""
file_address_path = os.path.abspath(file_address_path)
if not os.path.exists(file_address_path + os.sep + 'rear.bz2'):
jtyoui.download_dev_tencent('rear.bz2', 'zhangwei0530', 'logo', file_address_path,
'52288E9AD139B0BAE97A55997B69A51F')
bz = jtyoui.unbz2_one(file_address_path + os.sep + 'rear.bz2', None)
address = json.loads(bz[512:-1134], encoding='utf8')
return address
[文档]@deprecationWarning
def find_address(name):
"""查询地址
该函数也废除，建议使用：finds_address函数
输入一个地名，查到这个名字的详细地址：比如输入：大连市、朝阳区、遵义县、卡比村等
:param name: 输入一个地址。
:return: 地址的信息
"""
assert True if name else False, '输入的字符串不能为空'
global _Address
if not _Address:
if 'nt' == os.name:
path = r'D://'
else:
path = r'./'
_Address = load_address_file(path)
address = []
for province in _Address:
city_s = _Address[province]
for city in city_s:
districts = city_s[city]
for district in districts:
towns = districts[district]
for town in towns:
villages = towns[town]
for village in villages:
if name in village:
addr = province + ' ' + city + ' ' + district + ' ' + town + ' ' + village
address.append(addr)
if name in town:
addr = province + ' ' + city + ' ' + district + ' ' + town
address.append(addr)
if name in district:
addr = province + ' ' + city + ' ' + district
address.append(addr)
if name in city:
addr = province + ' ' + city
address.append(addr)
if name in province:
addr = province
address.append(addr)
return address
[文档]def find_identity_card_address(card_addr):
"""识别身份证地址
输入一个地址，判断是否是身份证上的地址，返回格式包括：
一级地址：省|自治区
二级地址：自治州|市|盟|地区|区划
三级地址：市|县|旗|区
四级地址：区|镇
五级地址：其他
:param card_addr: 一串地址
:return: 返回元组，包括五级地址
"""
names = re.match(re_id_card, card_addr)
province = names.group('province')
city = names.group('city')
county = names.group('county')
town = names.group('town')
village = names.group('village')
return province, city, county, town, village
[文档]def finds_address(data, name: str):
"""查询地址
输入一个地名，查到这个名字的详细地址：比如输入：大连市、朝阳区、遵义县、卡比村等
:param data: 地址数据
:param name: 输入一个地址名
:return: 地址的信息
"""
global _TREE
if not _TREE:
_TREE = jtyoui.dict_create_tree(data)
return _TREE.search_tree_value(name)
[文档]def supplement_address(address_ls, address_tree):
"""补全地址
输入零碎的地址信息。补全地址，比如输入：['山西', '文水', '孝义村'],补全为：山西省-吕梁市-文水县-孝义镇-孝义村委会
>>> Tree = jtyoui.dict_create_tree(jtyoui.load_address_file('D://'))
>>> print(supplement_address(['山西', '文水', '孝义'], Tree))
:param address_ls: 补全的地址名字
:param address_tree: 地址树，默认方法：dict_create_tree(load_address_file('D://'))
:return: 返回列表，列表里面是一个元组，第一个参数是地址名字，第二个参数是地址的权重，权重越大越相似，排序越前。
"""
max_addr, ls = {}, []
for index in range(len(address_ls)):
address_tree.search_tree(address_ls[index], ls)
while ls:
tree = ls.pop()
for params in address_ls:
if params == address_ls[index]:
continue
path = tree.node_parent_value()
if params in path:
max_addr[path] = max_addr.get(path, 0) + 1
addr = jtyoui.remove_subset(list(max_addr.keys()))
return sorted(((a, max_addr[a]) for a in addr), key=lambda k: k[1], reverse=True)
if __name__ == '__main__':
import pprint
load = load_address_file('D://')
pprint.pprint(find_address('晋安')) # 已废弃，建议用：finds_address
print(find_identity_card_address('贵州省贵阳市南明区花果园延安南路28号'))
print('---------------------搜索树查找，第一次比较慢------------------------------------------')
for t in finds_address(load, '晋安'):
print(t)
for t in finds_address(load, '遵义县'):
print(t)
Tree = jtyoui.dict_create_tree(load, jtyoui.Tree(value='Root'))
print(supplement_address(['山西', '文水', '孝义'], Tree))