nominatim/config.py

   1 """
   2 Nominatim configuration accessor.
   3 """
   4 import os
   5
   6 from dotenv import dotenv_values
   7
   8 class Configuration:
   9     """ Load and manage the project configuration.
  10
  11         Nominatim uses dotenv to configure the software. Configuration options
  12         are resolved in the following order:
  13
  14          * from the OS environment
  15          * from the .env file in the project directory of the installation
  16          * from the default installation in the configuration directory
  17
  18         All Nominatim configuration options are prefixed with 'NOMINATIM_' to
  19         avoid conflicts with other environment variables.
  20     """
  21
  22     def __init__(self, project_dir, config_dir):
  23         self._config = dotenv_values(str((config_dir / 'env.defaults').resolve()))
  24         if project_dir is not None:
  25             self._config.update(dotenv_values(str((project_dir / '.env').resolve())))
  26
  27     def __getattr__(self, name):
  28         name = 'NOMINATIM_' + name
  29
  30         return os.environ.get(name) or self._config[name]
  31
  32     def get_libpq_dsn(self):
  33         """ Get configured database DSN converted into the key/value format
  34             understood by libpq and psycopg.
  35         """
  36         dsn = self.DATABASE_DSN
  37
  38         if dsn.startswith('pgsql:'):
  39             # Old PHP DSN format. Convert before returning.
  40             return dsn[6:].replace(';', ' ')
  41
  42         return dsn
  43
  44     def get_os_env(self):
  45         """ Return a copy of the OS environment with the Nominatim configuration
  46             merged in.
  47         """
  48         env = dict(self._config)
  49         env.update(os.environ)
  50
  51         return env