def _cat(self, path, patterns=[]):
"""Returns cat() output without assertions."""
with StringIO() as out:
writer = self._make_writer(out)
pycat.cat(path, generate_tokens, patterns, writer)
actual = out.getvalue()
return actual
#!/usr/bin/env python3
"""Writes ansi-colored .py file to stdout.
Styles are described in pycat/styles. Patterns for regions to be
styled are defined in pycat/patterns.
"""
import sys
import ptTools
from ptTools.writers import AnsiParseTreeWriter
from ptTools.tokenizer import Tokenizer
from ptTools.examples import pycat
from ptTools.examples.pycat.patterns import node_patterns
from ptTools.examples.pycat.patterns import token_patterns
if __name__ == '__main__':
FILENAME = sys.argv[1]
WRITER = AnsiParseTreeWriter(sys.stdout)
TOKENIZER = Tokenizer(token_patterns, attr_type=dict)
pycat.cat(FILENAME, TOKENIZER, node_patterns, WRITER)
