def __call__()

in faker_datasets/__init__.py [0:0]

• 13 lines of code
• 6 McCabe index (conditional complexity)

    def __call__(self, func):
        @wraps(func)
        def _func(faker, *args, **kwargs):
            args = func.datasets + args[len(func.datasets) :]
            return func(faker, *args, **kwargs)

        def set_datasets(datasets):
            if not hasattr(_func, "datasets"):
                raise ValueError("Use @with_datasets first")
            func.datasets = tuple([x for x in d if self.match(x)] for d in _func.datasets)
            if hasattr(func, "set_datasets"):
                func.set_datasets(datasets)

        _func.set_datasets = set_datasets
        return _func