shaba/openuds
1
0
Fork 1
mirror of https://github.com/dkmstr/openuds.git synced 2025-01-06 13:17:54 +03:00
Code Issues Releases Wiki Activity

Added "sorted_by" parameter to filter helpers

Browse Source
This commit is contained in:
Adolfo Gómez García 2024-06-30 14:30:56 +02:00
parent b5fd7c39de
commit f23d8dd66b
No known key found for this signature in database
GPG Key ID: DD1ABF20724CDA23
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/src/tests/utils/__init__.py
View File

@ -173,19 +173,25 @@ def search_item_by_attr(lst: list[T], attribute: str, value: typing.Any, **kwarg
return item return item
raise ValueError(f'Item with {attribute}=="{value}" not found in list {str(lst)[:100]}') raise ValueError(f'Item with {attribute}=="{value}" not found in list {str(lst)[:100]}')
def filter_list_by_attr(lst: list[T], attribute: str, value: typing.Any, **kwargs: typing.Any) -> list[T]: def filter_list_by_attr(lst: list[T], attribute: str, value: typing.Any,*, sorted_by: str = '', **kwargs: typing.Any) -> list[T]:
""" """
Returns a list of items from a list of items Returns a list of items from a list of items
kwargs are not used, just to let use it as partial on fixtures kwargs are not used, just to let use it as partial on fixtures
""" """
return [item for item in lst if getattr(item, attribute) == value or value is None] values = [item for item in lst if getattr(item, attribute) == value or value is None]
if sorted_by:
values.sort(key=lambda x: getattr(x, sorted_by))
return values
def filter_list_by_attr_list(lst: list[T], attribute: str, values: list[typing.Any], **kwargs: typing.Any) -> list[T]: def filter_list_by_attr_list(lst: list[T], attribute: str, values: list[typing.Any], *, sorted_by: str = '', **kwargs: typing.Any) -> list[T]:
""" """
Returns a list of items from a list of items Returns a list of items from a list of items
kwargs are not used, just to let use it as partial on fixtures kwargs are not used, just to let use it as partial on fixtures
""" """
return [item for item in lst if getattr(item, attribute) in values] values = [item for item in lst if getattr(item, attribute) in values]
if sorted_by:
values.sort(key=lambda x: getattr(x, sorted_by))
return values
def check_userinterface_values(obj: ui.UserInterface, values: ui.gui.ValuesDictType) -> None: def check_userinterface_values(obj: ui.UserInterface, values: ui.gui.ValuesDictType) -> None:
""" """