replace_args()

Complexity

Conditions

7

Size

Total Lines

23

Duplication

Lines	0
Ratio	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
B	_replace_args()	0	13	6
A	_replace()	0	5	3
B	_wrapper()	0	11	5

import functools

__all__ = [
    'replace_args',
]


def replace_args(attribute=None):
    """
    Decorator to Apply replacements in a list of command line arguments.

    :param attribute: Class attribute name which stores replacement rules.
    :type attribute: ``str``
    :return:
    :rtype: ``callable``
    """
    def _replace_args(f):
        @functools.wraps(f)
        def _wrapper(self, *args):
            def _replace(arg):
                for rule in rules:
                    if arg.startswith(rule):
                        return arg.replace(rule, rules[rule])
                return arg
            rules = getattr(self, attribute)
            if not rules:
                return f(self, *args)
            return map(_replace, f(self, *args))
        return _wrapper
    return _replace_args