yadunut/clan-core
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Classgen: export field type definitions

Browse Source
This commit is contained in:
Johannes Kirschbauer
2025-05-20 13:31:47 +02:00
parent 7f62a7993a
commit 30e74c408c
1 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
pkgs/classgen/main.py
View File

@@ -65,8 +65,8 @@ def field_def_from_default_type(
field_name: str, field_name: str,
field_types: set[str], field_types: set[str],
class_name: str, class_name: str,
finalize_field: Callable[..., str], finalize_field: Callable[..., tuple[str, str]],
) -> str | None: ) -> tuple[str, str] | None:
if "dict" in str(field_types): if "dict" in str(field_types):
return finalize_field( return finalize_field(
field_types=field_types, field_types=field_types,
@@ -127,8 +127,8 @@ def field_def_from_default_value(
field_name: str, field_name: str,
field_types: set[str], field_types: set[str],
nested_class_name: str, nested_class_name: str,
finalize_field: Callable[..., str], finalize_field: Callable[..., tuple[str, str]],
) -> str | None: ) -> tuple[str, str] | None:
# default_value = prop_info.get("default") # default_value = prop_info.get("default")
if default_value is None: if default_value is None:
return finalize_field( return finalize_field(
@@ -184,7 +184,7 @@ def get_field_def(
default: str | None = None, default: str | None = None,
default_factory: str | None = None, default_factory: str | None = None,
type_appendix: str = "", type_appendix: str = "",
) -> str: ) -> tuple[str, str]:
if "None" in field_types or default or default_factory: if "None" in field_types or default or default_factory:
if "None" in field_types: if "None" in field_types:
field_types.remove("None") field_types.remove("None")
@@ -193,7 +193,7 @@ def get_field_def(
else: else:
serialised_types = " | ".join(field_types) + type_appendix serialised_types = " | ".join(field_types) + type_appendix
return f"{field_name}: {serialised_types}" return (field_name, serialised_types)
# Recursive function to generate dataclasses from JSON schema # Recursive function to generate dataclasses from JSON schema
@@ -204,8 +204,8 @@ def generate_dataclass(
) -> str: ) -> str:
properties = schema.get("properties", {}) properties = schema.get("properties", {})
required_fields = [] required_fields: list[tuple[str, str]] = []
fields_with_default = [] fields_with_default: list[tuple[str, str]] = []
nested_classes: list[str] = [] nested_classes: list[str] = []
# if We are at the top level, and the attribute name is in shallow # if We are at the top level, and the attribute name is in shallow
@@ -218,7 +218,7 @@ def generate_dataclass(
field_name = prop.replace("-", "_") field_name = prop.replace("-", "_")
if len(attr_path) == 0 and prop not in attrs: if len(attr_path) == 0 and prop not in attrs:
field_def = f"{field_name}: NotRequired[dict[str, Any]]" field_def = field_name, "NotRequired[dict[str, Any]]"
fields_with_default.append(field_def) fields_with_default.append(field_def)
# breakpoint() # breakpoint()
continue continue
@@ -345,9 +345,18 @@ def generate_dataclass(
) )
required_fields.append(field_def) required_fields.append(field_def)
# breakpoint() # Join field name with type to form a complete field declaration
# e.g. "name: str"
fields_str = "\n ".join(required_fields + fields_with_default) all_field_declarations = [
f"{n}: {t}" for n, t in (required_fields + fields_with_default)
]
hoisted_types: str = "\n".join(
[
f"{class_name}{n.capitalize()}Type = {x}"
for n, x in (required_fields + fields_with_default)
]
)
fields_str = "\n ".join(all_field_declarations)
nested_classes_str = "\n\n".join(nested_classes) nested_classes_str = "\n\n".join(nested_classes)
class_def = f"\nclass {class_name}(TypedDict):\n" class_def = f"\nclass {class_name}(TypedDict):\n"
@@ -356,6 +365,8 @@ def generate_dataclass(
else: else:
class_def += f" {fields_str}" class_def += f" {fields_str}"
class_def += f"\n\n{hoisted_types}\n"
return f"{nested_classes_str}\n\n{class_def}" if nested_classes_str else class_def return f"{nested_classes_str}\n\n{class_def}" if nested_classes_str else class_def