on -j define __jobs instead of _smp_mflags

[packages/rpm-build-tools.git] / check-shebang.py

diff --git a/check-shebang.py b/check-shebang.py
index fe04522e6beb257258c4a1a7aae4074ef4f17e21..90075bc28f7b155bc0ef357609d2d608c225d200 100755 (executable)
--- a/check-shebang.py
+++ b/check-shebang.py
@@ -1,8 +1,9 @@
 #!/usr/bin/python3
 
-# thisscript.py --root=~/tmp/somepackage ~/rpm/BUILD/somepackage/
+# thisscript.py --buildroot=~/tmp/somepackage ~/rpm/BUILD/somepackage/
 
 import argparse
+import hashlib
 import io
 import os
 import re
@@ -17,19 +18,32 @@ rep = {
     'python2': [],
     'python3': [],
     'perl': [],
+    'ruby': [],
+    'bash': [],
 }
 
 skip_files = [".h", ".c", ".cc", ".gif", ".png", ".jpg", ".ko", ".gz", ".o"]
 
+def hash(fname):
+    hash_alg = hashlib.sha256()
+    with open(fname, "rb") as f:
+        for chunk in iter(lambda: f.read(4096), b""):
+            hash_alg.update(chunk)
+    return hash_alg.hexdigest()
+
+rpm_build_root_files = []
 if args.buildroot:
     print("%s: Caching `%s' files..." % (sys.argv[0], args.buildroot))
-    rpm_build_root_files = []
     for root, dirs, files in os.walk(args.buildroot):
         for name in files:
             fname, fext = os.path.splitext(name)
             if fext in skip_files:
                 continue
-            rpm_build_root_files.append(fname)
+            fpath = os.path.join(root, name)
+            try:
+                rpm_build_root_files.append(hash(fpath))
+            except FileNotFoundError:
+                pass
     print("%s: Caching done." % (sys.argv[0]))
 
 for root, dirs, files in os.walk(args.sourcedir):
@@ -39,47 +53,52 @@ for root, dirs, files in os.walk(args.sourcedir):
             continue
 
         fpath = os.path.join(root, name)
-        with open(fpath, 'rt', encoding='utf-8', errors='replace') as f:
-            try:
-                fline = f.read(128)
-                f = io.StringIO(fline)
-                shebang = f.readline()
-            except UnicodeDecodeError as e:
-                print("%s: skipping file `%s': %s" % (sys.argv[0], fpath, e), file=sys.stderr)
-                continue
-            if re.compile(r'^#!/usr/bin/env python\s').match(shebang) \
-                    or re.compile(r'^#!/usr/bin/env python2\s').match(shebang) \
-                    or re.compile(r'^#!/usr/bin/python\s').match(shebang):
-                rep['python2'].append(fpath)
-            elif re.compile(r'^#!/usr/bin/env python3\s').match(shebang):
-                rep['python3'].append(fpath)
-            elif re.compile(r'^#!/usr/bin/env perl\s').match(shebang):
-                rep['perl'].append(fpath)
+        try:
+            with open(fpath, 'rt', encoding='utf-8', errors='replace') as f:
+                try:
+                    fline = f.read(128)
+                    f = io.StringIO(fline)
+                    shebang = f.readline()
+                except UnicodeDecodeError as e:
+                    print("%s: skipping file `%s': %s" % (sys.argv[0], fpath, e), file=sys.stderr)
+                    continue
+                if re.compile(r'^#!\s*/usr/bin/env python\s').match(shebang) \
+                        or re.compile(r'^#!\s*/usr/bin/env\s+python2\s').match(shebang) \
+                        or re.compile(r'^#!\s*/usr/bin/python\s').match(shebang):
+                    rep['python2'].append(fpath)
+                elif re.compile(r'^#!\s*/usr/bin/env\s+python3\s').match(shebang):
+                    rep['python3'].append(fpath)
+                elif re.compile(r'^#!\s*/usr/bin/env\s+perl\s').match(shebang):
+                    rep['perl'].append(fpath)
+                elif re.compile(r'^#!\s*/usr/bin/env\s+ruby\s').match(shebang):
+                    rep['ruby'].append(fpath)
+                elif re.compile(r'^#!\s*/usr/bin/env\s+bash\s').match(shebang):
+                    rep['bash'].append(fpath)
+        except FileNotFoundError:
+            pass
 
-def gf(files):
+def gf(cmd, files):
     newfiles = []
     for f in files:
-        if not rpm_build_root_files or os.path.basename(f) in rpm_build_root_files:
+        if not rpm_build_root_files or hash(f) in rpm_build_root_files:
             newfiles.append(f)
     newfiles.sort()
+    if not newfiles:
+        return
+    print(cmd)
     for i in range(0, len(newfiles) - 1):
-        print("\t%s \\\n" % os.path.relpath(newfiles[i], start=args.sourcedir), end='')
-    print("\t%s\n" % os.path.relpath(newfiles[len(newfiles) - 1], start=args.sourcedir))
+        print("      %s \\\n" % os.path.relpath(newfiles[i], start=args.sourcedir), end='')
+    print("      %s\n" % os.path.relpath(newfiles[len(newfiles) - 1], start=args.sourcedir))
 
-print("\n# Copy from here:", file=sys.stderr)
+print("\n# Copy from here:")
 print("# %s " % sys.argv[0], end='')
 if args.buildroot:
-    print("--root=%s " % args.buildroot, end='')
+    print("--buildroot=%s " % args.buildroot, end='')
 print("%s\n" % args.sourcedir)
 
-if rep['python2']:
-    print("sed -i -e '1s,#!/usr/bin/env python2,%{__python},' -e '1s,#!/usr/bin/env python,%{__python},' -e '1s,#!/usr/bin/python,%{__python},' \\")
-    gf(rep['python2'])
-
-if rep['python3']:
-    print("sed -i -e '1s,#!/usr/bin/env python3,%{__python3},' \\")
-    gf(rep['python3'])
-
-if rep['perl']:
-    print("sed -i -e '1s,#!/usr/bin/env perl,%{__perl},' \\")
-    gf(rep['perl'])
+gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+python2(\s|$),#!%{__python}\\1,' -e '1s,#!\s*/usr/bin/env\s+python(\s|$),#!%{__python}\\1,' -e '1s,#!\s*/usr/bin/python(\s|$),#!%{__python}\\1,' \\",
+   rep['python2'])
+gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+python3(\s|$),#!%{__python3}\\1,' \\", rep['python3'])
+gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+perl(\s|$),#!%{__perl}\\1,' \\", rep['perl'])
+gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+ruby(\s|$),#!%{__ruby}\\1,' \\", rep['ruby'])
+gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+bash(\s|$),#!/bin/bash\\1,' \\", rep['bash'])