Add -b/--enable-beta option for enabling beta functionalities.

This commit is contained in:
Darko Poljak 2016-08-15 14:20:35 +02:00
commit 1c07b63f1d
4 changed files with 35 additions and 6 deletions

View file

@ -71,6 +71,8 @@ def inspect_ssh_mux_opts():
class Config(object):
"""Cdist main class to hold arbitrary data"""
BETA_ARGS = ['jobs']
def __init__(self, local, remote, dry_run=False, jobs=None):
self.local = local
@ -109,6 +111,19 @@ class Config(object):
for host in source:
yield host
@classmethod
def _check_beta(cls, args_dict):
if 'beta' not in args_dict:
args_dict['beta'] = False
# Check only if beta is not enabled: if beta option is specified then
# raise error.
if not args_dict['beta']:
err_msg = ("\'{}\' is beta, but beta is not enabled. If you want "
"to use it please enable beta functionalities.")
for arg in cls.BETA_ARGS:
if arg in args_dict:
raise cdist.Error(err_msg.format(arg))
@classmethod
def commandline(cls, args):
"""Configure remote system"""
@ -120,6 +135,10 @@ class Config(object):
if args.manifest == '-' and args.hostfile == '-':
raise cdist.Error(("Cannot read both, manifest and host file, "
"from stdin"))
args_dict = vars(args)
cls._check_beta(args_dict)
# if no host source is specified then read hosts from stdin
if not (args.hostfile or args.host):
args.hostfile = '-'
@ -148,7 +167,6 @@ class Config(object):
args.remote_exec_pattern = None
args.remote_copy_pattern = None
args_dict = vars(args)
# if remote-exec and/or remote-copy args are None then user
# didn't specify command line options nor env vars:
# inspect multiplexing options for default cdist.REMOTE_COPY/EXEC