From 62a5f6002da2e6e6e3e937b0d206f2398f602f9b Mon Sep 17 00:00:00 2001 From: Al Date: Mon, 12 Sep 2016 16:42:42 -0400 Subject: [PATCH] [geonames] adding query method to GeoNamesDB --- scripts/geodata/geonames/db.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/geodata/geonames/db.py b/scripts/geodata/geonames/db.py index e83c6de3..36bb7b0c 100644 --- a/scripts/geodata/geonames/db.py +++ b/scripts/geodata/geonames/db.py @@ -17,8 +17,11 @@ class GeoNamesDB(object): def __init__(self, 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): - cursor = self.db.execute(self.names_query, [geonames_id]) + cursor = self.query(self.names_query, geonames_id) language_names = defaultdict(list) for language, name, is_preferred, is_short in cursor: language_names[language].append((name,