def _get_mommy_args(self, model, _create, **kwargs): if issubclass(model, Model): factory = self.get_factory(model) elif issubclass(model, DjangoMommyFactory): factory = model elif issubclass(model, Factory): return Factory.attributes(_create, kwargs) else: raise TypeError('argument must be a model instance or a ' 'DjangoMommyFactory subclass') kwargs = factory._get_mommy_args(_create, **kwargs) return kwargs