X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/7222235579bc0d287c86fea6d7e22959132c7377..a79a3210e69c648e3e5c1123b4049ce0419b484c:/nominatim/clicmd/args.py diff --git a/nominatim/clicmd/args.py b/nominatim/clicmd/args.py index f9b94ef2..d1f47ba0 100644 --- a/nominatim/clicmd/args.py +++ b/nominatim/clicmd/args.py @@ -1,7 +1,18 @@ +# SPDX-License-Identifier: GPL-2.0-only +# +# This file is part of Nominatim. (https://nominatim.org) +# +# Copyright (C) 2022 by the Nominatim developer community. +# For a full list of authors see the git log. """ Provides custom functions over command-line arguments. """ +import logging +from pathlib import Path +from nominatim.errors import UsageError + +LOG = logging.getLogger() class NominatimArgs: """ Customized namespace class for the nominatim command line tool @@ -18,5 +29,27 @@ class NominatimArgs: osm2pgsql_style=self.config.get_import_style_file(), threads=self.threads or default_threads, dsn=self.config.get_libpq_dsn(), - flatnode_file=self.config.FLATNODE_FILE) + flatnode_file=str(self.config.get_path('FLATNODE_FILE')), + tablespaces=dict(slim_data=self.config.TABLESPACE_OSM_DATA, + slim_index=self.config.TABLESPACE_OSM_INDEX, + main_data=self.config.TABLESPACE_PLACE_DATA, + main_index=self.config.TABLESPACE_PLACE_INDEX + ) + ) + + + def get_osm_file_list(self): + """ Return the --osm-file argument as a list of Paths or None + if no argument was given. The function also checks if the files + exist and raises a UsageError if one cannot be found. + """ + if not self.osm_file: + return None + + files = [Path(f) for f in self.osm_file] + for fname in files: + if not fname.is_file(): + LOG.fatal("OSM file '%s' does not exist.", fname) + raise UsageError('Cannot access file.') + return files