implementaion of 'nominatim index'

[nominatim.git] / nominatim / admin / exec_utils.py

diff --git a/nominatim/admin/exec_utils.py b/nominatim/admin/exec_utils.py
index 91225b7aa9268d10a4a93191483ab504ed734e9c..0158227957f04bc6afa76f820e276672213a1e7f 100644 (file)
--- a/nominatim/admin/exec_utils.py
+++ b/nominatim/admin/exec_utils.py
@@ -1,21 +1,27 @@
 """
 Helper functions for executing external programs.
 """
 """
 Helper functions for executing external programs.
 """

-from pathlib import Path

 import subprocess
 
 import subprocess
 

-def run_legacy_script(script, *args, nominatim_env=None):
-        """ Run a Nominatim PHP script with the given arguments.
-        """
-        cmd = ['/usr/bin/env', 'php', '-Cq',
-               nominatim_env.phplib_dir / 'admin' / script]
-        cmd.extend(args)
+def run_legacy_script(script, *args, nominatim_env=None, throw_on_fail=False):
+    """ Run a Nominatim PHP script with the given arguments.

 
 

-        env = nominatim_env.config.get_os_env()
-        env['NOMINATIM_DATADIR'] = str(nominatim_env.data_dir)
-        env['NOMINATIM_BINDIR'] = str(nominatim_env.data_dir / 'utils')
+        Returns the exit code of the script. If `throw_on_fail` is True
+        then throw a `CalledProcessError` on a non-zero exit.
+    """
+    cmd = ['/usr/bin/env', 'php', '-Cq',
+           nominatim_env.phplib_dir / 'admin' / script]
+    cmd.extend([str(a) for a in args])

 
 

-        proc = subprocess.run(cmd, cwd=str(nominatim_env.project_dir), env=env)
+    env = nominatim_env.config.get_os_env()
+    env['NOMINATIM_DATADIR'] = str(nominatim_env.data_dir)
+    env['NOMINATIM_BINDIR'] = str(nominatim_env.data_dir / 'utils')
+    if not env['NOMINATIM_DATABASE_MODULE_PATH']:
+        env['NOMINATIM_DATABASE_MODULE_PATH'] = nominatim_env.module_dir
+    if not env['NOMINATIM_OSM2PGSQL_BINARY']:
+        env['NOMINATIM_OSM2PGSQL_BINARY'] = nominatim_env.osm2pgsql_path

 
 

-        return proc.returncode
+    proc = subprocess.run(cmd, cwd=str(nominatim_env.project_dir), env=env,
+                          check=throw_on_fail)

 
 

+    return proc.returncode