mirror of
https://github.com/Erreur32/cheat.git
synced 2024-11-18 09:50:44 +01:00
c1fbeffde5
If you use cheat to save some programming snippets this might be useful. For example if you have a long list of SQL Query cheats, you can do the following: Enter ```sql in the beginning of the file containing the cheats content. Example file: sql ```sql SELECT 17 & 16 = 16; SELECT 2+4+8+16 & 1 = 0;
74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
from __future__ import print_function
|
|
import os
|
|
import sys
|
|
|
|
|
|
def colorize(sheet_content):
|
|
""" Colorizes cheatsheet content if so configured """
|
|
|
|
# only colorize if so configured
|
|
if not 'CHEATCOLORS' in os.environ:
|
|
return sheet_content
|
|
|
|
try:
|
|
from pygments import highlight
|
|
from pygments.lexers import get_lexer_by_name
|
|
from pygments.formatters import TerminalFormatter
|
|
|
|
# if pygments can't load, just return the uncolorized text
|
|
except ImportError:
|
|
return sheet_content
|
|
|
|
first_line = sheet_content.splitlines()[0]
|
|
lexer = get_lexer_by_name('bash')
|
|
if first_line.startswith('```'):
|
|
sheet_content = '\n'.join(sheet_content.split('\n')[1:])
|
|
try:
|
|
lexer = get_lexer_by_name(first_line[3:])
|
|
except Exception:
|
|
pass
|
|
|
|
return highlight(sheet_content, lexer, TerminalFormatter())
|
|
|
|
|
|
def die(message):
|
|
""" Prints a message to stderr and then terminates """
|
|
warn(message)
|
|
exit(1)
|
|
|
|
|
|
def editor():
|
|
""" Determines the user's preferred editor """
|
|
if 'EDITOR' not in os.environ:
|
|
die(
|
|
'In order to create/edit a cheatsheet you must set your EDITOR '
|
|
'environment variable to your editor\'s path.'
|
|
)
|
|
|
|
elif os.environ['EDITOR'] == "":
|
|
die(
|
|
'Your EDITOR environment variable is set to an empty string. It must '
|
|
'be set to your editor\'s path.'
|
|
)
|
|
|
|
else:
|
|
return os.environ['EDITOR']
|
|
|
|
|
|
def prompt_yes_or_no(question):
|
|
""" Prompts the user with a yes-or-no question """
|
|
# Support Python 2 and 3 input
|
|
# Default to Python 2's input()
|
|
get_input = raw_input
|
|
|
|
# If this is Python 3, use input()
|
|
if sys.version_info[:2] >= (3, 0):
|
|
get_input = input
|
|
|
|
print(question)
|
|
return get_input('[y/n] ') == 'y'
|
|
|
|
|
|
def warn(message):
|
|
""" Prints a message to stderr """
|
|
print((message), file=sys.stderr)
|