工厂男孩覆盖属性

Question

我想将 FactoryBoy 与用户/配置文件模式一起使用，如其文档中所述。

class ProfileFactory(DjangoModelFactory):
    class Meta:
        model = Profile     
    first_name = lazy_attribute(lambda o: faker.first_name())
    user = SubFactory('profiles.tests.factories.UserFactory', profile=None)
class UserFactory(DjangoModelFactory):
    class Meta:
        model = settings.AUTH_USER_MODEL
    username = lazy_attribute(lambda o: faker.first_name())
    profile = RelatedFactory(ProfileFactory, 'user')

现在我希望能够通过仅提供用户名来创建用户，例如

u = UserFactory(username='alice')

并自动获取用户名作为个人资料中的first_name（='Alice' ）

或像这样的first_name

u = UserFactory(profile__first_name='Bob')

并获取根据给定first_name设置的用户名（='bob'

或者，如果我很活泼，请同时提供用户名和first_name当然保留两者

u = UserFactory(username='alice', profile__first_name='Bob')

到目前为止，我还没有找到实施此行为的正确位置。我尝试覆盖_generate()，但发现这不是正确的位置，因为我的懒惰属性此时已经启动了。此外，我无法覆盖__call__()或__new__()我的UserFactory。

也许这一切都是因为我对python相当陌生，所以我会感谢任何帮助，将我推向正确的方向。

Answer 1

尽管我在这里回答自己的第一个问题有点愚蠢，但我想我自己找到了答案：

看起来覆盖以实现我想要的东西的最佳方法是从BaseFactory attributes，因为所有构建/创建函数都调用它，我仍然得到"纯粹"kwargs作为extra摆弄它们。

也许有人仍然有更好的答案，但在此之前这是我的版本。

@classmethod
def attributes(cls, create=False, extra=None):
    if extra:
        if 'profile__first_name' in extra and 'username' not in extra:
            extra['username'] = extra['profile__first_name'].lower()
        elif 'username' in extra and 'profile__first_name' not in extra:
            extra['profile__first_name'] = extra['username'].title()
    return super().attributes(create, extra)