X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/b7c83d35808c1706bc6e92d21c74765e23103446..8c54f9d7044c7f354fcad43de033b4a1cfc1ac56:/nominatim/tools/database_import.py?ds=sidebyside diff --git a/nominatim/tools/database_import.py b/nominatim/tools/database_import.py index 1f39f381..de7e6a4a 100644 --- a/nominatim/tools/database_import.py +++ b/nominatim/tools/database_import.py @@ -24,8 +24,7 @@ from nominatim.db.sql_preprocessor import SQLPreprocessor from nominatim.tools.exec_utils import run_osm2pgsql from nominatim.errors import UsageError from nominatim.version import POSTGRESQL_REQUIRED_VERSION, \ - POSTGIS_REQUIRED_VERSION, \ - HSTORE_REQUIRED_VERSION + POSTGIS_REQUIRED_VERSION LOG = logging.getLogger() @@ -40,7 +39,14 @@ def _require_version(module: str, actual: Tuple[int, int], expected: Tuple[int, raise UsageError(f'{module} is too old.') -def check_existing_database_plugins(dsn: str): +def _require_loaded(extension_name: str, conn: Connection) -> None: + """ Check that the given extension is loaded. """ + if not conn.extension_loaded(extension_name): + LOG.fatal('Required module %s is not loaded.', extension_name) + raise UsageError(f'{extension_name} is not loaded.') + + +def check_existing_database_plugins(dsn: str) -> None: """ Check that the database has the required plugins installed.""" with connect(dsn) as conn: _require_version('PostgreSQL server', @@ -49,9 +55,7 @@ def check_existing_database_plugins(dsn: str): _require_version('PostGIS', conn.postgis_version_tuple(), POSTGIS_REQUIRED_VERSION) - _require_version('hstore', - conn.hstore_version_tuple(), - HSTORE_REQUIRED_VERSION) + _require_loaded('hstore', conn) def setup_database_skeleton(dsn: str, rouser: Optional[str] = None) -> None: