- Fixed GoConfig/PythonConfig __dict__ error with slots=True dataclasses

Author: MS-32154

json_explorer/codegen/languages/go/config.py

@@ -65,27 +65,41 @@ class GoConfig:
     use_pointers_for_optional: bool = True
 
     # Declaration
-    type_map: dict = field(init=False)
+    type_map: dict = field(init=False, repr=False, default_factory=dict)
 
     def __post_init__(self) -> None:
         """Build type map with configured types."""
-        self.type_map = GO_TYPE_MAP.copy()
-        self.type_map[FieldType.INTEGER] = self.int_type
-        self.type_map[FieldType.FLOAT] = self.float_type
-        self.type_map[FieldType.STRING] = self.string_type
-        self.type_map[FieldType.BOOLEAN] = self.bool_type
-        self.type_map[FieldType.TIMESTAMP] = self.time_type
-        self.type_map[FieldType.UNKNOWN] = self.unknown_type
-        self.type_map[FieldType.CONFLICT] = self.unknown_type
+        type_map = GO_TYPE_MAP.copy()
+        type_map[FieldType.INTEGER] = self.int_type
+        type_map[FieldType.FLOAT] = self.float_type
+        type_map[FieldType.STRING] = self.string_type
+        type_map[FieldType.BOOLEAN] = self.bool_type
+        type_map[FieldType.TIMESTAMP] = self.time_type
+        type_map[FieldType.UNKNOWN] = self.unknown_type
+        type_map[FieldType.CONFLICT] = self.unknown_type
+
+        object.__setattr__(self, "type_map", type_map)
 
         logger.debug(
             f"GoConfig initialized: int={self.int_type}, "
             f"float={self.float_type}, pointers={self.use_pointers_for_optional}"
         )
 
