tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] KeyError

Browse Source
This commit is contained in:
Al
2015-11-27 14:40:11 -05:00
parent a77bc03977
commit 3cd1fee89d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/geodata/osm/osm_address_training_data.py
View File

@@ -597,15 +597,17 @@ def build_address_format_training_data(admin_rtree, language_rtree, neighborhood
''' '''
if non_local_language or (AddressFormatter.CITY not in address_components and random.random() < 0.2): if non_local_language or (AddressFormatter.CITY not in address_components and random.random() < 0.2):
lang = non_local_language or language
quattroshapes_cities = quattroshapes_rtree.point_in_poly(latitude, longitude, return_all=True) quattroshapes_cities = quattroshapes_rtree.point_in_poly(latitude, longitude, return_all=True)
for result in quattroshapes_cities: for result in quattroshapes_cities:
if result.get(quattroshapes_rtree.LEVEL) == quattroshapes_rtree.LOCALITY and quattroshapes_rtree.GEONAMES_ID in result: if result.get(quattroshapes_rtree.LEVEL) == quattroshapes_rtree.LOCALITY and quattroshapes_rtree.GEONAMES_ID in result:
geonames_id = int(result[quattroshapes_rtree.GEONAMES_ID].split(',')[0]) geonames_id = int(result[quattroshapes_rtree.GEONAMES_ID].split(',')[0])
names = geonames.get_alternate_names(geonames_id) names = geonames.get_alternate_names(geonames_id)
if not names or language not in names:
if not names or lang not in names:
continue continue
city = names[non_local_language or language][0][0] city = names[lang][0][0]
if not city or not city.strip(): if not city or not city.strip():
continue continue
address_components[AddressFormatter.CITY] = city address_components[AddressFormatter.CITY] = city