]> git.openstreetmap.org Git - nominatim.git/blob - nominatim/clicmd/index.py
Merge pull request #2297 from lonvia/update-deployment-docs
[nominatim.git] / nominatim / clicmd / index.py
1 """
2 Implementation of the 'index' subcommand.
3 """
4 import psutil
5
6 from nominatim.db import status
7 from nominatim.db.connection import connect
8
9 # Do not repeat documentation of subcommand classes.
10 # pylint: disable=C0111
11 # Using non-top-level imports to avoid eventually unused imports.
12 # pylint: disable=E0012,C0415
13
14
15 class UpdateIndex:
16     """\
17     Reindex all new and modified data.
18     """
19
20     @staticmethod
21     def add_args(parser):
22         group = parser.add_argument_group('Filter arguments')
23         group.add_argument('--boundaries-only', action='store_true',
24                            help="""Index only administrative boundaries.""")
25         group.add_argument('--no-boundaries', action='store_true',
26                            help="""Index everything except administrative boundaries.""")
27         group.add_argument('--minrank', '-r', type=int, metavar='RANK', default=0,
28                            help='Minimum/starting rank')
29         group.add_argument('--maxrank', '-R', type=int, metavar='RANK', default=30,
30                            help='Maximum/finishing rank')
31
32     @staticmethod
33     def run(args):
34         from ..indexer.indexer import Indexer
35
36         indexer = Indexer(args.config.get_libpq_dsn(),
37                           args.threads or psutil.cpu_count() or 1)
38
39         if not args.no_boundaries:
40             indexer.index_boundaries(args.minrank, args.maxrank)
41         if not args.boundaries_only:
42             indexer.index_by_rank(args.minrank, args.maxrank)
43
44         if not args.no_boundaries and not args.boundaries_only \
45            and args.minrank == 0 and args.maxrank == 30:
46             with connect(args.config.get_libpq_dsn()) as conn:
47                 status.set_indexed(conn, True)
48
49         return 0