mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-03-10 12:04:06 +00:00
simplify constructor for SQL preprocessor
Use sql path from config.
This commit is contained in:
@@ -75,9 +75,9 @@ class SQLPreprocessor: # pylint: disable=too-few-public-methods
|
|||||||
and follows its syntax.
|
and follows its syntax.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, conn, config, sqllib_dir):
|
def __init__(self, conn, config):
|
||||||
self.env = jinja2.Environment(autoescape=False,
|
self.env = jinja2.Environment(autoescape=False,
|
||||||
loader=jinja2.FileSystemLoader(str(sqllib_dir)))
|
loader=jinja2.FileSystemLoader(str(config.lib_dir.sql)))
|
||||||
|
|
||||||
db_info = {}
|
db_info = {}
|
||||||
db_info['partitions'] = _get_partitions(conn)
|
db_info['partitions'] = _get_partitions(conn)
|
||||||
|
|||||||
@@ -184,7 +184,7 @@ def create_tables(conn, config, sqllib_dir, reverse_only=False):
|
|||||||
When `reverse_only` is True, then the main table for searching will
|
When `reverse_only` is True, then the main table for searching will
|
||||||
be skipped and only reverse search is possible.
|
be skipped and only reverse search is possible.
|
||||||
"""
|
"""
|
||||||
sql = SQLPreprocessor(conn, config, sqllib_dir)
|
sql = SQLPreprocessor(conn, config)
|
||||||
sql.env.globals['db']['reverse_only'] = reverse_only
|
sql.env.globals['db']['reverse_only'] = reverse_only
|
||||||
|
|
||||||
sql.run_sql_file(conn, 'tables.sql')
|
sql.run_sql_file(conn, 'tables.sql')
|
||||||
@@ -194,14 +194,14 @@ def create_table_triggers(conn, config, sqllib_dir):
|
|||||||
""" Create the triggers for the tables. The trigger functions must already
|
""" Create the triggers for the tables. The trigger functions must already
|
||||||
have been imported with refresh.create_functions().
|
have been imported with refresh.create_functions().
|
||||||
"""
|
"""
|
||||||
sql = SQLPreprocessor(conn, config, sqllib_dir)
|
sql = SQLPreprocessor(conn, config)
|
||||||
sql.run_sql_file(conn, 'table-triggers.sql')
|
sql.run_sql_file(conn, 'table-triggers.sql')
|
||||||
|
|
||||||
|
|
||||||
def create_partition_tables(conn, config, sqllib_dir):
|
def create_partition_tables(conn, config, sqllib_dir):
|
||||||
""" Create tables that have explicit partitioning.
|
""" Create tables that have explicit partitioning.
|
||||||
"""
|
"""
|
||||||
sql = SQLPreprocessor(conn, config, sqllib_dir)
|
sql = SQLPreprocessor(conn, config)
|
||||||
sql.run_sql_file(conn, 'partition-tables.src.sql')
|
sql.run_sql_file(conn, 'partition-tables.src.sql')
|
||||||
|
|
||||||
|
|
||||||
@@ -303,7 +303,7 @@ def create_search_indices(conn, config, sqllib_dir, drop=False):
|
|||||||
cur.execute('DROP INDEX "{}"'.format(idx))
|
cur.execute('DROP INDEX "{}"'.format(idx))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
|
|
||||||
sql = SQLPreprocessor(conn, config, sqllib_dir)
|
sql = SQLPreprocessor(conn, config)
|
||||||
|
|
||||||
sql.run_sql_file(conn, 'indices.sql', drop=drop)
|
sql.run_sql_file(conn, 'indices.sql', drop=drop)
|
||||||
|
|
||||||
|
|||||||
@@ -81,7 +81,7 @@ def create_functions(conn, config, sqllib_dir,
|
|||||||
enable_diff_updates=True, enable_debug=False):
|
enable_diff_updates=True, enable_debug=False):
|
||||||
""" (Re)create the PL/pgSQL functions.
|
""" (Re)create the PL/pgSQL functions.
|
||||||
"""
|
"""
|
||||||
sql = SQLPreprocessor(conn, config, sqllib_dir)
|
sql = SQLPreprocessor(conn, config)
|
||||||
|
|
||||||
sql.run_sql_file(conn, 'functions.sql',
|
sql.run_sql_file(conn, 'functions.sql',
|
||||||
disable_diff_updates=not enable_diff_updates,
|
disable_diff_updates=not enable_diff_updates,
|
||||||
|
|||||||
@@ -86,7 +86,7 @@ def add_tiger_data(data_dir, config, threads):
|
|||||||
return
|
return
|
||||||
|
|
||||||
with connect(dsn) as conn:
|
with connect(dsn) as conn:
|
||||||
sql = SQLPreprocessor(conn, config, config.lib_dir.sql)
|
sql = SQLPreprocessor(conn, config)
|
||||||
sql.run_sql_file(conn, 'tiger_import_start.sql')
|
sql.run_sql_file(conn, 'tiger_import_start.sql')
|
||||||
|
|
||||||
# Reading sql_files and then for each file line handling
|
# Reading sql_files and then for each file line handling
|
||||||
@@ -116,5 +116,5 @@ def add_tiger_data(data_dir, config, threads):
|
|||||||
print('\n')
|
print('\n')
|
||||||
LOG.warning("Creating indexes on Tiger data")
|
LOG.warning("Creating indexes on Tiger data")
|
||||||
with connect(dsn) as conn:
|
with connect(dsn) as conn:
|
||||||
sql = SQLPreprocessor(conn, config, config.lib_dir.sql)
|
sql = SQLPreprocessor(conn, config)
|
||||||
sql.run_sql_file(conn, 'tiger_import_finish.sql')
|
sql.run_sql_file(conn, 'tiger_import_finish.sql')
|
||||||
|
|||||||
@@ -280,7 +280,11 @@ def osm2pgsql_options(temp_db):
|
|||||||
main_data='', main_index=''))
|
main_data='', main_index=''))
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def sql_preprocessor(temp_db_conn, tmp_path, def_config, monkeypatch, table_factory):
|
def sql_preprocessor(temp_db_conn, tmp_path, monkeypatch, table_factory):
|
||||||
monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
|
monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
|
||||||
table_factory('country_name', 'partition INT', (0, 1, 2))
|
table_factory('country_name', 'partition INT', (0, 1, 2))
|
||||||
return SQLPreprocessor(temp_db_conn, def_config, tmp_path)
|
cfg = Configuration(None, SRC_DIR.resolve() / 'settings')
|
||||||
|
cfg.set_libdirs(module='.', osm2pgsql='.', php=SRC_DIR / 'lib-php',
|
||||||
|
sql=tmp_path, data=SRC_DIR / 'data')
|
||||||
|
|
||||||
|
return SQLPreprocessor(temp_db_conn, cfg)
|
||||||
|
|||||||
@@ -5,6 +5,11 @@ import pytest
|
|||||||
|
|
||||||
from nominatim.tools.refresh import create_functions
|
from nominatim.tools.refresh import create_functions
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def sql_tmp_path(tmp_path, def_config):
|
||||||
|
def_config.lib_dir.sql = tmp_path
|
||||||
|
return tmp_path
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def conn(temp_db_conn, table_factory, monkeypatch):
|
def conn(temp_db_conn, table_factory, monkeypatch):
|
||||||
monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
|
monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
|
||||||
@@ -12,8 +17,8 @@ def conn(temp_db_conn, table_factory, monkeypatch):
|
|||||||
return temp_db_conn
|
return temp_db_conn
|
||||||
|
|
||||||
|
|
||||||
def test_create_functions(temp_db_cursor, conn, def_config, tmp_path):
|
def test_create_functions(temp_db_cursor, conn, def_config, sql_tmp_path):
|
||||||
sqlfile = tmp_path / 'functions.sql'
|
sqlfile = sql_tmp_path / 'functions.sql'
|
||||||
sqlfile.write_text("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER
|
sqlfile.write_text("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER
|
||||||
AS $$
|
AS $$
|
||||||
BEGIN
|
BEGIN
|
||||||
@@ -22,14 +27,14 @@ def test_create_functions(temp_db_cursor, conn, def_config, tmp_path):
|
|||||||
$$ LANGUAGE plpgsql IMMUTABLE;
|
$$ LANGUAGE plpgsql IMMUTABLE;
|
||||||
""")
|
""")
|
||||||
|
|
||||||
create_functions(conn, def_config, tmp_path)
|
create_functions(conn, def_config, sql_tmp_path)
|
||||||
|
|
||||||
assert temp_db_cursor.scalar('SELECT test()') == 43
|
assert temp_db_cursor.scalar('SELECT test()') == 43
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("dbg,ret", ((True, 43), (False, 22)))
|
@pytest.mark.parametrize("dbg,ret", ((True, 43), (False, 22)))
|
||||||
def test_create_functions_with_template(temp_db_cursor, conn, def_config, tmp_path, dbg, ret):
|
def test_create_functions_with_template(temp_db_cursor, conn, def_config, sql_tmp_path, dbg, ret):
|
||||||
sqlfile = tmp_path / 'functions.sql'
|
sqlfile = sql_tmp_path / 'functions.sql'
|
||||||
sqlfile.write_text("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER
|
sqlfile.write_text("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER
|
||||||
AS $$
|
AS $$
|
||||||
BEGIN
|
BEGIN
|
||||||
@@ -42,6 +47,6 @@ def test_create_functions_with_template(temp_db_cursor, conn, def_config, tmp_pa
|
|||||||
$$ LANGUAGE plpgsql IMMUTABLE;
|
$$ LANGUAGE plpgsql IMMUTABLE;
|
||||||
""")
|
""")
|
||||||
|
|
||||||
create_functions(conn, def_config, tmp_path, enable_debug=dbg)
|
create_functions(conn, def_config, sql_tmp_path, enable_debug=dbg)
|
||||||
|
|
||||||
assert temp_db_cursor.scalar('SELECT test()') == ret
|
assert temp_db_cursor.scalar('SELECT test()') == ret
|
||||||
|
|||||||
Reference in New Issue
Block a user