Make Pygments a soft dependency.

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

9
cheat
View File

@ -2,10 +2,14 @@
import os
import sys
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)
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:

View File

@ -12,6 +12,5 @@ setup(name='cheat',
packages=['cheatsheets'],
package_data={'cheatsheets': [f for f in os.listdir('cheatsheets')
if '.' not in f]},
install_requires=['Pygments>=1.6'],
scripts=['cheat']
)