diff options
Diffstat (limited to 'rainbow_parenthesis/__main__.py')
-rw-r--r-- | rainbow_parenthesis/__main__.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/rainbow_parenthesis/__main__.py b/rainbow_parenthesis/__main__.py new file mode 100644 index 0000000..251731a --- /dev/null +++ b/rainbow_parenthesis/__main__.py @@ -0,0 +1,23 @@ +""" +Entry point for rainbow parenthesis. + +Reads a string from stdin, and outputs it to stdout with all +parenthesis prettily colored. +""" + +from . import colorize, Colored +from . import term +import argparse + +parser = argparse.ArgumentParser(prog='rainbow') +parser.add_argument('input', type=argparse.FileType('r'), + nargs='?', default='-') +args = parser.parse_args() + + +for item in colorize(args.input): + match item: + case Colored(): + print(term.colorize(item), end='') + case s: + print(s, end='') |