| 1 | import __main__ |
|---|
| 2 | import argparse |
|---|
| 3 | import os |
|---|
| 4 | import re |
|---|
| 5 | import stat |
|---|
| 6 | |
|---|
| 7 | from subprocess import Popen, PIPE, STDOUT |
|---|
| 8 | |
|---|
| 9 | # helper functions to run terminal commands |
|---|
| 10 | def sh(cmd, dry_run = False, print2stdout = True): |
|---|
| 11 | if dry_run : # if this is a dry_run, only print the commands that would be ran |
|---|
| 12 | print("cmd: %s" % cmd) |
|---|
| 13 | return 0, None |
|---|
| 14 | else : # otherwise create a pipe and run the desired command |
|---|
| 15 | proc = Popen(cmd, stdout=None if print2stdout else PIPE, stderr=STDOUT, shell=True) |
|---|
| 16 | out, err = proc.communicate() |
|---|
| 17 | return proc.returncode, out |
|---|
| 18 | |
|---|
| 19 | # Remove 1 or more files silently |
|---|
| 20 | def rm( files, dry_run = False ): |
|---|
| 21 | try: |
|---|
| 22 | for file in files: |
|---|
| 23 | sh("rm -f %s > /dev/null 2>&1" % file, dry_run) |
|---|
| 24 | except TypeError: |
|---|
| 25 | sh("rm -f %s > /dev/null 2>&1" % files, dry_run) |
|---|
| 26 | |
|---|
| 27 | def chdir( dest = __main__.__file__ ): |
|---|
| 28 | abspath = os.path.abspath(dest) |
|---|
| 29 | dname = os.path.dirname(abspath) |
|---|
| 30 | os.chdir(dname) |
|---|
| 31 | |
|---|
| 32 | # helper function to replace patterns in a file |
|---|
| 33 | def file_replace(fname, pat, s_after): |
|---|
| 34 | # first, see if the pattern is even in the file. |
|---|
| 35 | with open(fname) as f: |
|---|
| 36 | if not any(re.search(pat, line) for line in f): |
|---|
| 37 | return # pattern does not occur in file so we are done. |
|---|
| 38 | |
|---|
| 39 | # pattern is in the file, so perform replace operation. |
|---|
| 40 | with open(fname) as f: |
|---|
| 41 | out_fname = fname + ".tmp" |
|---|
| 42 | out = open(out_fname, "w") |
|---|
| 43 | for line in f: |
|---|
| 44 | out.write(re.sub(pat, s_after, line)) |
|---|
| 45 | out.close() |
|---|
| 46 | os.rename(out_fname, fname) |
|---|
| 47 | |
|---|
| 48 | # helper function to check if a files contains only a spacific string |
|---|
| 49 | def fileContainsOnly(file, text) : |
|---|
| 50 | with open(file) as f: |
|---|
| 51 | ff = f.read().strip() |
|---|
| 52 | result = ff == text.strip() |
|---|
| 53 | |
|---|
| 54 | return result; |
|---|
| 55 | |
|---|
| 56 | # check whether or not a file is executable |
|---|
| 57 | def fileIsExecutable(file) : |
|---|
| 58 | try : |
|---|
| 59 | fileinfo = os.stat(file) |
|---|
| 60 | return bool(fileinfo.st_mode & stat.S_IXUSR) |
|---|
| 61 | except Exception as inst: |
|---|
| 62 | print(type(inst)) # the exception instance |
|---|
| 63 | print(inst.args) # arguments stored in .args |
|---|
| 64 | print(inst) |
|---|
| 65 | return False |
|---|
| 66 | |
|---|
| 67 | # check if arguments is yes or no |
|---|
| 68 | def yes_no(string): |
|---|
| 69 | if string == "yes" : |
|---|
| 70 | return True |
|---|
| 71 | if string == "no" : |
|---|
| 72 | return False |
|---|
| 73 | raise argparse.ArgumentTypeError(msg) |
|---|
| 74 | return False |
|---|
| 75 | |
|---|
| 76 | # diff two files |
|---|
| 77 | def diff( lhs, rhs, dry_run ): |
|---|
| 78 | # diff the output of the files |
|---|
| 79 | diff_cmd = ("diff --ignore-all-space " |
|---|
| 80 | "--ignore-blank-lines " |
|---|
| 81 | "--old-group-format='\t\tmissing lines :\n" |
|---|
| 82 | "%%<' \\\n" |
|---|
| 83 | "--new-group-format='\t\tnew lines :\n" |
|---|
| 84 | "%%>' \\\n" |
|---|
| 85 | "--unchanged-group-format='%%=' \\" |
|---|
| 86 | "--changed-group-format='\t\texpected :\n" |
|---|
| 87 | "%%<" |
|---|
| 88 | "\t\tgot :\n" |
|---|
| 89 | "%%>\n' \\\n" |
|---|
| 90 | "--new-line-format='\t\t%%dn\t%%L' \\\n" |
|---|
| 91 | "--old-line-format='\t\t%%dn\t%%L' \\\n" |
|---|
| 92 | "--unchanged-line-format='' \\\n" |
|---|
| 93 | "%s %s") |
|---|
| 94 | |
|---|
| 95 | # fetch return code and error from the diff command |
|---|
| 96 | return sh(diff_cmd % (lhs, rhs), dry_run, False) |
|---|
