Allow custom base class for models about datamodel-code-generator HOT 3 CLOSED

@MrMrRobat
support and publish the feature. Thank you

Example

$ datamodel-codegen --input ./tests/data/api.yaml --base-class custom_module.CustomBase
Output:

class Pet(CustomBase):
    id: int
    name: str
    tag: Optional[str] = None
...

from datamodel-code-generator.

@MrMrRobat
Sounds great to me.

This can be approached by renaming BaseModel in a result manually, but if it not hard to implement I'd preffered to have this built-in

Did you expect to use inherited BasdModel?

I think the code generator should able to receive the alternative base class as arguments or a config file.

from datamodel-code-generator.

Looks great, thank you!

from datamodel-code-generator.