src/nominatim_db/clicmd/convert.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2024 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Implementation of the 'convert' subcommand.
   9 """
  10 from typing import Set, Any, Union, Optional, Sequence
  11 import argparse
  12 import asyncio
  13 from pathlib import Path
  14
  15 from ..errors import UsageError
  16 from .args import NominatimArgs
  17
  18
  19 class WithAction(argparse.Action):
  20     """ Special action that saves a list of flags, given on the command-line
  21         as `--with-foo` or `--without-foo`.
  22     """
  23     def __init__(self, option_strings: Sequence[str], dest: Any,
  24                  default: bool = True, **kwargs: Any) -> None:
  25         if 'nargs' in kwargs:
  26             raise ValueError("nargs not allowed.")
  27         if option_strings is None:
  28             raise ValueError("Positional parameter not allowed.")
  29
  30         self.dest_set = kwargs.pop('dest_set')
  31         full_option_strings = []
  32         for opt in option_strings:
  33             if not opt.startswith('--'):
  34                 raise ValueError("short-form options not allowed")
  35             if default:
  36                 self.dest_set.add(opt[2:])
  37             full_option_strings.append(f"--with-{opt[2:]}")
  38             full_option_strings.append(f"--without-{opt[2:]}")
  39
  40         super().__init__(full_option_strings, argparse.SUPPRESS, nargs=0, **kwargs)
  41
  42     def __call__(self, parser: argparse.ArgumentParser, namespace: argparse.Namespace,
  43                  values: Union[str, Sequence[Any], None],
  44                  option_string: Optional[str] = None) -> None:
  45         assert option_string
  46         if option_string.startswith('--with-'):
  47             self.dest_set.add(option_string[7:])
  48         if option_string.startswith('--without-'):
  49             self.dest_set.discard(option_string[10:])
  50
  51
  52 class ConvertDB:
  53     """ Convert an existing database into a different format. (EXPERIMENTAL)
  54
  55         Dump a read-only version of the database in a different format.
  56         At the moment only a SQLite database suitable for reverse lookup
  57         can be created.
  58     """
  59
  60     def __init__(self) -> None:
  61         self.options: Set[str] = set()
  62
  63     def add_args(self, parser: argparse.ArgumentParser) -> None:
  64         parser.add_argument('--format', default='sqlite',
  65                             choices=('sqlite', ),
  66                             help='Format of the output database (must be sqlite currently)')
  67         parser.add_argument('--output', '-o', required=True, type=Path,
  68                             help='File to write the database to.')
  69         group = parser.add_argument_group('Switches to define database layout'
  70                                           '(currently no effect)')
  71         group.add_argument('--reverse', action=WithAction, dest_set=self.options, default=True,
  72                            help='Enable/disable support for reverse and lookup API'
  73                                 ' (default: enabled)')
  74         group.add_argument('--search', action=WithAction, dest_set=self.options, default=True,
  75                            help='Enable/disable support for search API (default: disabled)')
  76         group.add_argument('--details', action=WithAction, dest_set=self.options, default=True,
  77                            help='Enable/disable support for details API (default: enabled)')
  78
  79     def run(self, args: NominatimArgs) -> int:
  80         if args.output.exists():
  81             raise UsageError(f"File '{args.output}' already exists. Refusing to overwrite.")
  82
  83         if args.format == 'sqlite':
  84             from ..tools import convert_sqlite
  85
  86             asyncio.run(convert_sqlite.convert(args.project_dir, args.output, self.options))
  87             return 0
  88
  89         return 1