aboutsummaryrefslogtreecommitdiff
path: root/tools/cru-py/cru/_util/_lang.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/cru-py/cru/_util/_lang.py')
-rw-r--r--tools/cru-py/cru/_util/_lang.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/tools/cru-py/cru/_util/_lang.py b/tools/cru-py/cru/_util/_lang.py
new file mode 100644
index 0000000..925ba00
--- /dev/null
+++ b/tools/cru-py/cru/_util/_lang.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)