How to get dict dump of Factory

Hello,

I need to get a full dump of a Factory as a dict so that I can serialize it to json. FactoryName.attributes() works great except when there is a SubField, in which case it evaluates it and returns an instantiated version of the object as opposed to a dictionary of key: value of the SubField attributes.

I tried to hack around this by traversing the Factory myself but it turns out that I’ll have to duplicate a lot of code that already exists. Do you see an easy way to do this?

[aalvrz]

@rbarrois

I am trying to use the factory.build(dict, FACTORY_CLASS=MyFactory) method in my REST API tests. However, the dictionary returned by that method contains a reference to an instance of an object in SubFactory() fields:

{'isbn': '12345', 'materia': <Materia: materia_2>, 'precio': 99.99, 'titulo': 'titulo_0', 'bisac_primario': <CodigoBisac: bisac_2>, 'idioma': <Idioma: I_2 - I_2>, 'moneda': <Moneda: moneda_2>, 'volumen': 1, 'edicion': 3, 'autor': <Autor: autor_2>}

I think it would be better if this would return the ID of this object instance instead.

Is there any built-in method that does this?

[chanyou0311]

@hectorcanto Great way!

I used that code as a reference and created a function that also outputs SubFactory as a dict.

from functools import partial
from typing import Any, Dict

from factory import Factory
from factory.base import StubObject


def generate_dict_factory(factory: Factory):
    def convert_dict_from_stub(stub: StubObject) -> Dict[str, Any]:
        stub_dict = stub.__dict__
        for key, value in stub_dict.items():
            if isinstance(value, StubObject):
                stub_dict[key] = convert_dict_from_stub(value)
        return stub_dict

    def dict_factory(factory, **kwargs):
        stub = factory.stub(**kwargs)
        stub_dict = convert_dict_from_stub(stub)
        return stub_dict

    return partial(dict_factory, factory)

# example of usage
UserDictFactory = generate_dict_factory(UserFactory)

We look forward to helping you.