Support type hinting?

I can’t seem to have PyCharm play nice with the objects generated by Factory Boy; it would be nice to either automatically support type hinting somehow (ideally, it would simply be the model defined as a base for the factory) or at least add some documentation to explain how to have type checkers understand that, after p = PersonFactory(), variuable p is actually an instance of Person rather than of PersonFactory.

(Note: I’m not submitting a PR because I haven’t figured out a solution)

[kamilglod]

my workaround for missing General protocol is as follow:

T = TypeVar("T")


class BaseFactory(Generic[T], factory.Factory):
    @classmethod
    def create(cls, **kwargs) -> T:
        return super().create(**kwargs)


class RoleFactory(BaseFactory[Role]):
    class Meta:
        model = Role

    id = factory.Sequence(lambda n: n)
    name = factory.Sequence(lambda n: f"Role {n}")

one drawback of this solution is that you always need to use Factory.create() instead of Factory() as it’s not possible (at least I didn’t find a proper solution) to override __new__() method with TypeVar.

[michaeloliverx]

Typing support would be really nice!