util.py

import inspect
import re
import clint.textui
import sys
import yaml
from clint.textui import colored
import json
import pygments
from pygments.lexers import JsonLexer, YamlLexer
from pygments.formatters import TerminalFormatter
import time


def caller():
    # note, need extra 'f_back' because this itself is a function...
    frame = inspect.currentframe().f_back.f_back
    return frame.f_globals, frame.f_locals


def fmt(s, context=None):
    """
    Interpolate arbitrary expressions inside '{}' in strings (similar to Ruby or
    Scala). It should go without saying that because this uses `eval`, it is 
    unsafe to use with any user-provided inputs. It is also probably terribly 
    slow because it's using `inspect` to get the caller's local variables.

    Example:
        x = 1
        print fmt('x is {x}, x+1 is {x+1}')
        > x is  1, x+1 is 2
    """
    context = context or caller()
    return re.compile(r"#{(.*?)}").sub(lambda m: str(eval(m.group(1), *context)), str(s))


def puts(s, newline=True, flush=False):
    clint.textui.puts(fmt(s, caller()), newline)
    if flush:
        sys.stdout.flush()


def puts_err(s):
    clint.textui.puts_err(fmt(s, caller()))


def pretty_json(value):
    return pygments.highlight(unicode(json.dumps(value, indent=2, sort_keys=True), 'UTF-8'), JsonLexer(), TerminalFormatter(bg="dark"))


def pretty_yaml(value):
    return pygments.highlight(unicode(yaml.safe_dump(value, indent=2, default_flow_style=False), 'UTF-8'), YamlLexer(), TerminalFormatter(bg="dark"))


def heading(text):
    return colored.black(text, bold=True)


def note(text):
    return colored.black(text)


ANSISEQ = re.compile(r'(\x9B|\x1B\[)[0-?]*[ -\/]*[@-~]')

def strip_ansi(text):
    return ANSISEQ.sub('', text)


def now():
    return time.strftime("%I:%M:%S", time.localtime())