llvm/llvm/utils/lit/lit/builtin_commands/cat.py

import getopt
import sys

try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO


def convertToCaretAndMNotation(data):
    newdata = StringIO()
    if isinstance(data, str):
        data = bytearray(data.encode())

    for intval in data:
        if intval == 9 or intval == 10:
            newdata.write(chr(intval))
            continue
        if intval > 127:
            intval = intval - 128
            newdata.write("M-")
        if intval < 32:
            newdata.write("^")
            newdata.write(chr(intval + 64))
        elif intval == 127:
            newdata.write("^?")
        else:
            newdata.write(chr(intval))

    return newdata.getvalue().encode()


def main(argv):
    arguments = argv[1:]
    short_options = "v"
    long_options = ["show-nonprinting"]
    show_nonprinting = False

    try:
        options, filenames = getopt.gnu_getopt(arguments, short_options, long_options)
    except getopt.GetoptError as err:
        sys.stderr.write("Unsupported: 'cat':  %s\n" % str(err))
        sys.exit(1)

    for option, value in options:
        if option == "-v" or option == "--show-nonprinting":
            show_nonprinting = True

    writer = getattr(sys.stdout, "buffer", None)
    if writer is None:
        writer = sys.stdout
        if sys.platform == "win32":
            import os, msvcrt

            msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
    for filename in filenames:
        try:
            contents = None
            is_text = False
            try:
                if sys.platform != "win32":
                    fileToCat = open(filename, "r")
                    contents = fileToCat.read()
                    is_text = True
            except:
                pass

            if contents is None:
                fileToCat = open(filename, "rb")
                contents = fileToCat.read()

            if show_nonprinting:
                contents = convertToCaretAndMNotation(contents)
            elif is_text:
                contents = contents.encode()
            writer.write(contents)
            sys.stdout.flush()
            fileToCat.close()
        except IOError as error:
            sys.stderr.write(str(error))
            sys.exit(1)


if __name__ == "__main__":
    main(sys.argv)