diff options
| author | Yuqian Yang <crupest@crupest.life> | 2025-02-23 16:40:32 +0800 | 
|---|---|---|
| committer | Yuqian Yang <crupest@crupest.life> | 2025-02-23 16:40:32 +0800 | 
| commit | 90868bf85dc295f70620dbcbd5790999fe239550 (patch) | |
| tree | 08c0f73597a751acff14a4d224446e87b2d8775d /python/cru/_helper.py | |
| parent | 1e9b2436eaffa4130f6a69c3a108f6feb9dd4ac8 (diff) | |
| download | crupest-90868bf85dc295f70620dbcbd5790999fe239550.tar.gz crupest-90868bf85dc295f70620dbcbd5790999fe239550.tar.bz2 crupest-90868bf85dc295f70620dbcbd5790999fe239550.zip | |
feat(python): move python codes.
Diffstat (limited to 'python/cru/_helper.py')
| -rw-r--r-- | python/cru/_helper.py | 16 | 
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) | 
