mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-12-24 05:11:58 +00:00
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
|
#!/usr/bin/env python
|
||
|
import sys
|
||
|
import itertools
|
||
|
import termcolor
|
||
|
|
||
|
def parse(line):
|
||
|
left = line[0:line.find('=')].strip()
|
||
|
right = line[line.find('=')+1:].strip('\'"\n ')
|
||
|
try:
|
||
|
cmd = next(part for part in right.split() if len([char for char in '=<>' if char in part])==0)
|
||
|
except StopIteration:
|
||
|
cmd = right
|
||
|
return (left, right, cmd)
|
||
|
|
||
|
def cheatsheet(lines):
|
||
|
exps = [ parse(line) for line in lines ]
|
||
|
cheatsheet = {'_default': []}
|
||
|
for key, group in itertools.groupby(exps, lambda exp:exp[2]):
|
||
|
group_list = [ item for item in group ]
|
||
|
if len(group_list)==1:
|
||
|
target_aliases = cheatsheet['_default']
|
||
|
else:
|
||
|
if key not in cheatsheet:
|
||
|
cheatsheet[key] = []
|
||
|
target_aliases = cheatsheet[key]
|
||
|
target_aliases.extend(group_list)
|
||
|
return cheatsheet
|
||
|
|
||
|
def pretty_print_group(key, aliases, hightlight=None):
|
||
|
if len(aliases) == 0:
|
||
|
return
|
||
|
group_hl_formatter = lambda g, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'red') for part in ('[%s]' % g).split(hl)])
|
||
|
alias_hl_formatter = lambda alias, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'green') for part in ('\t%s = %s' % alias[0:2]).split(hl)])
|
||
|
group_formatter = lambda g: termcolor.colored('[%s]' % g, 'red')
|
||
|
alias_formatter = lambda alias: termcolor.colored('\t%s = %s' % alias[0:2], 'green')
|
||
|
if hightlight and len(hightlight)>0:
|
||
|
print (group_hl_formatter(key, hightlight))
|
||
|
print ('\n'.join([alias_hl_formatter(alias, hightlight) for alias in aliases]))
|
||
|
else:
|
||
|
print (group_formatter(key))
|
||
|
print ('\n'.join([alias_formatter(alias) for alias in aliases]))
|
||
|
print ('')
|
||
|
|
||
|
def pretty_print(cheatsheet, wfilter):
|
||
|
sorted_key = sorted(cheatsheet.keys())
|
||
|
for key in sorted_key:
|
||
|
aliases = cheatsheet.get(key)
|
||
|
if not wfilter:
|
||
|
pretty_print_group(key, aliases, wfilter)
|
||
|
else:
|
||
|
pretty_print_group(key, [ alias for alias in aliases if alias[0].find(wfilter)>-1 or alias[1].find(wfilter)>-1], wfilter)
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
lines = sys.stdin.readlines()
|
||
|
pretty_print(cheatsheet(lines), sys.argv[1] if len(sys.argv)>1 else None)
|