模型和工厂的真实来源之一

如何解决模型和工厂的真实来源之一

在键入提示 API 的 JSON 响应时，我发现自己编写了很多相同的代码，例如：

class Foo(TypedDict):
    foo: int
    baz: str

来自 json 响应：

{
   'foo': 1,'baz': 'somestr'
}

我现在想写一些工厂，这样我就可以模拟和测试这个端点。所以我必须写一个非常相似的：

class FooFactory(factory.DictFactory):
    foo = Faker("random_int")
    baz = Faker("text")

要同时维护它们是非常痛苦的。有没有办法让工厂使用 Foo 作为“灵感” - 即类型提示是 str 所以使用 Faker("text")，类型提示是 int所以使用 Faker("random_int") 等... ?

我试过了：

class FooFactory(factory.DictFactory):
    class Meta:
       model = Foo

但这似乎并没有什么值得注意的 - 因为它只是在我的测试中返回 {}。

---

此外，我不限于 TypedDicts 或 factoryboy。我只是不想写同样的东西两次:)

解决方法

这是 factory_boy 的计划功能，请参阅 https://github.com/FactoryBoy/factory_boy/issues/836。

该问题指向了可以通过子类化 Factory 类在您的代码中重用的草案实现。