EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Tools / scripts / pathfix.py

#! /usr/bin/env python\r
\r
# Change the #! line occurring in Python scripts.  The new interpreter\r
# pathname must be given with a -i option.\r
#\r
# Command line arguments are files or directories to be processed.\r
# Directories are searched recursively for files whose name looks\r
# like a python module.\r
# Symbolic links are always ignored (except as explicit directory\r
# arguments).  Of course, the original file is kept as a back-up\r
# (with a "~" attached to its name).\r
#\r
# Undoubtedly you can do this using find and sed or perl, but this is\r
# a nice example of Python code that recurses down a directory tree\r
# and uses regular expressions.  Also note several subtleties like\r
# preserving the file's mode and avoiding to even write a temp file\r
# when no changes are needed for a file.\r
#\r
# NB: by changing only the function fixfile() you can turn this\r
# into a program for a different change to Python programs...\r
\r
import sys\r
import re\r
import os\r
from stat import *\r
import getopt\r
\r
err = sys.stderr.write\r
dbg = err\r
rep = sys.stdout.write\r
\r
new_interpreter = None\r
\r
def main():\r
    global new_interpreter\r
    usage = ('usage: %s -i /interpreter file-or-directory ...\n' %\r
             sys.argv[0])\r
    try:\r
        opts, args = getopt.getopt(sys.argv[1:], 'i:')\r
    except getopt.error, msg:\r
        err(msg + '\n')\r
        err(usage)\r
        sys.exit(2)\r
    for o, a in opts:\r
        if o == '-i':\r
            new_interpreter = a\r
    if not new_interpreter or new_interpreter[0] != '/' or not args:\r
        err('-i option or file-or-directory missing\n')\r
        err(usage)\r
        sys.exit(2)\r
    bad = 0\r
    for arg in args:\r
        if os.path.isdir(arg):\r
            if recursedown(arg): bad = 1\r
        elif os.path.islink(arg):\r
            err(arg + ': will not process symbolic links\n')\r
            bad = 1\r
        else:\r
            if fix(arg): bad = 1\r
    sys.exit(bad)\r
\r
ispythonprog = re.compile('^[a-zA-Z0-9_]+\.py$')\r
def ispython(name):\r
    return ispythonprog.match(name) >= 0\r
\r
def recursedown(dirname):\r
    dbg('recursedown(%r)\n' % (dirname,))\r
    bad = 0\r
    try:\r
        names = os.listdir(dirname)\r
    except os.error, msg:\r
        err('%s: cannot list directory: %r\n' % (dirname, msg))\r
        return 1\r
    names.sort()\r
    subdirs = []\r
    for name in names:\r
        if name in (os.curdir, os.pardir): continue\r
        fullname = os.path.join(dirname, name)\r
        if os.path.islink(fullname): pass\r
        elif os.path.isdir(fullname):\r
            subdirs.append(fullname)\r
        elif ispython(name):\r
            if fix(fullname): bad = 1\r
    for fullname in subdirs:\r
        if recursedown(fullname): bad = 1\r
    return bad\r
\r
def fix(filename):\r
##  dbg('fix(%r)\n' % (filename,))\r
    try:\r
        f = open(filename, 'r')\r
    except IOError, msg:\r
        err('%s: cannot open: %r\n' % (filename, msg))\r
        return 1\r
    line = f.readline()\r
    fixed = fixline(line)\r
    if line == fixed:\r
        rep(filename+': no change\n')\r
        f.close()\r
        return\r
    head, tail = os.path.split(filename)\r
    tempname = os.path.join(head, '@' + tail)\r
    try:\r
        g = open(tempname, 'w')\r
    except IOError, msg:\r
        f.close()\r
        err('%s: cannot create: %r\n' % (tempname, msg))\r
        return 1\r
    rep(filename + ': updating\n')\r
    g.write(fixed)\r
    BUFSIZE = 8*1024\r
    while 1:\r
        buf = f.read(BUFSIZE)\r
        if not buf: break\r
        g.write(buf)\r
    g.close()\r
    f.close()\r
\r
    # Finishing touch -- move files\r
\r
    # First copy the file's mode to the temp file\r
    try:\r
        statbuf = os.stat(filename)\r
        os.chmod(tempname, statbuf[ST_MODE] & 07777)\r
    except os.error, msg:\r
        err('%s: warning: chmod failed (%r)\n' % (tempname, msg))\r
    # Then make a backup of the original file as filename~\r
    try:\r
        os.rename(filename, filename + '~')\r
    except os.error, msg:\r
        err('%s: warning: backup failed (%r)\n' % (filename, msg))\r
    # Now move the temp file to the original file\r
    try:\r
        os.rename(tempname, filename)\r
    except os.error, msg:\r
        err('%s: rename failed (%r)\n' % (filename, msg))\r
        return 1\r
    # Return succes\r
    return 0\r
\r
def fixline(line):\r
    if not line.startswith('#!'):\r
        return line\r
    if "python" not in line:\r
        return line\r
    return '#! %s\n' % new_interpreter\r
\r
if __name__ == '__main__':\r
    main()\r