ceph/teuthology/task/args.py

"""
These routines only appear to be used by the peering_speed tests.
"""
def gen_args(name, args):
    """
    Called from argify to generate arguments.
    """
    usage = [""]
    usage += [name + ':']
    usage += \
        ["    {key}: <{usage}> ({default})".format(
            key=key, usage=_usage, default=default)
         for (key, _usage, default, _) in args]
    usage.append('')
    usage.append(name + ':')
    usage += \
        ["    {key}: {default}".format(
                key = key, default = default)
         for (key, _, default, _) in args]
    usage = '\n'.join('    ' + i for i in usage)
    def ret(config):
        """
        return an object with attributes set from args. 
        """
        class Object(object):
            """
            simple object
            """
            pass
        obj = Object()
        for (key, usage, default, conv) in args:
            if key in config:
                setattr(obj, key, conv(config[key]))
            else:
                setattr(obj, key, conv(default))
        return obj
    return usage, ret

def argify(name, args):
    """
    Object used as a decorator for the peering speed tests.
    See peering_spee_test.py
    """
    (usage, config_func) = gen_args(name, args)
    def ret1(f):
        """
        Wrapper to handle doc and usage information
        """
        def ret2(**kwargs):
            """
            Call f (the parameter passed to ret1) 
            """
            config = kwargs.get('config', {})
            if config is None:
                config = {}
            kwargs['config'] = config_func(config)
            return f(**kwargs)
        ret2.__doc__ = f.__doc__ + usage
        return ret2
    return ret1