From 1e9b2436eaffa4130f6a69c3a108f6feb9dd4ac8 Mon Sep 17 00:00:00 2001 From: Yuqian Yang Date: Sat, 22 Feb 2025 18:11:35 +0800 Subject: feat(services): refactor structure. --- services/manager/_helper.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 services/manager/_helper.py (limited to 'services/manager/_helper.py') diff --git a/services/manager/_helper.py b/services/manager/_helper.py new file mode 100644 index 0000000..43baf46 --- /dev/null +++ b/services/manager/_helper.py @@ -0,0 +1,16 @@ +from collections.abc import Callable +from typing import Any, Iterable, TypeVar, cast + +_T = TypeVar("_T") +_D = TypeVar("_D") + + +def remove_element( + iterable: Iterable[_T | None], to_rm: Iterable[Any], des: type[_D] | None = None +) -> _D: + to_rm = set(to_rm) + return cast(Callable[..., _D], des or list)(v for v in iterable if v not in to_rm) + + +def remove_none(iterable: Iterable[_T | None], des: type[_D] | None = None) -> _D: + return cast(Callable[..., _D], des or list)(v for v in iterable if v is not None) -- cgit v1.2.3