Shellmen/src/libs/PyInquirer/utils.py

36 lines
930 B
Python
Raw Normal View History

# -*- coding: utf-8 -*-
from __future__ import print_function
import json
import sys
from pprint import pprint
from pygments import highlight, lexers, formatters
__version__ = '0.1.2'
PY3 = sys.version_info[0] >= 3
def format_json(data):
return json.dumps(data, sort_keys=True, indent=4)
def colorize_json(data):
if PY3:
if isinstance(data, bytes):
data = data.decode('UTF-8')
else:
if not isinstance(data, unicode):
data = unicode(data, 'UTF-8')
colorful_json = highlight(data,
lexers.JsonLexer(),
formatters.TerminalFormatter())
return colorful_json
def print_json(data):
#colorful_json = highlight(unicode(format_json(data), 'UTF-8'),
# lexers.JsonLexer(),
# formatters.TerminalFormatter())
pprint(colorize_json(format_json(data)))