aboutsummaryrefslogtreecommitdiff
path: root/python/cru/_helper.py
diff options
context:
space:
mode:
authorYuqian Yang <crupest@crupest.life>2025-02-23 16:40:32 +0800
committerYuqian Yang <crupest@crupest.life>2025-02-23 16:40:32 +0800
commit8d85090fc6a41aaa9366250982e41a5534bed283 (patch)
tree22051ae36ba29c25121257ef4ca5fe69fc107e87 /python/cru/_helper.py
parent687ae968e10e1738c105fc0538fc29ccb08bfbbf (diff)
downloadcrupest-8d85090fc6a41aaa9366250982e41a5534bed283.tar.gz
crupest-8d85090fc6a41aaa9366250982e41a5534bed283.tar.bz2
crupest-8d85090fc6a41aaa9366250982e41a5534bed283.zip
feat(python): move python codes.
Diffstat (limited to 'python/cru/_helper.py')
-rw-r--r--python/cru/_helper.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/python/cru/_helper.py b/python/cru/_helper.py
new file mode 100644
index 0000000..43baf46
--- /dev/null
+++ b/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)