Integrated OSM views into importance computation

[nominatim.git] / nominatim / clicmd / setup.py

diff --git a/nominatim/clicmd/setup.py b/nominatim/clicmd/setup.py
index c1cbab21c377601c8ebafdc3a22fd2136aeb3ee3..c870fb17132bf44511828a11e9eb7c93d7af5507 100644 (file)
--- a/nominatim/clicmd/setup.py
+++ b/nominatim/clicmd/setup.py
@@ -105,18 +105,18 @@ class SetupAll:
             if refresh.import_wikipedia_articles(args.config.get_libpq_dsn(),
                                                  data_path) > 0:
                 LOG.error('Wikipedia importance dump file not found. '
-                          'Calculating importance values of locations will not \
-                            use Wikipedia importance data.')
+                          'Calculating importance values of locations will not '
+                          'use Wikipedia importance data.')
 
             LOG.warning('Importing OSM views GeoTIFF data')
             data_path = Path(args.project_dir)
-            with connect(args.config.get_libpq_dsn()) as conn:
-                if refresh.import_osm_views_geotiff(conn, data_path) == 1:
-                    LOG.error('OSM views GeoTIFF file not found. '
-                          'Calculating importance values of locations will not use OSM views data.')
-                elif refresh.import_osm_views_geotiff(conn, data_path) == 2:
-                    LOG.error('PostGIS version number is less than 3. '
-                          'Calculating importance values of locations will not use OSM views data.')
+            num = refresh.import_osm_views_geotiff(args.config.get_libpq_dsn(), data_path)
+            if num == 1:
+                LOG.error('OSM views GeoTIFF file not found. '
+                        'Calculating importance values of locations will not use OSM views data.')
+            elif num == 2:
+                LOG.error('PostGIS version number is less than 3. '
+                        'Calculating importance values of locations will not use OSM views data.')
 
         if args.continue_at is None or args.continue_at == 'load-data':
             LOG.warning('Initialise tables')