2 Implementation of the 'transition' subcommand.
4 This subcommand provides standins for functions that were available
5 through the PHP scripts but are now no longer directly accessible.
6 This module will be removed as soon as the transition phase is over.
9 from pathlib import Path
11 from ..db.connection import connect
12 from ..db import status
13 from ..errors import UsageError
15 # Do not repeat documentation of subcommand classes.
16 # pylint: disable=C0111
17 # Using non-top-level imports to avoid eventually unused imports.
18 # pylint: disable=E0012,C0415
20 LOG = logging.getLogger()
22 class AdminTransition:
24 Internal functions for code transition. Do not use.
29 group = parser.add_argument_group('Sub-functions')
30 group.add_argument('--create-db', action='store_true',
31 help='Create nominatim db')
32 group.add_argument('--setup-db', action='store_true',
33 help='Build a blank nominatim db')
34 group.add_argument('--import-data', action='store_true',
35 help='Import a osm file')
36 group.add_argument('--load-data', action='store_true',
37 help='Copy data to live tables from import table')
38 group.add_argument('--create-tables', action='store_true',
39 help='Create main tables')
40 group.add_argument('--create-partition-tables', action='store_true',
41 help='Create required partition tables')
42 group.add_argument('--index', action='store_true',
43 help='Index the data')
44 group = parser.add_argument_group('Options')
45 group.add_argument('--no-partitions', action='store_true',
46 help='Do not partition search indices')
47 group.add_argument('--osm-file', metavar='FILE',
48 help='File to import')
49 group.add_argument('--drop', action='store_true',
50 help='Drop tables needed for updates, making the database readonly')
51 group.add_argument('--osm2pgsql-cache', metavar='SIZE', type=int,
52 help='Size of cache to be used by osm2pgsql (in MB)')
53 group.add_argument('--no-analyse', action='store_true',
54 help='Do not perform analyse operations during index')
55 group.add_argument('--ignore-errors', action='store_true',
56 help="Ignore certain erros on import.")
57 group.add_argument('--reverse-only', action='store_true',
58 help='Do not create search tables and indexes')
62 from ..tools import database_import
63 from ..tools import refresh
66 LOG.warning('Create DB')
67 database_import.create_db(args.config.get_libpq_dsn())
70 LOG.warning('Setup DB')
72 with connect(args.config.get_libpq_dsn()) as conn:
73 database_import.setup_extensions(conn)
74 database_import.install_module(args.module_dir, args.project_dir,
75 args.config.DATABASE_MODULE_PATH,
78 database_import.import_base_data(args.config.get_libpq_dsn(),
79 args.data_dir, args.no_partitions)
82 LOG.warning('Import data')
84 raise UsageError('Missing required --osm-file argument')
85 database_import.import_osm_data(Path(args.osm_file),
86 args.osm2pgsql_options(0, 1),
88 ignore_errors=args.ignore_errors)
90 if args.create_tables:
91 LOG.warning('Create Tables')
92 with connect(args.config.get_libpq_dsn()) as conn:
93 database_import.create_tables(conn, args.config, args.sqllib_dir, args.reverse_only)
94 refresh.load_address_levels_from_file(conn, Path(args.config.ADDRESS_LEVEL_CONFIG))
95 refresh.create_functions(conn, args.config, args.sqllib_dir,
96 enable_diff_updates=False)
97 database_import.create_table_triggers(conn, args.config, args.sqllib_dir)
99 if args.create_partition_tables:
100 LOG.warning('Create Partition Tables')
101 with connect(args.config.get_libpq_dsn()) as conn:
102 database_import.create_partition_tables(conn, args.config, args.sqllib_dir)
105 LOG.warning('Load data')
106 with connect(args.config.get_libpq_dsn()) as conn:
107 database_import.truncate_data_tables(conn, args.config.MAX_WORD_FREQUENCY)
108 database_import.load_data(args.config.get_libpq_dsn(),
112 with connect(args.config.get_libpq_dsn()) as conn:
114 status.set_status(conn, status.compute_database_date(conn))
115 except Exception as exc: # pylint: disable=broad-except
116 LOG.error('Cannot determine date of database: %s', exc)
119 LOG.warning('Indexing')
120 from ..indexer.indexer import Indexer
121 indexer = Indexer(args.config.get_libpq_dsn(), args.threads or 1)