Git/cheat
1
0
Fork 0
mirror of https://github.com/cheat/cheat.git synced 2025-09-06 03:52:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/liuyang1/cheat into liuyang1-master

Browse Source 
Resolving merge-conflicts.
This commit is contained in:
Chris Lane
2019-01-11 17:00:39 -05:00
parent 1224908445 c0fe871b33
commit 22b64d2d08
2 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
cheat/utils.py
View File

@ -21,13 +21,18 @@ def highlight(needle, haystack):
# if the import succeeds, colorize the needle in haystack
return haystack.replace(needle, colored(needle, os.environ.get('CHEAT_HIGHLIGHT')));
def colorize(sheet_content):
""" Colorizes cheatsheet content if so configured """
# if colorization is not configured, exit early
if os.environ.get('CHEATCOLORS') != 'true':
# only colorize if configured to do so, and if stdout is a tty
if os.environ.get('CHEATCOLORS') != 'true' or not sys.stdout.isatty():
return sheet_content
# don't attempt to colorize an empty cheatsheet
if not sheet_content.strip():
return ""
# otherwise, attempt to import the pygments library
try:
from pygments import highlight
@ -41,6 +46,8 @@ def colorize(sheet_content):
# otherwise, attempt to colorize
first_line = sheet_content.splitlines()[0]
lexer = get_lexer_by_name('bash')
# apply syntax-highlighting if the first line is a code-fence
if first_line.startswith('```'):
sheet_content = '\n'.join(sheet_content.split('\n')[1:-2])
try: