X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/c85b74497b385a633ed3f487c77cc0736d5f7328..105597be056bf9cbf3bb4c15f9bc516479e48ee2:/nominatim/tools/refresh.py diff --git a/nominatim/tools/refresh.py b/nominatim/tools/refresh.py index a3b6c4f0..43e5b1eb 100644 --- a/nominatim/tools/refresh.py +++ b/nominatim/tools/refresh.py @@ -15,10 +15,10 @@ from pathlib import Path from psycopg2 import sql as pysql from nominatim.config import Configuration -from nominatim.db.connection import Connection +from nominatim.db.connection import Connection, connect from nominatim.db.utils import execute_file from nominatim.db.sql_preprocessor import SQLPreprocessor -from nominatim.version import version_str +from nominatim.version import NOMINATIM_VERSION LOG = logging.getLogger() @@ -120,6 +120,7 @@ PHP_CONST_DEFS = ( ('Search_NameOnlySearchFrequencyThreshold', 'SEARCH_NAME_ONLY_THRESHOLD', str), ('Use_US_Tiger_Data', 'USE_US_TIGER_DATA', bool), ('MapIcon_URL', 'MAPICON_URL', str), + ('Search_WithinCountries', 'SEARCH_WITHIN_COUNTRIES', bool), ) @@ -146,23 +147,23 @@ def import_wikipedia_articles(dsn: str, data_path: Path, ignore_errors: bool = F return 0 -def import_osm_views_geotiff(dsn, data_path, ignore_errors=False): - """ Replaces the OSM views table with new data. - - Returns 0 if all was well and 1 if the GeoTIFF file could not +def import_secondary_importance(dsn: str, data_path: Path, ignore_errors: bool = False) -> int: + """ Replaces the secondary importance raster data table with new data. + + Returns 0 if all was well and 1 if the raster SQL file could not be found. Throws an exception if there was an error reading the file. """ - datafile = data_path / 'osmviews.tiff' - + datafile = data_path / 'secondary_importance.sql.gz' if not datafile.exists(): return 1 - pre_code = """BEGIN; - DROP TABLE IF EXISTS "osmviews"; - """ - post_code = "COMMIT" - execute_file(dsn, datafile, ignore_errors=ignore_errors, - pre_code=pre_code, post_code=post_code) + with connect(dsn) as conn: + postgis_version = conn.postgis_version_tuple() + if postgis_version[0] < 3: + LOG.error('PostGIS version is too old for using OSM raster data.') + return 2 + + execute_file(dsn, datafile, ignore_errors=ignore_errors) return 0 @@ -176,7 +177,7 @@ def recompute_importance(conn: Connection) -> None: cur.execute(""" UPDATE placex SET (wikipedia, importance) = (SELECT wikipedia, importance - FROM compute_importance(extratags, country_code, osm_type, osm_id)) + FROM compute_importance(extratags, country_code, rank_search, centroid)) """) cur.execute(""" UPDATE placex s SET wikipedia = d.wikipedia, importance = d.importance @@ -216,30 +217,33 @@ def setup_website(basedir: Path, config: Configuration, conn: Connection) -> Non LOG.info('Creating website directory.') basedir.mkdir() - template = dedent(f"""\ + assert config.project_dir is not None + basedata = dedent(f"""\