Git/cheat
1
0
Fork 0
mirror of https://github.com/cheat/cheat.git synced 2024-12-18 18:55:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'more_lexers' of git://github.com/cnicolov/cheat into cnicolov-more_lexers

Browse Source
This commit is contained in:
Chris Lane 2017-02-28 18:22:58 -05:00
commit 7abb663bf4
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cheat/utils.py
View File

@ -13,14 +13,23 @@ def colorize(sheet_content):
try:
from pygments import highlight
from pygments.lexers import BashLexer
from pygments.lexers import get_lexer_by_name
from pygments.formatters import TerminalFormatter
# if pygments can't load, just return the uncolorized text
except ImportError:
return sheet_content
return highlight(sheet_content, BashLexer(), TerminalFormatter())
first_line = sheet_content.splitlines()[0]
lexer = get_lexer_by_name('bash')
if first_line.startswith('```'):
sheet_content = '\n'.join(sheet_content.split('\n')[1:])
try:
lexer = get_lexer_by_name(first_line[3:])
except Exception:
pass
return highlight(sheet_content, lexer, TerminalFormatter())
def die(message):