X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/cc0bdd34e98c9308a6fd52580c620cfc1f28ed12..019a68a4bbc90834386dec3f4fac04dd64afc5ce:/nominatim/tools/database_import.py diff --git a/nominatim/tools/database_import.py b/nominatim/tools/database_import.py index dbbb196e..de7e6a4a 100644 --- a/nominatim/tools/database_import.py +++ b/nominatim/tools/database_import.py @@ -23,7 +23,8 @@ from nominatim.db.async_connection import DBConnection 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 +from nominatim.version import POSTGRESQL_REQUIRED_VERSION, \ + POSTGIS_REQUIRED_VERSION LOG = logging.getLogger() @@ -38,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', @@ -47,6 +55,7 @@ def check_existing_database_plugins(dsn: str): _require_version('PostGIS', conn.postgis_version_tuple(), POSTGIS_REQUIRED_VERSION) + _require_loaded('hstore', conn) def setup_database_skeleton(dsn: str, rouser: Optional[str] = None) -> None: