Merge branch '3787-docker-docs' closes #3787
[arvados.git] / sdk / python / arvados / commands / _util.py
1 #!/usr/bin/env python
2
3 import argparse
4 import errno
5 import os
6
7 def _pos_int(s):
8     num = int(s)
9     if num < 0:
10         raise ValueError("can't accept negative value: %s" % (num,))
11     return num
12
13 retry_opt = argparse.ArgumentParser(add_help=False)
14 retry_opt.add_argument('--retries', type=_pos_int, default=3, help="""
15 Maximum number of times to retry server requests that encounter temporary
16 failures (e.g., server down).  Default 3.""")
17
18 def _ignore_error(error):
19     return None
20
21 def _raise_error(error):
22     raise error
23
24 def make_home_conf_dir(path, mode=None, errors='ignore'):
25     # Make the directory path under the user's home directory, making parent
26     # directories as needed.
27     # If the directory is newly created, and a mode is specified, chmod it
28     # with those permissions.
29     # If there's an error, return None if errors is 'ignore', else raise an
30     # exception.
31     error_handler = _ignore_error if (errors == 'ignore') else _raise_error
32     tilde_path = os.path.join('~', path)
33     abs_path = os.path.expanduser(tilde_path)
34     if abs_path == tilde_path:
35         return error_handler(ValueError("no home directory available"))
36     try:
37         os.makedirs(abs_path)
38     except OSError as error:
39         if error.errno != errno.EEXIST:
40             return error_handler(error)
41     else:
42         if mode is not None:
43             os.chmod(abs_path, mode)
44     return abs_path