Print diff of __init__.pyi

pyteal/__init__.pyi

@@ -23,6 +23,7 @@ __all__ = [
     "Bytes",
     "Int",
     "EnumInt",
+    "MethodSignature",
     "Arg",
     "TxnType",
     "TxnField",

scripts/generate_init.py

@@ -1,4 +1,4 @@
-import argparse, os, sys
+import argparse, os, sys, difflib
 from pyteal import __all__ as static_all
 
 
@@ -30,7 +30,7 @@
 init_file = os.path.join(orig_dir, py_file)
 
 
-def generate_tmp():
+def generate_init_pyi() -> str:
     with open(init_file, "r") as f:
         init_contents = f.read()
 
@@ -47,19 +47,29 @@ def generate_tmp():
     )
 
 
-def is_different(regen):
+def is_different(regen: str) -> bool:
     if not os.path.exists(orig_file):
         return True
 
     with open(orig_file, "r") as f:
         orig_lines = f.readlines()
 
-    curr_lines = regen.split("\n")
+    curr_lines = regen.splitlines(keepends=True)
 
-    return orig_lines == curr_lines
+    diff = list(
+        difflib.unified_diff(
+            orig_lines, curr_lines, fromfile="original", tofile="generated", n=3
+        )
+    )
+
+    if len(diff) != 0:
+        print("".join(diff), end="")
+        return True
+
+    return False
 
 
-def overwrite(regen):
+def overwrite(regen: str):
     with open(orig_file, "w") as f:
         f.write(regen)
 
@@ -74,7 +84,7 @@ def overwrite(regen):
     )
     args = parser.parse_args()
 
-    regen = generate_tmp()
+    regen = generate_init_pyi()
 
     if args.check:
         if is_different(regen):