|
12 | 12 | STEP3_FILE = Path(__file__).resolve().parent / "steps_3_token_classes_info_readable.py" |
13 | 13 | STEP4_FILE = Path(__file__).resolve().parent / "steps_4_token_classes_proto_attributes.py" |
14 | 14 |
|
| 15 | +RED = "\033[91m" |
| 16 | +GREEN = "\033[92m" |
| 17 | +YELLOW = "\033[93m" |
| 18 | +BLUE = "\033[94m" |
| 19 | +RESET = "\033[0m" |
| 20 | + |
15 | 21 |
|
16 | 22 | def load_module(file_path: Path, module_name: str): |
17 | 23 | if not file_path.exists(): |
@@ -104,20 +110,20 @@ def collect_sdk_classes(step3_module): |
104 | 110 | extra_sdk_methods = [m for m in sdk_methods if m not in predicted_setters] |
105 | 111 |
|
106 | 112 | print(f"\n💠 {sdk_class_name} vs {proto_class_name}") |
107 | | - print("✅ SDK Attributes:", sdk_attrs) |
108 | | - print("✅ SDK Setters:", sdk_class_info.get("setters", [])) |
| 113 | + print(f"{GREEN}SDK Attributes: {sdk_attrs}{RESET}") |
| 114 | + print(f"{GREEN}SDK Setters: {sdk_class_info.get('setters', [])}{RESET}") |
109 | 115 | print("✅ SDK Other Methods:", sdk_class_info.get("other_methods", [])) |
110 | 116 | print("📦 Actual Proto Attributes:", actual_proto_attrs) |
111 | 117 | print("📦 Predicted Proto Setters:", predicted_setters) |
112 | 118 |
|
113 | 119 | if missing_attrs or missing_setters: |
114 | | - print("⚠️ Missing in SDK:") |
| 120 | + print(f"{RED}⚠️ Missing in SDK:{RESET}") |
115 | 121 | if missing_attrs: |
116 | | - print(" - Attributes:", missing_attrs) |
| 122 | + print(f"{RED} - Attributes: {missing_attrs}{RESET}") |
117 | 123 | if missing_setters: |
118 | | - print(" - Predicted Setters / Methods:", missing_setters) |
| 124 | + print(f"{RED} - Predicted Setters / Methods: {missing_setters}{RESET}") |
119 | 125 | else: |
120 | | - print("✅ SDK fully covers proto attributes and predicted setters") |
| 126 | + print(f"{GREEN}✅ SDK fully covers proto attributes and predicted setters{RESET}") |
121 | 127 |
|
122 | 128 | if extra_sdk_methods: |
123 | 129 | print("✨ Extra SDK methods beyond proto setters:", extra_sdk_methods) |
0 commit comments