X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/bef1aebf1c74277f3f9b8744a2a92dca0ef3f255..7205491b8495e48c62b28373d1746e77d475582b:/nominatim/db/sql_preprocessor.py?ds=sidebyside diff --git a/nominatim/db/sql_preprocessor.py b/nominatim/db/sql_preprocessor.py index 31b4a8c0..839f682d 100644 --- a/nominatim/db/sql_preprocessor.py +++ b/nominatim/db/sql_preprocessor.py @@ -57,9 +57,11 @@ def _setup_postgresql_features(conn: Connection) -> Dict[str, Any]: """ pg_version = conn.server_version_tuple() postgis_version = conn.postgis_version_tuple() + pg11plus = pg_version >= (11, 0, 0) + ps3 = postgis_version >= (3, 0) return { - 'has_index_non_key_column': pg_version >= (11, 0, 0), - 'spgist_geom' : 'SPGIST' if postgis_version >= (3, 0) else 'GIST' + 'has_index_non_key_column': pg11plus, + 'spgist_geom' : 'SPGIST' if pg11plus and ps3 else 'GIST' } class SQLPreprocessor: @@ -88,6 +90,18 @@ class SQLPreprocessor: self.env.globals['postgres'] = _setup_postgresql_features(conn) + def run_string(self, conn: Connection, template: str, **kwargs: Any) -> None: + """ Execute the given SQL template string on the connection. + The keyword arguments may supply additional parameters + for preprocessing. + """ + sql = self.env.from_string(template).render(**kwargs) + + with conn.cursor() as cur: + cur.execute(sql) + conn.commit() + + def run_sql_file(self, conn: Connection, name: str, **kwargs: Any) -> None: """ Execute the given SQL file on the connection. The keyword arguments may supply additional parameters for preprocessing. @@ -101,7 +115,7 @@ class SQLPreprocessor: def run_parallel_sql_file(self, dsn: str, name: str, num_threads: int = 1, **kwargs: Any) -> None: - """ Execure the given SQL files using parallel asynchronous connections. + """ Execute the given SQL files using parallel asynchronous connections. The keyword arguments may supply additional parameters for preprocessing.