+    def to_dict(self) -> dict:
+        """Convert to dict, excluding computed fields."""
+        return {
+            "int_type": self.int_type,
+            "float_type": self.float_type,
+            "string_type": self.string_type,
+            "bool_type": self.bool_type,
+            "time_type": self.time_type,
+            "unknown_type": self.unknown_type,
+            "use_pointers_for_optional": self.use_pointers_for_optional,
+        }
+
     def get_go_type(
         self,
-        field_type: FieldType,
+        field_type: FieldType | None,
         *,
         is_optional: bool = False,
         is_array: bool = False,
@@ -104,10 +118,18 @@ def get_go_type(
             Go type string (e.g., "string", "*int64", "[]User")
         """
         if is_array:
-            base = element_type or self.type_map.get(field_type, self.unknown_type)
+            base = (
+                element_type or self.type_map.get(field_type, self.unknown_type)
+                if field_type
+                else self.unknown_type
+            )
             return f"[]{base}"
 
-        go_type = self.type_map.get(field_type, self.unknown_type)
+        # Handle None field_type
+        if field_type is None:
+            go_type = self.unknown_type
+        else:
+            go_type = self.type_map.get(field_type, self.unknown_type)
 
         # Add pointer for optional fields if configured
         if is_optional and self.use_pointers_for_optional:

json_explorer/codegen/languages/go/generator.py

@@ -314,9 +314,8 @@ def create_web_api_generator() -> GoGenerator:
         generate_json_tags=True,
         json_tag_omitempty=True,
         add_comments=True,
-        language_config=get_web_api_config().__dict__,
+        language_config=get_web_api_config().to_dict(),
     )
-
     return GoGenerator(config)
 
 
@@ -329,7 +328,6 @@ def create_strict_generator() -> GoGenerator:
         generate_json_tags=True,
         json_tag_omitempty=False,
         add_comments=True,
-        language_config=get_strict_config().__dict__,
+        language_config=get_strict_config().to_dict(),
     )
-
     return GoGenerator(config)

json_explorer/codegen/languages/python/config.py

@@ -123,29 +123,54 @@ class PythonConfig:
     typeddict_total: bool = False
 
     # Declaration
-    type_map: dict = field(init=False)
+    type_map: dict = field(init=False, repr=False, default_factory=dict)
 
     def __post_init__(self) -> None:
         """Initialize type map and validate style."""
         # Convert string style to enum if needed
         if isinstance(self.style, str):
-            self.style = PythonStyle(self.style)
+            object.__setattr__(self, "style", PythonStyle(self.style))
 
         # Build type map
-        self.type_map = PYTHON_TYPE_MAP.copy()
-        self.type_map[FieldType.INTEGER] = self.int_type
-        self.type_map[FieldType.FLOAT] = self.float_type
-        self.type_map[FieldType.STRING] = self.string_type
-        self.type_map[FieldType.BOOLEAN] = self.bool_type
-        self.type_map[FieldType.TIMESTAMP] = self.datetime_type
-        self.type_map[FieldType.UNKNOWN] = self.unknown_type
-        self.type_map[FieldType.CONFLICT] = self.unknown_type
+        type_map = PYTHON_TYPE_MAP.copy()
+        type_map[FieldType.INTEGER] = self.int_type
+        type_map[FieldType.FLOAT] = self.float_type
+        type_map[FieldType.STRING] = self.string_type
+        type_map[FieldType.BOOLEAN] = self.bool_type
+        type_map[FieldType.TIMESTAMP] = self.datetime_type
+        type_map[FieldType.UNKNOWN] = self.unknown_type
+        type_map[FieldType.CONFLICT] = self.unknown_type
+
+        object.__setattr__(self, "type_map", type_map)
 
         logger.debug(
             f"PythonConfig initialized: style={self.style.value}, "
             f"optional={self.use_optional}"
         )
 
+    def to_dict(self) -> dict:
+        """Convert to dict, excluding computed fields."""
+        return {
+            "style": (
+                self.style.value if isinstance(self.style, PythonStyle) else self.style
+            ),
+            "int_type": self.int_type,
+            "float_type": self.float_type,
+            "string_type": self.string_type,
+            "bool_type": self.bool_type,
+            "datetime_type": self.datetime_type,
+            "unknown_type": self.unknown_type,
+            "use_optional": self.use_optional,
+            "dataclass_frozen": self.dataclass_frozen,
+            "dataclass_slots": self.dataclass_slots,
+            "dataclass_kw_only": self.dataclass_kw_only,
+            "pydantic_use_field": self.pydantic_use_field,
+            "pydantic_use_alias": self.pydantic_use_alias,
+            "pydantic_config_dict": self.pydantic_config_dict,
+            "pydantic_extra_forbid": self.pydantic_extra_forbid,
+            "typeddict_total": self.typeddict_total,
+        }
+
     def get_python_type(
         self,
         field_type: FieldType,
@@ -244,6 +269,22 @@ def _extract_base_types(self, type_string: str) -> set[str]:
 # ============================================================================
 # Preset Configurations
 # ============================================================================
+def get_python_generator_config(**overrides) -> dict:
+    """
+    Get GeneratorConfig dict with Python-friendly defaults.
+
+    Python conventions:
+    - Classes: PascalCase (struct_case="pascal")
+    - Fields: snake_case (field_case="snake")
+    """
+    defaults = {
+        "package_name": "models",
+        "struct_case": "pascal",
+        "field_case": "snake",  # Python convention
+        "add_comments": True,
+    }
+    defaults.update(overrides)
+    return defaults
 
 
 def get_dataclass_config() -> PythonConfig:

json_explorer/codegen/languages/python/generator.py

@@ -11,7 +11,7 @@
 from ...core.config import GeneratorConfig
 from ...core.generator import CodeGenerator
 from ...core.schema import Schema, Field, FieldType
-from .config import PythonConfig, PythonStyle
+from .config import PythonConfig, PythonStyle, get_python_generator_config
 from .naming import create_python_name_tracker
 
 logger = logging.getLogger(__name__)
@@ -391,9 +391,8 @@ def create_dataclass_generator() -> PythonGenerator:
     from .config import get_dataclass_config
 
     config = GeneratorConfig(
-        package_name="models",
-        add_comments=True,
-        language_config=get_dataclass_config().__dict__,
+        **get_python_generator_config(),
+        language_config=get_dataclass_config().to_dict(),
     )
 
     return PythonGenerator(config)
@@ -404,9 +403,8 @@ def create_pydantic_generator() -> PythonGenerator:
     from .config import get_pydantic_config
 
     config = GeneratorConfig(
-        package_name="models",
-        add_comments=True,
-        language_config=get_pydantic_config().__dict__,
+        **get_python_generator_config(),
+        language_config=get_pydantic_config().to_dict(),
     )
 
     return PythonGenerator(config)
@@ -417,9 +415,8 @@ def create_typeddict_generator() -> PythonGenerator:
     from .config import get_typeddict_config
 
     config = GeneratorConfig(
-        package_name="types",
-        add_comments=True,
-        language_config=get_typeddict_config().__dict__,
+        **get_python_generator_config(),
+        language_config=get_typeddict_config().to_dict(),
     )
 
     return PythonGenerator(config)