diff options
author | Yuqian Yang <crupest@crupest.life> | 2025-06-10 16:34:42 +0800 |
---|---|---|
committer | Yuqian Yang <crupest@crupest.life> | 2025-06-10 16:34:42 +0800 |
commit | 4c207e6433a73ae7c75dc75fca3631e2f02dca95 (patch) | |
tree | ec36daff383438b1a8df2f08c8077d9edfb9ce65 /store/works/python/cru/_helper.py | |
parent | b63999fd70ed7878569b969c79820e85621dcc2b (diff) | |
download | crupest-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.py | 16 |
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) |