nominatim/tools/country_info.py

   1 """
   2 Functions for importing and managing static country information.
   3 """
   4 from nominatim.db import utils as db_utils
   5 from nominatim.db.connection import connect
   6
   7 def setup_country_tables(dsn, sql_dir, ignore_partitions=False):
   8     """ Create and populate the tables with basic static data that provides
   9         the background for geocoding. Data is assumed to not yet exist.
  10     """
  11     db_utils.execute_file(dsn, sql_dir / 'country_name.sql')
  12     db_utils.execute_file(dsn, sql_dir / 'country_osm_grid.sql.gz')
  13
  14     if ignore_partitions:
  15         with connect(dsn) as conn:
  16             with conn.cursor() as cur:
  17                 cur.execute('UPDATE country_name SET partition = 0')
  18             conn.commit()