fix: resolve bare OSW ID type hints in generated Python code

When child schemas override inherited properties with only options
(e.g. {"options": {"hidden": true}}), datamodel-code-generator uses
the JSON schema file stem (OSW ID) instead of the schema title as the
type name. This adds a post-processing step that builds a UUID-to-class
mapping from generated class definitions and replaces bare OSW IDs.

Author: simontaurus

src/osw/core.py

@@ -39,7 +39,10 @@
 
 import osw.model.entity as model
 from osw.defaults import params as default_params
-from osw.utils.code_postprocessing import remove_constraints_from_forward_refs
+from osw.utils.code_postprocessing import (
+    remove_constraints_from_forward_refs,
+    resolve_osw_id_type_hints,
+)
 from osw.utils.oold import (
     AggregateGeneratedSchemasParam,
     AggregateGeneratedSchemasParamMode,
@@ -908,6 +911,11 @@ def _fetch_schema(
                 content = all_content
 
             if fetchSchemaParam.final:
+                # Resolve bare OSW ID type hints (e.g. OSW3886...)
+                # with actual class names (e.g. RiskAssessmentProcess)
+                # using UUID annotations from generated class definitions
+                content = resolve_osw_id_type_hints(content)
+
                 # Cleanup the combined content
                 # find all "<cls>.update_forward_refs()" lines,
                 # remove duplicates and put them to EOF

src/osw/utils/code_postprocessing.py

@@ -44,3 +44,61 @@ def replace_field(match):
         return match.group(0)
 
     return re.sub(pattern, replace_field, src_code)
+
+
+def resolve_osw_id_type_hints(content: str) -> str:
+    """Replace bare OSW ID type hints with their actual class names.
+
+    When datamodel-code-generator processes child schemas that override inherited
+    properties with only options (e.g. {"options": {"hidden": true}}), it may use
+    the JSON schema file stem (an OSW ID like OSW3886740859ae459588fee73d3bb3c83e)
+    instead of the schema title (e.g. RiskAssessmentProcess) as the type name.
+
+    This function builds a mapping from OSW IDs to class names by extracting UUIDs
+    from class Config.schema_extra annotations in the generated code, then replaces
+    any bare OSW ID references with the resolved class name.
+
+    Parameters
+    ----------
+    content : str
+        The generated Python source code (may contain multiple classes).
+
+    Returns
+    -------
+    str
+        The source code with bare OSW ID type hints replaced by class names.
+    """
+    # Build mapping: OSW ID -> class name from schema_extra uuid annotations
+    # Each generated class has: class Foo(...): class Config: schema_extra = {"uuid": "..."}
+    class_starts = list(re.finditer(r"^class\s+(\w+)\s*\(", content, re.MULTILINE))
+    osw_id_to_class = {}
+    for i, match in enumerate(class_starts):
+        class_name = match.group(1)
+        start = match.start()
+        end = class_starts[i + 1].start() if i + 1 < len(class_starts) else len(content)
+        class_block = content[start:end]
+        uuid_match = re.search(
+            r'"uuid":\s*"([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}'
+            r"-[0-9a-f]{4}-[0-9a-f]{12})" + '"',
+            class_block,
+        )
+        if uuid_match:
+            osw_id = "OSW" + uuid_match.group(1).replace("-", "")
+            osw_id_to_class[osw_id] = class_name
+
+    if not osw_id_to_class:
+        return content
+
+    # Find bare OSW IDs used as identifiers (not inside "Category:OSW..." strings)
+    bare_osw_pattern = re.compile(r"(?<![:\w])OSW[0-9a-f]{32}(?!\w)")
+    unresolved = set(bare_osw_pattern.findall(content))
+
+    for osw_id in unresolved:
+        if osw_id in osw_id_to_class:
+            content = re.sub(
+                r"(?<![:\w])" + re.escape(osw_id) + r"(?!\w)",
+                osw_id_to_class[osw_id],
+                content,
+            )
+
+    return content