mirror of
https://github.com/httpie/cli.git
synced 2024-11-24 08:22:22 +02:00
31 lines
789 B
Python
31 lines
789 B
Python
import os
|
|
import json
|
|
|
|
import pygments
|
|
|
|
from pygments import token
|
|
from pygments.util import ClassNotFound
|
|
from pygments.styles import get_style_by_name, STYLE_MAP
|
|
from pygments.lexers import get_lexer_for_mimetype, HttpLexer
|
|
from pygments.formatters.terminal import TerminalFormatter
|
|
from pygments.formatters.terminal256 import Terminal256Formatter
|
|
|
|
from . import solarized
|
|
|
|
|
|
DEFAULT_STYLE = 'solarized'
|
|
AVAILABLE_STYLES = [DEFAULT_STYLE] + list(STYLE_MAP.keys())
|
|
FORMATTER = (Terminal256Formatter
|
|
if '256color' in os.environ.get('TERM', '')
|
|
else TerminalFormatter)
|
|
|
|
|
|
def formatter(style_name):
|
|
if style_name == 'solarized':
|
|
style = solarized.SolarizedStyle
|
|
|
|
else:
|
|
style = get_style_by_name(style_name)
|
|
|
|
return FORMATTER(style=style)
|