Support NaiveDateTime about datamodel-code-generator HOT 1 OPEN

I know this was mentioned in the original PR #1735 and also brought up in #1944 (but closed), I have a similar need. I can't modify the source system to properly generate timezone strict output (all datetimes are already generated in UTC, but don't serialize to JSON that keeps the AwareDatetime happy). Datetimes that are in ISO 8601 format like "2024-05-29T18:00:29.526990Z" are fine with AwareDatetime at least. But I can't guarantee.

As a workaround to consume the JSON, while I dislike it, I currently have a script that calls datamodel_code_generator.codegen.generate programmatically instead of via direct CLI, so I can monkeypatch the datamodel_code_generator.model.pydantic_v2.types.IMPORT_AWARE_DATETIME value before generating like:

    # Patch to use naive datetime instead of default AwareDatetime
    from datamodel_code_generator.imports import Import
    from datamodel_code_generator.model.pydantic_v2 import types as pydantic_v2_types
    pydantic_v2_types.IMPORT_AWARE_DATETIME = Import.from_full_path('pydantic.NaiveDatetime')

    codegen.generate(
        ...
    )

Then my outputs properly use NaiveDatetime like I want.

Similarly, if I wanted to use the original standard datetime.datetime, just use Import.from_full_path('datetime.datetime')

Alternatively, I could have patched/overridden the pydantic_v2.types.DataTypeManager class and its type_map_factory method, but that seemed more complicated to subclass and inject into the generate/parser.

Obviously long-term, it would be great to have an official way to customize.

from datamodel-code-generator.