echosystem/cheat
1
0
Fork 0
mirror of https://github.com/cheat/cheat.git synced 2024-11-16 08:58:28 +01:00
Code Issues Projects Releases Packages Wiki Activity

Moved around some code blocks for the sake of readabilty.

Browse source
This commit is contained in:
Chris Lane 2013-09-14 08:56:36 -04:00
parent 2a9220bb6d
commit 48cdbdd836
1 changed files with 36 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
cheat
View file

@ -18,6 +18,7 @@ if os.name == 'posix' and 'CHEATCOLORS' in os.environ:
except ImportError:
pass
def cheat_directories():
"Assembles a list of directories containing cheatsheets."
default_directories = [DEFAULT_CHEAT_DIR]
@ -47,19 +48,6 @@ def cheat_files(cheat_directories):
and not cheat.startswith('__')]))
return cheats
def edit_cheatsheet(cheat, cheatsheets):
"Edit a pre-existing cheatsheet."
if cheat not in cheatsheets:
print ('No cheatsheet found for "%s"' % cheat)
ans = raw_input('Create cheatsheet for "%s" (Y/N)? ' % cheat)
if ans.lower() in ['y', 'yes']:
create_cheatsheet(cheat, cheatsheets)
else:
exit()
subprocess.call([os.environ['EDITOR'],
os.path.join(cheatsheets[cheat], cheat)])
exit()
def create_cheatsheet(cheat, cheatsheets):
"Create a new cheatsheet."
@ -83,6 +71,41 @@ def create_cheatsheet(cheat, cheatsheets):
exit()
def edit_cheatsheet(cheat, cheatsheets):
"Edit a pre-existing cheatsheet."
if cheat not in cheatsheets:
print ('No cheatsheet found for "%s"' % cheat)
ans = raw_input('Create cheatsheet for "%s" (Y/N)? ' % cheat)
if ans.lower() in ['y', 'yes']:
create_cheatsheet(cheat, cheatsheets)
else:
exit()
subprocess.call([os.environ['EDITOR'],
os.path.join(cheatsheets[cheat], cheat)])
exit()
def pretty_print(filename):
"Applies syntax highlighting to a cheatsheet and writes it to stdout"
try:
if os.path.splitext(filename)[1]:
lexer = get_lexer_for_filename(filename)
else:
# shell is a sensible default when there is no extension
lexer = get_lexer_for_filename(filename + '.sh')
except ClassNotFound:
lexer = TextLexer()
with open(filename) as istream:
code = istream.read()
fmt = TerminalFormatter()
highlight(code, lexer, fmt, sys.stdout)
def main():
# assemble a keyphrase out of all params passed to the script
keyphrase = ' '.join(sys.argv[1:])
@ -161,23 +184,5 @@ def main():
print ('No cheatsheet found for %s.' % keyphrase)
def pretty_print(filename):
try:
if os.path.splitext(filename)[1]:
lexer = get_lexer_for_filename(filename)
else:
# shell is a sensible default when there is no extension
lexer = get_lexer_for_filename(filename + '.sh')
except ClassNotFound:
lexer = TextLexer()
with open(filename) as istream:
code = istream.read()
fmt = TerminalFormatter()
highlight(code, lexer, fmt, sys.stdout)
if __name__ == '__main__':
main()