nominatim/clicmd/add_data.py

   1 """
   2 Implementation of the 'add-data' subcommand.
   3 """
   4 import logging
   5
   6 from nominatim.tools.exec_utils import run_legacy_script
   7
   8 # Do not repeat documentation of subcommand classes.
   9 # pylint: disable=C0111
  10 # Using non-top-level imports to avoid eventually unused imports.
  11 # pylint: disable=E0012,C0415
  12
  13 LOG = logging.getLogger()
  14
  15 class UpdateAddData:
  16     """\
  17     Add additional data from a file or an online source.
  18
  19     Data is only imported, not indexed. You need to call `nominatim index`
  20     to complete the process.
  21     """
  22
  23     @staticmethod
  24     def add_args(parser):
  25         group_name = parser.add_argument_group('Source')
  26         group = group_name.add_mutually_exclusive_group(required=True)
  27         group.add_argument('--file', metavar='FILE',
  28                            help='Import data from an OSM file')
  29         group.add_argument('--diff', metavar='FILE',
  30                            help='Import data from an OSM diff file')
  31         group.add_argument('--node', metavar='ID', type=int,
  32                            help='Import a single node from the API')
  33         group.add_argument('--way', metavar='ID', type=int,
  34                            help='Import a single way from the API')
  35         group.add_argument('--relation', metavar='ID', type=int,
  36                            help='Import a single relation from the API')
  37         group.add_argument('--tiger-data', metavar='DIR',
  38                            help='Add housenumbers from the US TIGER census database.')
  39         group = parser.add_argument_group('Extra arguments')
  40         group.add_argument('--use-main-api', action='store_true',
  41                            help='Use OSM API instead of Overpass to download objects')
  42
  43     @staticmethod
  44     def run(args):
  45         from nominatim.tokenizer import factory as tokenizer_factory
  46         from nominatim.tools import tiger_data
  47
  48         if args.tiger_data:
  49             tokenizer = tokenizer_factory.get_tokenizer_for_db(args.config)
  50             return tiger_data.add_tiger_data(args.tiger_data,
  51                                              args.config, args.threads or 1,
  52                                              tokenizer)
  53
  54         params = ['update.php']
  55         if args.file:
  56             params.extend(('--import-file', args.file))
  57         elif args.diff:
  58             params.extend(('--import-diff', args.diff))
  59         elif args.node:
  60             params.extend(('--import-node', args.node))
  61         elif args.way:
  62             params.extend(('--import-way', args.way))
  63         elif args.relation:
  64             params.extend(('--import-relation', args.relation))
  65         if args.use_main_api:
  66             params.append('--use-main-api')
  67         return run_legacy_script(*params, nominatim_env=args)