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

fix except case

Browse Source 
- when redirect stdout to pipe but not tty, it throw exception.
- when have no content, it throw exception.
- remove reductant newline at end of file
This commit is contained in:
liuyang1
2018-06-13 18:56:54 +08:00
committed by ly
parent aa1e12625e
commit c0fe871b33
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
cheat/utils.py
View File

@ -10,6 +10,9 @@ def colorize(sheet_content):
# only colorize if so configured
if not 'CHEATCOLORS' in os.environ:
return sheet_content
# only colorize if stdout is tty
if not sys.stdout.isatty():
return sheet_content
try:
from pygments import highlight
@ -20,8 +23,11 @@ def colorize(sheet_content):
except ImportError:
return sheet_content
first_line = sheet_content.splitlines()[0]
lexer = get_lexer_by_name('bash')
lines = sheet_content.splitlines()
if len(lines) == 0:
return ""
first_line = lines[0]
lexer = get_lexer_by_name('bash')
if first_line.startswith('```'):
sheet_content = '\n'.join(sheet_content.split('\n')[1:-2])
try: