Add support for Pydantic models in stubgen
#19095
Conversation
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
teplandr
changed the title
stubgen
|
Hello @JelleZijlstra! Hope you're doing well 🙂 |
|
Hello @hauntsaninja! Hope you're doing well! |
|
Hi @sobolevn! You're my last hope 🙂 |
test-data/unit/stubgen.test
Outdated
|
|
||
| class User(BaseUser): | ||
| name: str | ||
| email: str |
There was a problem hiding this comment.
| email: str | |
| email: str = '@' |
There was a problem hiding this comment.
There is an issue connected with this change. Currently, the code only sets processing_pydantic_model = True when the direct base class is pydantic.BaseModel, but it doesn't handle cases where a class inherits from another class that inherits from pydantic.BaseModel. I tried a few ideas but they aren't working. Could you please give me a clue how to tackle the problem?
| name: str | ||
| email: str | ||
|
|
||
| [case testPydanticModelWithMethods] |
There was a problem hiding this comment.
We can just add some methods to existing models. Let's reduce the amount of tests without reducing test features.
This PR adds support for generating stubs for Pydantic models in
stubgen.Changes
pydantic.BaseModeldirectly or indirectlyFixes #16968