Merge remote-tracking branch 'upstream/master'

[nominatim.git] / nominatim / tools / database_import.py

diff --git a/nominatim/tools/database_import.py b/nominatim/tools/database_import.py
index 9d4cad1693bbc6c089ac930c57add5f3493dc5ce..433cd8afaca30372ab58698821fee06092748b1a 100644 (file)
--- a/nominatim/tools/database_import.py
+++ b/nominatim/tools/database_import.py
@@ -312,32 +312,29 @@ def create_country_names(conn, config):
     """
 
     with conn.cursor() as cur:
-        cur.execute("""select
-                       getorcreate_country(make_standard_name('uk')
-                       , 'gb')""")
-        cur.execute("""select getorcreate_country(make_standard_name('united states'), 'us')""")
-        cur.execute("""select count(*) from
-                       (select getorcreate_country(make_standard_name(country_code),
-                       country_code) from country_name where country_code is not null) as x""")
-        cur.execute("""select count(*) from
-                        (select getorcreate_country(make_standard_name(name->'name'),
-                        country_code) from country_name where name ? 'name') as x""")
-        sql_statement = """select count(*) from (select getorcreate_country(make_standard_name(v)
-                           , country_code) from (select country_code, skeys(name)
-                           as k, svals(name) as v from country_name) x where k """
+        cur.execute("""SELECT getorcreate_country(make_standard_name('uk'), 'gb')""")
+        cur.execute("""SELECT getorcreate_country(make_standard_name('united states'), 'us')""")
+        cur.execute("""SELECT COUNT(*) FROM
+                       (SELECT getorcreate_country(make_standard_name(country_code),
+                       country_code) FROM country_name WHERE country_code is not null) AS x""")
+        cur.execute("""SELECT COUNT(*) FROM
+                       (SELECT getorcreate_country(make_standard_name(name->'name'), country_code) 
+                       FROM country_name WHERE name ? 'name') AS x""")
+        sql_statement = """SELECT COUNT(*) FROM (SELECT getorcreate_country(make_standard_name(v),
+                           country_code) FROM (SELECT country_code, skeys(name)
+                           AS k, svals(name) AS v FROM country_name) x WHERE k"""
 
         languages = config.LANGUAGES
 
         if languages:
-            sql_statement += 'in '
-            delim = '('
+            sql_statement = "{} IN (".format(sql_statement)
+            delim = ''
             for language in languages.split(','):
-                sql_statement += delim + "'name:" + language + "'"
-                delim = ','
-
-            sql_statement += ')'
+                sql_statement = "{}{}'name:{}'".format(sql_statement, delim, language)
+                delim = ', '
+            sql_statement = '{})'.format(sql_statement)
         else:
-            sql_statement += "like 'name:%'"
-        sql_statement += ') v'
+            sql_statement = "{} LIKE 'name:%'".format(sql_statement)
+        sql_statement = "{}) v".format(sql_statement)
         cur.execute(sql_statement)
     conn.commit()