fix llvm license for .py

Author: xurui1995

scripts/license.py

@@ -45,19 +45,32 @@
     "===-*===",
 ]
 
+llvm_license_py: list[str] = [
+    "#  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.",
+    "#  See https://llvm.org/LICENSE.txt for license information.",
+    "#  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception"
+]
+
+
 def check_license(filepath: str, license: list[str], var: Dict[str, str], re_line: Set[int]):
     with open(filepath, 'r') as f:
         idx: int = 0
         for line in f.readlines():
             lic: str = license[idx]
-            # replace the variable defined in license
-            for k, v in var.items():
-                lic = lic.replace(k, v)
-            if idx in re_line:
-                if re.search(lic, line) is not None and ("RE_WIDTH" not in var or int(var["RE_WIDTH"]) + 1 == len(line)):
-                    idx += 1
-            elif line.find(lic) != -1:
+            # check the string directly
+            if license == llvm_license_py:
+                if lic != line[:-1]:
+                    return False
                 idx += 1
+            else:
+                # replace the variable defined in license
+                for k, v in var.items():
+                    lic = lic.replace(k, v)
+                if idx in re_line:
+                    if re.search(lic, line) is not None and ("RE_WIDTH" not in var or int(var["RE_WIDTH"]) + 1 == len(line)):
+                        idx += 1
+                elif line.find(lic) != -1:
+                    idx += 1
             if idx == len(license):
                 return True
         return False
@@ -82,22 +95,24 @@ def fix_intel_license(var: Dict[str, str]):
 def fix_llvm_license(var: Dict[str, str]):
     lang: str = var['$LANG']
     cmt: str = "//" # comment char
-    if lang == "Python":
-        cmt = "#"
-    elif lang == "C\\+\\+":
+    if lang == "C\\+\\+":
         lang = "C++"
+        
+    if lang == "Python":
+        for i in range(len(llvm_license_py)):
+            print((llvm_license_py[i]))
+    else:
+        part1 = "%s===-- %s - DESC " % (cmt, var['$FILE'])
+        part3 = "-*- %s -*-===%s" % (lang, cmt)
+        part2 = "-" * (WIDTH - len(part1) - len(part3))
 
-    part1 = "%s===-- %s - DESC " % (cmt, var['$FILE'])
-    part3 = "-*- %s -*-===%s" % (lang, cmt)
-    part2 = "-" * (WIDTH - len(part1) - len(part3))
-
-    print(part1 + part2 + part3)
-    for i in range(1, len(llvm_license) - 1):
-        print((cmt + " " + llvm_license[i]).rstrip())
-    part1 = cmt + "==="
-    part3 = "===" + cmt
-    part2 = "-" * (WIDTH - len(part1) - len(part3))
-    print(part1 + part2 + part3)
+        print(part1 + part2 + part3)
+        for i in range(1, len(llvm_license) - 1):
+            print((cmt + " " + llvm_license[i]).rstrip())
+        part1 = cmt + "==="
+        part3 = "===" + cmt
+        part2 = "-" * (WIDTH - len(part1) - len(part3))
+        print(part1 + part2 + part3)
 
 def use_llvm_license(path: str) -> bool:
     for folder in ["lib/gc/", 'include/gc/', 'unittests/', 'python/gc_mlir']:
@@ -131,15 +146,18 @@ def use_llvm_license(path: str) -> bool:
 
     is_llvm_license = use_llvm_license(filepath)
     if is_llvm_license:
-        # llvm license, only check python/cpp now
-        lic = llvm_license
-        re_line.add(0)
-        re_line.add(6)
-        var['$FILE'] = name
-        # the line we read contains a '\n' character, so the expected length should be 81
-        var['RE_WIDTH'] = str(WIDTH)
-        if name.endswith(".td"):
-            var['$LANG'] = "tablegen"
+        # llvm license, only check python/cpp now   
+        if name.endswith(".py"):
+            lic = llvm_license_py
+        else:
+            lic = llvm_license
+            re_line.add(0)
+            re_line.add(6)
+            var['$FILE'] = name
+            # the line we read contains a '\n' character, so the expected length should be 81
+            var['RE_WIDTH'] = str(WIDTH)
+            if name.endswith(".td"):
+                var['$LANG'] = "tablegen"
     else:
         # intel license
         lic = intel_license