aboutsummaryrefslogtreecommitdiff
path: root/store/works/python/cru/_helper.py
diff options
context:
space:
mode:
authorYuqian Yang <crupest@crupest.life>2025-06-10 16:34:42 +0800
committerYuqian Yang <crupest@crupest.life>2025-06-10 16:34:42 +0800
commit4abd6020df12427aed62599a68abba87b1ccc3b8 (patch)
treee2504019eb8663ef0e49d409b2ca279030c70c60 /store/works/python/cru/_helper.py
parent73d711416dc50378321982c88fe01e48ea18e20a (diff)
downloadcrupest-4abd6020df12427aed62599a68abba87b1ccc3b8.tar.gz
crupest-4abd6020df12427aed62599a68abba87b1ccc3b8.tar.bz2
crupest-4abd6020df12427aed62599a68abba87b1ccc3b8.zip
refactor: bye, python!
Diffstat (limited to 'store/works/python/cru/_helper.py')
-rw-r--r--store/works/python/cru/_helper.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/store/works/python/cru/_helper.py b/store/works/python/cru/_helper.py
new file mode 100644
index 0000000..43baf46
--- /dev/null
+++ b/store/works/python/cru/_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)