X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/c7f40e3cee2853ade0e16a90ab7a3b29b02265f9..eacabb0e962cf5301c63622204dd8efa0ce0a7ac:/nominatim/db/utils.py

diff --git a/nominatim/db/utils.py b/nominatim/db/utils.py
index 6d2eb297..0a2e2c06 100644
--- a/nominatim/db/utils.py
+++ b/nominatim/db/utils.py
@@ -35,24 +35,25 @@ def execute_file(dsn, fname, ignore_errors=False, pre_code=None, post_code=None)
         cmd.append('--quiet')
     proc = subprocess.Popen(cmd, env=get_pg_env(dsn), stdin=subprocess.PIPE)
 
-    if not LOG.isEnabledFor(logging.INFO):
-        proc.stdin.write('set client_min_messages to WARNING;'.encode('utf-8'))
-
-    if pre_code:
-        proc.stdin.write((pre_code + ';').encode('utf-8'))
-
-    if fname.suffix == '.gz':
-        with gzip.open(str(fname), 'rb') as fdesc:
-            remain = _pipe_to_proc(proc, fdesc)
-    else:
-        with fname.open('rb') as fdesc:
-            remain = _pipe_to_proc(proc, fdesc)
-
-    if remain == 0 and post_code:
-        proc.stdin.write((';' + post_code).encode('utf-8'))
-
-    proc.stdin.close()
+    try:
+        if not LOG.isEnabledFor(logging.INFO):
+            proc.stdin.write('set client_min_messages to WARNING;'.encode('utf-8'))
+
+        if pre_code:
+            proc.stdin.write((pre_code + ';').encode('utf-8'))
+
+        if fname.suffix == '.gz':
+            with gzip.open(str(fname), 'rb') as fdesc:
+                remain = _pipe_to_proc(proc, fdesc)
+        else:
+            with fname.open('rb') as fdesc:
+                remain = _pipe_to_proc(proc, fdesc)
+
+        if remain == 0 and post_code:
+            proc.stdin.write((';' + post_code).encode('utf-8'))
+    finally:
+        proc.stdin.close()
+        ret = proc.wait()
 
-    ret = proc.wait()
     if ret != 0 or remain > 0:
         raise UsageError("Failed to execute SQL file.")