make DB helper functions free functions

[nominatim.git] / src / nominatim_db / clicmd / refresh.py

diff --git a/src/nominatim_db/clicmd/refresh.py b/src/nominatim_db/clicmd/refresh.py
index d5acf54b3a36a0e8b0832081a436ed4ce62c586f..363bad7853431851264f53eab83be7f277e46e07 100644 (file)
--- a/src/nominatim_db/clicmd/refresh.py
+++ b/src/nominatim_db/clicmd/refresh.py
@@ -13,7 +13,7 @@ import logging
 from pathlib import Path
 
 from ..config import Configuration
 from pathlib import Path
 
 from ..config import Configuration

-from ..db.connection import connect
+from ..db.connection import connect, table_exists

 from ..tokenizer.base import AbstractTokenizer
 from .args import NominatimArgs
 
 from ..tokenizer.base import AbstractTokenizer
 from .args import NominatimArgs
 


@@ -124,7 +124,7 @@ class UpdateRefresh:
             with connect(args.config.get_libpq_dsn()) as conn:
                 # If the table did not exist before, then the importance code
                 # needs to be enabled.
             with connect(args.config.get_libpq_dsn()) as conn:
                 # If the table did not exist before, then the importance code
                 # needs to be enabled.

-                if not conn.table_exists('secondary_importance'):
+                if not table_exists(conn, 'secondary_importance'):

                     args.functions = True
 
             LOG.warning('Import secondary importance raster data from %s', args.project_dir)
                     args.functions = True
 
             LOG.warning('Import secondary importance raster data from %s', args.project_dir)