sdk/python/arvados/config.py

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: Apache-2.0
   4
   5 # config.py - configuration settings and global variables for Arvados clients
   6 #
   7 # Arvados configuration settings are taken from $HOME/.config/arvados.
   8 # Environment variables override settings in the config file.
   9
  10 import os
  11 import re
  12
  13 _settings = None
  14 if os.environ.get('HOME') is not None:
  15     default_config_file = os.environ['HOME'] + '/.config/arvados/settings.conf'
  16 else:
  17     default_config_file = ''
  18
  19 KEEP_BLOCK_SIZE = 2**26
  20 EMPTY_BLOCK_LOCATOR = 'd41d8cd98f00b204e9800998ecf8427e+0'
  21
  22 def initialize(config_file=default_config_file):
  23     global _settings
  24     _settings = {}
  25
  26     # load the specified config file if available
  27     try:
  28         _settings = load(config_file)
  29     except IOError:
  30         pass
  31
  32     # override any settings with environment vars
  33     for var in os.environ:
  34         if var.startswith('ARVADOS_'):
  35             _settings[var] = os.environ[var]
  36
  37 def load(config_file):
  38     cfg = {}
  39     with open(config_file, "r") as f:
  40         for config_line in f:
  41             if re.match('^\s*$', config_line):
  42                 continue
  43             if re.match('^\s*#', config_line):
  44                 continue
  45             var, val = config_line.rstrip().split('=', 2)
  46             cfg[var] = val
  47     return cfg
  48
  49 def flag_is_true(key, d=None):
  50     if d is None:
  51         d = settings()
  52     return d.get(key, '').lower() in set(['1', 't', 'true', 'y', 'yes'])
  53
  54 def get(key, default_val=None):
  55     return settings().get(key, default_val)
  56
  57 def settings():
  58     if _settings is None:
  59         initialize()
  60     return _settings