holm/httpie-cli
1
1
Fork 0
forked from extern/httpie-cli
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor get_formatters_grouped

Browse Source
This commit is contained in:
Jakub Roztocil 2019-09-01 11:45:47 +02:00
parent 4dffac7a25
commit bd3208cf24
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
httpie/plugins/base.py
View File

@ -89,6 +89,7 @@ class ConverterPlugin(BasePlugin):
class FormatterPlugin(BasePlugin):
group_name = 'format'
def __init__(self, **kwargs):
"""

12
httpie/plugins/manager.py
View File

@ -1,4 +1,5 @@
from itertools import groupby
from operator import attrgetter
from typing import Dict, List, Type
from pkg_resources import iter_entry_points
@ -51,12 +52,11 @@ class PluginManager(list):
return self.filter(FormatterPlugin)
def get_formatters_grouped(self) -> Dict[str, List[Type[FormatterPlugin]]]:
groups = {}
for group_name, group in groupby(
self.get_formatters(),
key=lambda p: getattr(p, 'group_name', 'format')):
groups[group_name] = list(group)
return groups
return {
group_name: list(group)
for group_name, group
in groupby(self.get_formatters(), key=attrgetter('group_name'))
}
def get_converters(self) -> List[Type[ConverterPlugin]]:
return self.filter(ConverterPlugin)