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
commit4c207e6433a73ae7c75dc75fca3631e2f02dca95 (patch)
treeec36daff383438b1a8df2f08c8077d9edfb9ce65 /store/works/python/cru/_helper.py
parentb63999fd70ed7878569b969c79820e85621dcc2b (diff)
downloadcrupest-4c207e6433a73ae7c75dc75fca3631e2f02dca95.tar.gz
crupest-4c207e6433a73ae7c75dc75fca3631e2f02dca95.tar.bz2
crupest-4c207e6433a73ae7c75dc75fca3631e2f02dca95.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)