Changeset f85bc15 for src/tests/pybin/tools.py

Timestamp:

Jul 23, 2018, 4:46:36 PM (6 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, no_list, persistent-indexer, pthread-emulation, qualifiedEnum

Children:

552f5cb

Parents:

46782ab5

Message:

Test now work outside of tree except for io2

File:

• src/tests/pybin/tools.py (modified) (5 diffs)

src/tests/pybin/tools.py

@@ -9 +9 @@
 import stat
 import sys
+import fileinput
 
 from pybin import settings
@@ -33 +34 @@
                 out, err = proc.communicate()
                 return proc.returncode, out
+
+def is_ascii(fname):
+        if not os.path.isfile(fname):
+                return False
+
+        code, out = sh("file %s" % fname, print2stdout = False)
+        if code != 0:
+                return False
+
+        match = re.search(".*: (.*)", out)
+
+        if not match:
+                return False
+
+        return match.group(1) == "ASCII text"
 
 # Remove 1 or more files silently
@@ -105 +121 @@
 # helper function to replace patterns in a file
 def file_replace(fname, pat, s_after):
-    # first, see if the pattern is even in the file.
-    with open(fname) as f:
-        if not any(re.search(pat, line) for line in f):
-            return # pattern does not occur in file so we are done.
-
-    # pattern is in the file, so perform replace operation.
-    with open(fname) as f:
-        out_fname = fname + ".tmp"
-        out = open(out_fname, "w")
-        for line in f:
-            out.write(re.sub(pat, s_after, line))
-        out.close()
-        os.rename(out_fname, fname)
+        file = fileinput.FileInput(fname, inplace=True, backup='.bak')
+        for line in file:
+                print(line.replace(pat, s_after), end='')
+        file.close()
 
 # helper function to check if a files contains only a specific string
@@ -140 +147 @@
 # transform path to canonical form
 def canonicalPath(path):
-        return os.path.join('.', os.path.normpath(path) )
+        abspath = os.path.abspath(__main__.__file__)
+        dname = os.path.dirname(abspath)
+        return os.path.join(dname, os.path.normpath(path) )
 
 # compare path even if form is different
@@ -151 +160 @@
                 for name in names:
                         path = os.path.join(dirname, name)
-
                         op( path )
 
         # Start the walk
-        os.path.walk('.', step, '')
+        abspath = os.path.abspath(__main__.__file__)
+        dname = os.path.dirname(abspath)
+        os.path.walk(dname, step, '')
 
 ################################################################################