Customization/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-12-18 10:22:00 +00:00
Code Releases Activity

feat(aliases): add basic argument parsing (#10710)

Browse source
This commit is contained in:
Yuval 2022-10-18 19:40:02 +02:00 committed by GitHub
parent 8487a5536d
commit 4e4961e60c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 25 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
plugins/aliases/README.md
View file

@ -17,6 +17,12 @@ Requirements: Python needs to be installed.
- `acs`: show all aliases by group. - `acs`: show all aliases by group.
- `acs -h/--help`: print help mesage.
- `acs <keyword>`: filter aliases by `<keyword>` and highlight. - `acs <keyword>`: filter aliases by `<keyword>` and highlight.
- `acs -g <group>/--group <group`: show only aliases for group `<group>`. Multiple uses of the flag show all groups,
- `acs --groups-only`: show only group names
![screenshot](https://cloud.githubusercontent.com/assets/3602957/11581913/cb54fb8a-9a82-11e5-846b-5a67f67ad9ad.png) ![screenshot](https://cloud.githubusercontent.com/assets/3602957/11581913/cb54fb8a-9a82-11e5-846b-5a67f67ad9ad.png)

25
plugins/aliases/cheatsheet.py
View file

@ -2,6 +2,7 @@
import sys import sys
import itertools import itertools
import termcolor import termcolor
import argparse
def parse(line): def parse(line):
left = line[0:line.find('=')].strip() left = line[0:line.find('=')].strip()
@ -26,7 +27,7 @@ def cheatsheet(lines):
target_aliases.extend(group_list) target_aliases.extend(group_list)
return cheatsheet return cheatsheet
def pretty_print_group(key, aliases, highlight=None): def pretty_print_group(key, aliases, highlight=None, only_groupname=False):
if len(aliases) == 0: if len(aliases) == 0:
return return
group_hl_formatter = lambda g, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'red') for part in ('[%s]' % g).split(hl)]) group_hl_formatter = lambda g, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'red') for part in ('[%s]' % g).split(hl)])
@ -35,21 +36,33 @@ def pretty_print_group(key, aliases, highlight=None):
alias_formatter = lambda alias: termcolor.colored('\t%s = %s' % alias[0:2], 'green') alias_formatter = lambda alias: termcolor.colored('\t%s = %s' % alias[0:2], 'green')
if highlight and len(highlight)>0: if highlight and len(highlight)>0:
print (group_hl_formatter(key, highlight)) print (group_hl_formatter(key, highlight))
print ('\n'.join([alias_hl_formatter(alias, highlight) for alias in aliases])) if not only_groupname:
print ('\n'.join([alias_hl_formatter(alias, highlight) for alias in aliases]))
else: else:
print (group_formatter(key)) print (group_formatter(key))
print ('\n'.join([alias_formatter(alias) for alias in aliases])) if not only_groupname:
print ('\n'.join([alias_formatter(alias) for alias in aliases]))
print ('') print ('')
def pretty_print(cheatsheet, wfilter): def pretty_print(cheatsheet, wfilter, group_list=None, groups_only=False):
sorted_key = sorted(cheatsheet.keys()) sorted_key = sorted(cheatsheet.keys())
for key in sorted_key: for key in sorted_key:
if group_list and key not in group_list:
continue
aliases = cheatsheet.get(key) aliases = cheatsheet.get(key)
if not wfilter: if not wfilter:
pretty_print_group(key, aliases, wfilter) pretty_print_group(key, aliases, wfilter, groups_only)
else: else:
pretty_print_group(key, [ alias for alias in aliases if alias[0].find(wfilter)>-1 or alias[1].find(wfilter)>-1], wfilter) 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__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Pretty print aliases.")
parser.add_argument('filter', nargs="*", help="search aliases matching string")
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() lines = sys.stdin.readlines()
pretty_print(cheatsheet(lines), sys.argv[1] if len(sys.argv)>1 else None) group_list = args.group_list or None
wfilter = " ".join(args.filter) or None
pretty_print(cheatsheet(lines), wfilter, group_list, args.groups_only)