Skip to content

Commit defc6ef

Browse files
committed
feat: make colourful
Signed-off-by: exploreriii <133720349+exploreriii@users.noreply.github.com>
1 parent 9fef756 commit defc6ef

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

scripts/src_vs_proto/steps_6_token_burn.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,12 @@
1212
STEP3_FILE = Path(__file__).resolve().parent / "steps_3_token_classes_info_readable.py"
1313
STEP4_FILE = Path(__file__).resolve().parent / "steps_4_token_classes_proto_attributes.py"
1414

15+
RED = "\033[91m"
16+
GREEN = "\033[92m"
17+
YELLOW = "\033[93m"
18+
BLUE = "\033[94m"
19+
RESET = "\033[0m"
20+
1521

1622
def load_module(file_path: Path, module_name: str):
1723
if not file_path.exists():
@@ -104,20 +110,20 @@ def collect_sdk_classes(step3_module):
104110
extra_sdk_methods = [m for m in sdk_methods if m not in predicted_setters]
105111

106112
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}")
109115
print("✅ SDK Other Methods:", sdk_class_info.get("other_methods", []))
110116
print("📦 Actual Proto Attributes:", actual_proto_attrs)
111117
print("📦 Predicted Proto Setters:", predicted_setters)
112118

113119
if missing_attrs or missing_setters:
114-
print("⚠️ Missing in SDK:")
120+
print(f"{RED}⚠️ Missing in SDK:{RESET}")
115121
if missing_attrs:
116-
print(" - Attributes:", missing_attrs)
122+
print(f"{RED} - Attributes: {missing_attrs}{RESET}")
117123
if missing_setters:
118-
print(" - Predicted Setters / Methods:", missing_setters)
124+
print(f"{RED} - Predicted Setters / Methods: {missing_setters}{RESET}")
119125
else:
120-
print("✅ SDK fully covers proto attributes and predicted setters")
126+
print(f"{GREEN}✅ SDK fully covers proto attributes and predicted setters{RESET}")
121127

122128
if extra_sdk_methods:
123129
print("✨ Extra SDK methods beyond proto setters:", extra_sdk_methods)

0 commit comments

Comments
 (0)