为什么这个被弃用的Factory.set_creation_function的替代品不能与nosetests一起使用？

Question

Factory Boy已弃用set_creation_function（请参阅ChangeLog 2.6.1）并建议开发人员

  

替换factory.set_creation_function（SomeFactory，creation_function）   覆盖SomeFactory

的_create（）方法

我有i）许多衍生工厂类和ii）我的数据库会话在另一个模块中实例化，因此我尝试用https://github.com/mattupstate/overholt替换下面第二个代码块的工作示例。 PyCharm警告我没有使用“db”导入，所以我怀疑在设置sqlalchemy_session时它可能没有被正确解除引用？

使用nosetests 1.3.7（但现在不推荐使用FactoryBoy的set_creation_function）：

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)

不使用nosetests 2.x（看起来db没有被正确引用？）

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

示例无效代​​码的两个主要问题：

• 该类应来自SQLAlchemyModelFactory class
• _create()方法应定义为classmethod

修正版：

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

以下是覆盖_create()方法的模型工厂的sample。