convert connect() into a context manager

2021-02-23 10:11:21 +01:00
parent 204fe20b4b
commit e520613362
12 changed files with 53 additions and 59 deletions
--- a/nominatim/tools/check_database.py
+++ b/nominatim/tools/check_database.py
@@ -60,7 +60,7 @@ def check_database(config):
    """ Run a number of checks on the database and return the status.
    """
    try:
-        conn = connect(config.get_libpq_dsn())
+        conn = connect(config.get_libpq_dsn()).connection
    except UsageError as err:
        conn = _BadConnection(str(err))