#!/usr/bin/env python3 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, highlight=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 highlight and len(highlight)>0: print (group_hl_formatter(key, highlight)) print ('\n'.join([alias_hl_formatter(alias, highlight) 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)