python - one - ¿Por qué esta alternativa a la Factory.set_creation_function obsoleta no funciona con nosetest?

pagination in flask sqlalchemy

Question

Factory Boy desaprobó set_creation_function (ver ChangeLog 2.6.1 ) y recomienda que los desarrolladores

Reemplazar factory.set_creation_function (SomeFactory, creation_function) con una anulación del método _create () de SomeFactory

Tengo i) varias clases de fábrica de derivados y ii) mi sesión de db instanciada en otro módulo, así que traté de reemplazar el ejemplo de trabajo de https://github.com/mattupstate/overholt con el segundo bloque de código a continuación. PyCharm me está advirtiendo que la importación "db" no se está utilizando, así que sospecho que no se está desreferenciando correctamente cuando configuro sqlalchemy_session ?

Trabajar con nosetests 1.3.7 (pero la set_creation_function de FactoryBoy ahora está en desuso):

from myapp.core import db
def create_sqlalchemy_model_function(class_to_create, *args, **kwargs):
entity = class_to_create(**kwargs)
db.session.add(entity)
db.session.commit()
return entity
Factory.set_creation_function(create_sqlalchemy_model_function)

No funciona con nosetests 2.x (parece que db no está siendo referenciado correctamente?)

from factory.alchemy import SQLAlchemyModelFactory as Factory
from myapp.core import db
class Factory():
class Meta:
sqlalchemy_session = db.session
def _create(cls, model_class, *args, **kwargs):
entity = model_class(*args, **kwargs)
db.session.add(entity)
db.session.commit()
return entity

Answer 1

Dos problemas principales con su código no funcional de muestra:

• la clase debe derivarse de la clase SQLAlchemyModelFactory
• el método _create() debe definirse como classmethod

Versión arreglada:

from factory.alchemy import SQLAlchemyModelFactory as Factory
from myapp.core import db
class MyFactory(Factory):
class Meta:
sqlalchemy_session = db.session
@classmethod
def _create(cls, model_class, *args, **kwargs):
entity = model_class(*args, **kwargs)
db.session.add(entity)
db.session.commit()
return entity

Aquí también hay una muestra de la fábrica de modelos que _create() método _create() .