feat: expanded to other things

Signed-off-by: exploreriii <133720349+exploreriii@users.noreply.github.com>

Author: exploreriii

scripts/src_vs_proto/steps_6_token_burn.py

@@ -1,8 +1,7 @@
 #!/usr/bin/env python3
 """
-Step 6: Compare token transaction SDK classes (Step 3) with proto attributes (Step 4)
-         Accurate logging: actual proto attributes, predicted setters, SDK coverage,
-         and SDK-specific extra methods
+Step 6 (Fixed Dynamic): Compare *all* token transaction SDK classes with proto attributes.
+Now properly finds nested entries like token_burn_transaction.TokenBurnTransaction.
 """
 
 from pathlib import Path
@@ -13,6 +12,7 @@
 STEP3_FILE = Path(__file__).resolve().parent / "steps_3_token_classes_info_readable.py"
 STEP4_FILE = Path(__file__).resolve().parent / "steps_4_token_classes_proto_attributes.py"
 
+
 def load_module(file_path: Path, module_name: str):
     if not file_path.exists():
         raise FileNotFoundError(f"File not found: {file_path}")
@@ -22,39 +22,55 @@ def load_module(file_path: Path, module_name: str):
     spec.loader.exec_module(module)
     return module
 
+
 def camel_to_snake(name: str) -> str:
     s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
     s2 = re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1)
     return s2.lower()
 
+
 PROTO_TO_SDK_MAP = {
     "token": "token_id",
     "tokens": "token_ids",
     "serialNumbers": "serials",
     "account": "account_id",
-    "set_account": "set_account_id"
+    "set_account": "set_account_id",
 }
 
+
 def map_proto_to_sdk(proto_attr: str) -> str:
     return PROTO_TO_SDK_MAP.get(proto_attr, camel_to_snake(proto_attr))
 
+
 def predicted_proto_setters(proto_attrs):
-    """Return predicted setters from actual proto attributes"""
     return [f"set_{map_proto_to_sdk(a)}" for a in proto_attrs]
 
-SDK_MODULE_CLASS_MAP = {
-    "token_burn_transaction": "TokenBurnTransaction",
-    "token_associate_transaction": "TokenAssociateTransaction",
-    "token_dissociate_transaction": "TokenDissociateTransaction",
-    "token_mint_transaction": "TokenMintTransaction",
-}
 
-TRANSACTION_PROTO_MAP = {
-    "TokenBurnTransaction": "TokenBurnTransactionBody",
-    "TokenAssociateTransaction": "TokenAssociateTransactionBody",
-    "TokenDissociateTransaction": "TokenDissociateTransactionBody",
-    "TokenMintTransaction": "TokenMintTransactionBody",
-}
+def find_proto_class_name(sdk_class_name: str, proto_mappings: dict):
+    base = sdk_class_name.replace("Transaction", "TransactionBody")
+    if base in proto_mappings:
+        return base
+    for proto_name in proto_mappings.keys():
+        if proto_name.lower().startswith(sdk_class_name.lower().replace("transaction", "")):
+            return proto_name
+    return None
+
+
+def collect_sdk_classes(step3_module):
+    """
+    Recursively collect dict objects representing SDK classes
+    from the parsed Step 3 module.
+    """
+    sdk_classes = {}
+    for name, val in vars(step3_module).items():
+        if isinstance(val, dict) and "attributes" in val and "setters" in val:
+            sdk_classes[name] = val
+        elif isinstance(val, type(step3_module)):  # submodule-like object
+            for subname, subval in vars(val).items():
+                if isinstance(subval, dict) and "attributes" in subval:
+                    sdk_classes[f"{name}.{subname}"] = subval
+    return sdk_classes
+
 
 if __name__ == "__main__":
     step3_module = load_module(STEP3_FILE, "step3_token_classes")
@@ -65,33 +81,32 @@ def predicted_proto_setters(proto_attrs):
     for k in sorted(proto_mappings.keys()):
         print(" -", k)
 
-    for sdk_module_name, sdk_class_name in SDK_MODULE_CLASS_MAP.items():
-        sdk_module = getattr(step3_module, sdk_module_name, None)
-        sdk_class_info = getattr(sdk_module, sdk_class_name, None) if sdk_module else {'attributes': [], 'setters': [], 'other_methods': []}
-        sdk_class_info.setdefault('other_methods', [])
+    sdk_classes = collect_sdk_classes(step3_module)
+    print(f"\n🔍 Found {len(sdk_classes)} SDK token transaction classes to compare.\n")
 
-        proto_class_name = TRANSACTION_PROTO_MAP.get(sdk_class_name, "")
-        proto_info = proto_mappings.get(proto_class_name)
-        if proto_info is None:
-            print(f"\n⚠️ Proto mapping missing for {proto_class_name} (SDK {sdk_class_name})")
-            proto_info = {'attributes': [], 'setters': []}
+    for full_cls_name, sdk_class_info in sorted(sdk_classes.items()):
+        sdk_class_name = full_cls_name.split(".")[-1]
+        proto_class_name = find_proto_class_name(sdk_class_name, proto_mappings)
 
-        actual_proto_attrs = [map_proto_to_sdk(a) for a in proto_info['attributes']]
-        predicted_setters = predicted_proto_setters(proto_info['attributes'])
+        if not proto_class_name:
+            print(f"\n⚠️ No proto mapping found for SDK class {sdk_class_name}")
+            continue
 
-        sdk_methods = sdk_class_info['setters'] + sdk_class_info['other_methods']
+        proto_info = proto_mappings.get(proto_class_name, {"attributes": [], "setters": []})
+        actual_proto_attrs = [map_proto_to_sdk(a) for a in proto_info["attributes"]]
+        predicted_setters = predicted_proto_setters(proto_info["attributes"])
 
-        missing_attrs = [a for a in actual_proto_attrs if a not in sdk_class_info['attributes']]
-        missing_setters = [s for s in predicted_setters if s not in sdk_methods]
+        sdk_attrs = sdk_class_info.get("attributes", [])
+        sdk_methods = sdk_class_info.get("setters", []) + sdk_class_info.get("other_methods", [])
 
-        # Detect extra SDK helpers beyond predicted setters
+        missing_attrs = [a for a in actual_proto_attrs if a not in sdk_attrs]
+        missing_setters = [s for s in predicted_setters if s not in sdk_methods]
         extra_sdk_methods = [m for m in sdk_methods if m not in predicted_setters]
 
-        # Logging
         print(f"\n💠 {sdk_class_name} vs {proto_class_name}")
-        print("✅ SDK Attributes:", sdk_class_info['attributes'])
-        print("✅ SDK Setters:", sdk_class_info['setters'])
-        print("✅ SDK Other Methods:", sdk_class_info['other_methods'])
+        print("✅ SDK Attributes:", sdk_attrs)
+        print("✅ SDK Setters:", sdk_class_info.get("setters", []))
+        print("✅ SDK Other Methods:", sdk_class_info.get("other_methods", []))
         print("📦 Actual Proto Attributes:", actual_proto_attrs)
         print("📦 Predicted Proto Setters:", predicted_setters)