view prettyprint.py @ 254:156edbd0733a pyqt5

the pyqt5 branch was merged into the default branch
author Dirk Olmes <dirk@xanthippe.ping.de>
date Fri, 10 Nov 2017 01:45:55 +0100
parents 7c719c4f5655
children
line wrap: on
line source

# -*- coding: utf-8 -*-
import sys

newlineAfter = ['{', '[', ',']
newlineBefore = ['}', ']']

def prettyPrint(file):
    indent = 0
    doIndent = False
    printCommand = None
    for line in file:
        for char in line:
            if char in newlineAfter:
                if char is not ',':
                    indent = indent + 1
                doIndent = True
                printCommand = p_nl
            elif char in newlineBefore:
                indent = indent - 1
                doIndent = True
                printCommand = nl_p
            else:
                printCommand = p
                doIndent = False

            printCommand(char)
            if doIndent:
                for x in range(indent):
                    sys.stdout.write("    ")
                    doIndent = False

def p(char):
    sys.stdout.write(char)

def p_nl(char):
    sys.stdout.write(char)
    sys.stdout.write("\n")

def nl_p(char):
    sys.stdout.write("\n")
    sys.stdout.write(char)

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("usage: %s <file>" % (sys.argv[0]))
        sys.exit(1)

    filename = sys.argv[1]
    with open(filename) as file:
        prettyPrint(file)