18842: Clean up keep cache set logic a little more
[arvados.git] / 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