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 | 6f285760e1432a6c7f60d94c15af4bb3f29ce8fa (patch) | |
tree | ce5a6b3dc74a012766c6d09e7e97cf8eba25ff9e /tools/cru-py/cru/_lang.py | |
parent | d2ed5fa3e26c3a6d0ad700791bc7de955b8d68e0 (diff) | |
download | crupest-6f285760e1432a6c7f60d94c15af4bb3f29ce8fa.tar.gz crupest-6f285760e1432a6c7f60d94c15af4bb3f29ce8fa.tar.bz2 crupest-6f285760e1432a6c7f60d94c15af4bb3f29ce8fa.zip |
HALF WORK: 2024.11.28
Diffstat (limited to 'tools/cru-py/cru/_lang.py')
-rw-r--r-- | tools/cru-py/cru/_lang.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tools/cru-py/cru/_lang.py b/tools/cru-py/cru/_lang.py new file mode 100644 index 0000000..925ba00 --- /dev/null +++ b/tools/cru-py/cru/_lang.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) |