mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2026-05-29 04:53:17 +02:00
cc0f7eee90
Fixed the following issues: 1. gitstatus.py: - Fixed broad IOError exception handling to catch specific OSError/FileNotFoundError - Fixed unused variable warning by using _ for ignored stderr 2. proxy.py: - Fixed os.path.expandvars usage that doesn't expand ~, changed to os.path.expanduser - Fixed check_output call that was incorrectly passing a file path instead of command list - Fixed missing space in env variable assignment for aliases 3. update_emoji.py: - Fixed file handle leaks by using with statements for file operations - Removed unnecessary .keys() call in dict iteration 4. cheatsheet.py: - Fixed parse() function to handle lines without '=' character - Fixed inefficient list comprehension using any() instead - Fixed shadowing of built-in 'cheatsheet' function name 5. ssh-agent.py: - Modernized string formatting from .format() to f-string 6. ssh-proxy.py: - Fixed next() call without default value that crashes when no proxy env vars set - Fixed missing exit code propagation from subprocess.call Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
72 lines
3.1 KiB
Python
72 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import itertools
|
|
import termcolor
|
|
import argparse
|
|
|
|
def parse(line):
|
|
eq_pos = line.find('=')
|
|
if eq_pos == -1:
|
|
return (line.strip(), "", "")
|
|
left = line[0:eq_pos].strip()
|
|
right = line[eq_pos+1:].strip('\'"\n ')
|
|
try:
|
|
cmd = next(part for part in right.split() if not any(char in part for char in '=<>'))
|
|
except StopIteration:
|
|
cmd = right
|
|
return (left, right, cmd)
|
|
|
|
def cheatsheet(lines):
|
|
exps = [ parse(line) for line in lines ]
|
|
exps.sort(key=lambda exp:exp[2])
|
|
cheatsheet_data = {'_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_data['_default']
|
|
else:
|
|
if key not in cheatsheet_data:
|
|
cheatsheet_data[key] = []
|
|
target_aliases = cheatsheet_data[key]
|
|
target_aliases.extend(group_list)
|
|
return cheatsheet_data
|
|
|
|
def pretty_print_group(key, aliases, highlight=None, only_groupname=False):
|
|
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))
|
|
if not only_groupname:
|
|
print ('\n'.join([alias_hl_formatter(alias, highlight) for alias in aliases]))
|
|
else:
|
|
print (group_formatter(key))
|
|
if not only_groupname:
|
|
print ('\n'.join([alias_formatter(alias) for alias in aliases]))
|
|
print ('')
|
|
|
|
def pretty_print(cheatsheet, wfilter, group_list=None, groups_only=False):
|
|
sorted_key = sorted(cheatsheet.keys())
|
|
for key in sorted_key:
|
|
if group_list and key not in group_list:
|
|
continue
|
|
aliases = cheatsheet.get(key)
|
|
if not wfilter:
|
|
pretty_print_group(key, aliases, wfilter, groups_only)
|
|
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__':
|
|
parser = argparse.ArgumentParser(description="Pretty print aliases.", prog="als")
|
|
parser.add_argument('filter', nargs="*", metavar="<keyword>", help="search aliases matching keywords")
|
|
parser.add_argument('-g', '--group', dest="group_list", action='append', help="only print aliases in given groups")
|
|
parser.add_argument('--groups', dest='groups_only', action='store_true', help="only print alias groups")
|
|
args = parser.parse_args()
|
|
|
|
lines = sys.stdin.readlines()
|
|
group_list = args.group_list or None
|
|
wfilter = " ".join(args.filter) or None
|
|
pretty_print(cheatsheet(lines), wfilter, group_list, args.groups_only)
|