mirror of
https://github.com/Erreur32/cheat.git
synced 2024-11-17 09:28:35 +01:00
Make Pygments a soft dependency.
This commit is contained in:
parent
c6bb350a13
commit
39a15a669a
2 changed files with 14 additions and 6 deletions
19
cheat
19
cheat
|
@ -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:
|
||||
|
|
1
setup.py
1
setup.py
|
@ -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']
|
||||
)
|
||||
|
|
Loading…
Reference in a new issue