X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/b6ff697ff005629727a46dcc4c32f6a4f554b21b..899a04ad264c45197ec7509b164c934a05392f30:/nominatim/clicmd/args.py diff --git a/nominatim/clicmd/args.py b/nominatim/clicmd/args.py index 45599ad5..eb3a3b61 100644 --- a/nominatim/clicmd/args.py +++ b/nominatim/clicmd/args.py @@ -10,11 +10,13 @@ Provides custom functions over command-line arguments. from typing import Optional, List, Dict, Any, Sequence, Tuple import argparse import logging +from functools import reduce from pathlib import Path from nominatim.errors import UsageError from nominatim.config import Configuration from nominatim.typing import Protocol +import nominatim.api as napi LOG = logging.getLogger() @@ -42,13 +44,6 @@ class NominatimArgs: # Basic environment set by root program. config: Configuration project_dir: Path - module_dir: Path - osm2pgsql_path: Path - phplib_dir: Path - sqllib_dir: Path - data_dir: Path - config_dir: Path - phpcgi_path: Path # Global switches version: bool @@ -77,6 +72,7 @@ class NominatimArgs: check_database: bool migrate: bool collect_os_info: bool + clean_deleted: str analyse_indexing: bool target: Optional[str] osm_id: Optional[str] @@ -104,9 +100,9 @@ class NominatimArgs: output_all_postcodes: bool language: Optional[str] restrict_to_country: Optional[str] - restrict_to_osm_node: Optional[int] - restrict_to_osm_way: Optional[int] - restrict_to_osm_relation: Optional[int] + + # Arguments to 'convert' + output: Path # Arguments to 'refresh' postcodes: bool @@ -133,6 +129,7 @@ class NominatimArgs: # Arguments to 'serve' server: str + engine: str # Arguments to 'special-phrases import_from_wiki: bool @@ -150,6 +147,7 @@ class NominatimArgs: # Arguments to 'search' query: Optional[str] + amenity: Optional[str] street: Optional[str] city: Optional[str] county: Optional[str] @@ -158,7 +156,7 @@ class NominatimArgs: postalcode: Optional[str] countrycodes: Optional[str] exclude_place_ids: Optional[str] - limit: Optional[int] + limit: int viewbox: Optional[str] bounded: bool dedupe: bool @@ -167,12 +165,18 @@ class NominatimArgs: lat: float lon: float zoom: Optional[int] + layers: Optional[Sequence[str]] # Arguments to 'lookup' ids: Sequence[str] # Arguments to 'details' object_class: Optional[str] + linkedplaces: bool + hierarchy: bool + keywords: bool + polygon_geojson: bool + group_hierarchy: bool def osm2pgsql_options(self, default_cache: int, @@ -181,13 +185,12 @@ class NominatimArgs: from the command line arguments. The resulting dict can be further customized and then used in `run_osm2pgsql()`. """ - return dict(osm2pgsql=self.config.OSM2PGSQL_BINARY or self.osm2pgsql_path, + return dict(osm2pgsql=self.config.OSM2PGSQL_BINARY or self.config.lib_dir.osm2pgsql, osm2pgsql_cache=self.osm2pgsql_cache or default_cache, osm2pgsql_style=self.config.get_import_style_file(), osm2pgsql_style_path=self.config.config_dir, threads=self.threads or default_threads, dsn=self.config.get_libpq_dsn(), - forward_dependencies=self.config.get_bool('UPDATE_FORWARD_DEPENDENCIES'), flatnode_file=str(self.config.get_path('FLATNODE_FILE') or ''), tablespaces=dict(slim_data=self.config.TABLESPACE_OSM_DATA, slim_index=self.config.TABLESPACE_OSM_INDEX, @@ -212,3 +215,45 @@ class NominatimArgs: raise UsageError('Cannot access file.') return files + + + def get_geometry_output(self) -> napi.GeometryFormat: + """ Get the requested geometry output format in a API-compatible + format. + """ + if not self.polygon_output: + return napi.GeometryFormat.NONE + if self.polygon_output == 'geojson': + return napi.GeometryFormat.GEOJSON + if self.polygon_output == 'kml': + return napi.GeometryFormat.KML + if self.polygon_output == 'svg': + return napi.GeometryFormat.SVG + if self.polygon_output == 'text': + return napi.GeometryFormat.TEXT + + try: + return napi.GeometryFormat[self.polygon_output.upper()] + except KeyError as exp: + raise UsageError(f"Unknown polygon output format '{self.polygon_output}'.") from exp + + + def get_locales(self, default: Optional[str]) -> napi.Locales: + """ Get the locales from the language parameter. + """ + if self.lang: + return napi.Locales.from_accept_languages(self.lang) + if default: + return napi.Locales.from_accept_languages(default) + + return napi.Locales() + + + def get_layers(self, default: napi.DataLayer) -> Optional[napi.DataLayer]: + """ Get the list of selected layers as a DataLayer enum. + """ + if not self.layers: + return default + + return reduce(napi.DataLayer.__or__, + (napi.DataLayer[s.upper()] for s in self.layers))