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

Make Pygments a soft dependency.

Browse Source
This commit is contained in:
Lars Yencken
2013-08-21 19:33:09 +10:00
parent c6bb350a13
commit 39a15a669a
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
cheat
View File

@ -2,10 +2,14 @@
import os
import sys
from pygments import highlight
from pygments.util import ClassNotFound
from pygments.lexers import get_lexer_for_filename, TextLexer
from pygments.formatters import TerminalFormatter
try:
from pygments import highlight
from pygments.util import ClassNotFound
from pygments.lexers import get_lexer_for_filename, TextLexer
from pygments.formatters import TerminalFormatter
USE_PYGMENTS = True
except ImportError:
USE_PYGMENTS = False
def cheat_directories():
@ -63,7 +67,12 @@ def main():
# print the cheatsheet if it exists
if keyphrase in cheatsheets:
filename = os.path.join(cheatsheets[keyphrase], keyphrase)
pretty_print(filename)
if USE_PYGMENTS:
pretty_print(filename)
else:
with open(filename) as istream:
for l in istream:
sys.stdout.write(l)
# if it does not, say so
else: