Add SQLAlchemy support for get_or_create()

I’m not sure this should be opened as an issue, but I can’t figure out how to do it…

Is there any way to re-use existing records with factory_boy, if they already exist? I can’t find anything in the documentation.

Here’s my scenario:

I’ve got an AddressFactory for my Address model. Every Address has an associated Country. I’d like to have my AddressFactory create the Country if it doesn’t exist, and re-use it if it does.

I know that I can create a Country manually, and pass it to the AddressFactory to be used. This works ok for a first-level nesting, but it gets complicated when there’s e.g. a User who has an Address which has a Country. I’d like to avoid having to pass it all the way down the chain.

For most fields, I get around this problem by using sequences to ensure unique names, etc. In this case, I’d really like to re-use existing countries.

I’ve tried to come up with a way to do this dynamically using lazy attributes, but haven’t had any luck so far.

Thanks!

[rbarrois]

This has finally been added last year 😉

[darkanthey]

Another one SqlAlchemy avoid duplication implementation.

class StaticFactory(BaseFactory):
    __static_exclude = ('__static_exclude', '__static_cache',)
    __static_cache = {}
 
    @classmethod
    def _create(cls, model_class, *args, **kwargs):
        """Helper for avoid duplicate factory"""
 
        # Exclude static cache
        cls._meta.exclude += cls.__static_exclude
 
        _unique_key = None
 
        # Get first unique keys from table. It'll be cache key.
        for col in model_class.__table__.columns:
            if any([col.primary_key, col.unique]):
                _unique_key = kwargs.get(col.name)
                if _unique_key:
                    break
 
        _instance = cls.__static_cache.get(_unique_key)
        if _instance:
            return _instance
 
        _session = cls._meta.sqlalchemy_session
        with _session.no_autoflush:
            obj = model_class(*args, **kwargs)
            _session.add(obj)
            cls.__static_cache.update({_unique_key: obj})
            return obj
 
class LanguageFactory(StaticFactory):
   xxx