Fix all-export generation for hyphenated directories (#3033)

Fixes a bug where --all-exports-scope fails to generate __all__ imports
for packages originating from schema directories with hyphens in the
path.

Co-authored-by: Koudai Aono <koxudaxi@gmail.com>

Author: an5t

src/datamodel_code_generator/parser/base.py

@@ -2592,14 +2592,16 @@ def _collect_exports_for_init(
     ) -> list[tuple[str, tuple[str, ...], str]]:
         """Collect exports for __init__.py based on scope."""
         exports: list[tuple[str, tuple[str, ...], str]] = []
-        base = module[:-1] if module[-1] == "__init__.py" else module
+        normalized_module = tuple(part.replace("-", "_") for part in module)
+        base = normalized_module[:-1] if normalized_module[-1] == "__init__.py" else normalized_module
         base_len = len(base)
 
         for proc_module, _, proc_models, _, _, _ in processed_models:
-            if not proc_models or proc_module == module:
+            normalized_proc_module = tuple(part.replace("-", "_") for part in proc_module)
+            if not proc_models or normalized_proc_module == normalized_module:
                 continue
-            last = proc_module[-1]
-            prefix = proc_module[:-1] if last == "__init__.py" else (*proc_module[:-1], last[:-3])
+            last = normalized_proc_module[-1]
+            prefix = normalized_proc_module[:-1] if last == "__init__.py" else (*normalized_proc_module[:-1], last[:-3])
             if prefix[:base_len] != base or (depth := len(prefix) - base_len) < 1:
                 continue
             if scope == AllExportsScope.Children and depth != 1:

tests/data/expected/main/jsonschema/all_exports_hyphenated_directory/hyphenated_directory/__init__.py

@@ -0,0 +1,10 @@
+# generated by datamodel-codegen:
+#   filename:  all_exports_hyphenated_directory
+
+from __future__ import annotations
+
+from .order import Order
+
+__all__ = [
+    "Order",
+]

tests/data/expected/main/jsonschema/all_exports_hyphenated_directory/hyphenated_directory/order.py

@@ -0,0 +1,10 @@
+# generated by datamodel-codegen:
+#   filename:  hyphenated-directory/order.json
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class Order(BaseModel):
+    id: str

tests/data/jsonschema/all_exports_hyphenated_directory/hyphenated-directory/order.json

@@ -0,0 +1,11 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "type": "object",
+  "title": "Order",
+  "properties": {
+    "id": {
+      "type": "string"
+    }
+  },
+  "required": ["id"]
+}

tests/main/test_main_general.py

@@ -1288,6 +1288,21 @@ def test_all_exports_scope_children_with_local_models(output_dir: Path) -> None:
     )
 
 
+def test_all_exports_scope_children_jsonschema_hyphenated_package(output_dir: Path) -> None:
+    """Test --all-exports-scope=children with hyphenated JSON Schema directories."""
+    run_main_and_assert(
+        input_path=JSON_SCHEMA_DATA_PATH / "all_exports_hyphenated_directory",
+        output_path=output_dir,
+        input_file_type="jsonschema",
+        extra_args=[
+            "--disable-timestamp",
+            "--all-exports-scope",
+            "children",
+        ],
+        expected_directory=EXPECTED_MAIN_PATH / "jsonschema" / "all_exports_hyphenated_directory",
+    )
+
+
 def test_check_respects_pyproject_toml_settings(tmp_path: Path) -> None:
     """Test --check uses pyproject.toml formatter settings from output path.