diff options
| author | crupest <crupest@outlook.com> | 2024-11-11 01:12:29 +0800 |
|---|---|---|
| committer | Yuqian Yang <crupest@crupest.life> | 2024-12-18 18:31:27 +0800 |
| commit | 7692763f83ad1be735b0b9e9ab0af8ce666d8de8 (patch) | |
| tree | b45e6cb4312b050ac4e1ccf51106d44af9201b40 /tools/cru-py/cru/_helper.py | |
| parent | 297beaf0fedd297dde6d665bd13c929b95b3d1c4 (diff) | |
| download | crupest-7692763f83ad1be735b0b9e9ab0af8ce666d8de8.tar.gz crupest-7692763f83ad1be735b0b9e9ab0af8ce666d8de8.tar.bz2 crupest-7692763f83ad1be735b0b9e9ab0af8ce666d8de8.zip | |
HALF WORK: 2024.12.17
Diffstat (limited to 'tools/cru-py/cru/_helper.py')
| -rw-r--r-- | tools/cru-py/cru/_helper.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tools/cru-py/cru/_helper.py b/tools/cru-py/cru/_helper.py new file mode 100644 index 0000000..43baf46 --- /dev/null +++ b/tools/cru-py/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) |