check-shebang.py

   1 #!/usr/bin/python3
   2
   3 # thisscript.py --buildroot=~/tmp/somepackage ~/rpm/BUILD/somepackage/
   4
   5 import argparse
   6 import hashlib
   7 import io
   8 import os
   9 import re
  10 import sys
  11
  12 parser = argparse.ArgumentParser()
  13 parser.add_argument('sourcedir', help='RPM_SOURCE_DIR directory')
  14 parser.add_argument("--buildroot", help="RPM_BUILD_ROOT directory")
  15 args = parser.parse_args()
  16
  17 rep = {
  18     'python2': [],
  19     'python3': [],
  20     'perl': [],
  21     'ruby': [],
  22     'bash': [],
  23 }
  24
  25 skip_files = [".h", ".c", ".cc", ".gif", ".png", ".jpg", ".ko", ".gz", ".o"]
  26
  27 def hash(fname):
  28     hash_alg = hashlib.sha256()
  29     with open(fname, "rb") as f:
  30         for chunk in iter(lambda: f.read(4096), b""):
  31             hash_alg.update(chunk)
  32     return hash_alg.hexdigest()
  33
  34 rpm_build_root_files = []
  35 if args.buildroot:
  36     print("%s: Caching `%s' files..." % (sys.argv[0], args.buildroot))
  37     for root, dirs, files in os.walk(args.buildroot):
  38         for name in files:
  39             fname, fext = os.path.splitext(name)
  40             if fext in skip_files:
  41                 continue
  42             fpath = os.path.join(root, name)
  43             try:
  44                 rpm_build_root_files.append(hash(fpath))
  45             except FileNotFoundError:
  46                 pass
  47     print("%s: Caching done." % (sys.argv[0]))
  48
  49 for root, dirs, files in os.walk(args.sourcedir):
  50     for name in files:
  51         fname, fext = os.path.splitext(name)
  52         if fext in skip_files:
  53             continue
  54
  55         fpath = os.path.join(root, name)
  56         try:
  57             with open(fpath, 'rt', encoding='utf-8', errors='replace') as f:
  58                 try:
  59                     fline = f.read(128)
  60                     f = io.StringIO(fline)
  61                     shebang = f.readline()
  62                 except UnicodeDecodeError as e:
  63                     print("%s: skipping file `%s': %s" % (sys.argv[0], fpath, e), file=sys.stderr)
  64                     continue
  65                 if re.compile(r'^#!\s*/usr/bin/env python\s').match(shebang) \
  66                         or re.compile(r'^#!\s*/usr/bin/env\s+python2\s').match(shebang) \
  67                         or re.compile(r'^#!\s*/usr/bin/python\s').match(shebang):
  68                     rep['python2'].append(fpath)
  69                 elif re.compile(r'^#!\s*/usr/bin/env\s+python3\s').match(shebang):
  70                     rep['python3'].append(fpath)
  71                 elif re.compile(r'^#!\s*/usr/bin/env\s+perl\s').match(shebang):
  72                     rep['perl'].append(fpath)
  73                 elif re.compile(r'^#!\s*/usr/bin/env\s+ruby\s').match(shebang):
  74                     rep['ruby'].append(fpath)
  75                 elif re.compile(r'^#!\s*/usr/bin/env\s+bash\s').match(shebang):
  76                     rep['bash'].append(fpath)
  77         except FileNotFoundError:
  78             pass
  79
  80 def gf(cmd, files):
  81     newfiles = []
  82     for f in files:
  83         if not rpm_build_root_files or hash(f) in rpm_build_root_files:
  84             newfiles.append(f)
  85     newfiles.sort()
  86     if not newfiles:
  87         return
  88     print(cmd)
  89     for i in range(0, len(newfiles) - 1):
  90         print("      %s \\\n" % os.path.relpath(newfiles[i], start=args.sourcedir), end='')
  91     print("      %s\n" % os.path.relpath(newfiles[len(newfiles) - 1], start=args.sourcedir))
  92
  93 print("\n# Copy from here:")
  94 print("# %s " % sys.argv[0], end='')
  95 if args.buildroot:
  96     print("--buildroot=%s " % args.buildroot, end='')
  97 print("%s\n" % args.sourcedir)
  98
  99 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,' \\",
 100    rep['python2'])
 101 gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+python3(\s|$),#!%{__python3}\\1,' \\", rep['python3'])
 102 gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+perl(\s|$),#!%{__perl}\\1,' \\", rep['perl'])
 103 gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+ruby(\s|$),#!%{__ruby}\\1,' \\", rep['ruby'])
 104 gf("%{__sed} -E -i -e '1s,#!\s*/usr/bin/env\s+bash(\s|$),#!/bin/bash\\1,' \\", rep['bash'])