Exemple #1
0
_absent = object()


def bindStaticVariable(name, value):
    def decorator(fn):
        setattr(fn, name, value)
        return fn

    return decorator


@bindStaticVariable('formatter', Terminal256Formatter(style=SolarizedDark))
@bindStaticVariable(
    'lexer',
    PyLexer(ensurenl=False) if PYTHON2 else Py3Lexer(ensurenl=False))
def colorize(s):
    self = colorize
    return highlight(s, self.lexer, self.formatter)


@contextmanager
def supportTerminalColorsInWindows():
    # Filter and replace ANSI escape sequences on Windows with equivalent Win32
    # API calls. This code does nothing on non-Windows systems.
    colorama.init()
    yield
    colorama.deinit()


def stderrPrint(*args):
Exemple #2
0
import datetime as dt
import pprint

from pygments.lexers import Python3Lexer as Py3Lexer
from pygments.token import Token

lexer = Py3Lexer(ensurenl=False)

def remove_comments(code_text):
    toks = list(lexer.get_tokens(text=code_text))
    return ''.join([text for kind, text in toks if not kind in Token.Comment])

def timestamp():
    return str(dt.datetime.now())

def pformatml(v):
    """pretty print formatting for multi-line outputs"""
    v = pprint.pformat(v)
    if '\n' in v:
        v = '\n' + v
    return v

def colorize(instr, code = 223):
    # 224 is the same color jupyter uses for stderr.
    bgnd = u"\u001b[48;5;" + str(code) + "m "
    reset = "\u001b[0m"
    return bgnd + instr + reset