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

[geonames] adding query method to GeoNamesDB

Browse Source
This commit is contained in:
Al
2016-09-12 16:42:42 -04:00
parent e8408d39fd
commit 62a5f6002d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/geodata/geonames/db.py
View File

@@ -17,8 +17,11 @@ class GeoNamesDB(object):
def __init__(self, filename): def __init__(self, filename):
self.db = sqlite3.connect(filename) self.db = sqlite3.connect(filename)
def query(self, query, *params):
return self.db.execute(self.names_query, params)
def get_alternate_names(self, geonames_id): def get_alternate_names(self, geonames_id):
cursor = self.db.execute(self.names_query, [geonames_id]) cursor = self.query(self.names_query, geonames_id)
language_names = defaultdict(list) language_names = defaultdict(list)
for language, name, is_preferred, is_short in cursor: for language, name, is_preferred, is_short in cursor:
language_names[language].append((name, language_names[language].append((name,