def __call__()

in faker_datasets/__init__.py [0:0]

• 14 lines of code
• 4 McCabe index (conditional complexity)

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

        def set_datasets(datasets):
            try:
                func.datasets = tuple(datasets[name] for name in self.names)
            except KeyError as e:
                raise ValueError(f"dataset not found: '{e.args[0]}'") from None
            if hasattr(func, "set_datasets"):
                func.set_datasets(datasets)

        _func.set_datasets = set_datasets
        return _func