mirror of
https://github.com/davidhalter/typeshed.git
synced 2025-12-07 04:34:28 +08:00
102 lines
4.0 KiB
Python
102 lines
4.0 KiB
Python
# Stubs for itertools
|
|
|
|
# Based on http://docs.python.org/3.2/library/itertools.html
|
|
|
|
import sys
|
|
from typing import Any, Callable, Generic, Iterable, Iterator, Optional, Tuple, TypeVar, overload
|
|
|
|
_T = TypeVar("_T")
|
|
_S = TypeVar("_S")
|
|
_N = TypeVar("_N", int, float)
|
|
Predicate = Callable[[_T], object]
|
|
|
|
def count(start: _N = ..., step: _N = ...) -> Iterator[_N]: ... # more general types?
|
|
|
|
class cycle(Iterator[_T], Generic[_T]):
|
|
def __init__(self, iterable: Iterable[_T]) -> None: ...
|
|
def __next__(self) -> _T: ...
|
|
def __iter__(self) -> Iterator[_T]: ...
|
|
|
|
@overload
|
|
def repeat(object: _T) -> Iterator[_T]: ...
|
|
@overload
|
|
def repeat(object: _T, times: int) -> Iterator[_T]: ...
|
|
|
|
if sys.version_info >= (3, 8):
|
|
@overload
|
|
def accumulate(iterable: Iterable[_T], func: Callable[[_T, _T], _T] = ...) -> Iterator[_T]: ...
|
|
@overload
|
|
def accumulate(iterable: Iterable[_T], func: Callable[[_S, _T], _S], initial: Optional[_S]) -> Iterator[_S]: ...
|
|
|
|
else:
|
|
def accumulate(iterable: Iterable[_T], func: Callable[[_T, _T], _T] = ...) -> Iterator[_T]: ...
|
|
|
|
class chain(Iterator[_T], Generic[_T]):
|
|
def __init__(self, *iterables: Iterable[_T]) -> None: ...
|
|
def __next__(self) -> _T: ...
|
|
def __iter__(self) -> Iterator[_T]: ...
|
|
@staticmethod
|
|
def from_iterable(iterable: Iterable[Iterable[_S]]) -> Iterator[_S]: ...
|
|
|
|
def compress(data: Iterable[_T], selectors: Iterable[Any]) -> Iterator[_T]: ...
|
|
def dropwhile(predicate: Predicate[_T], iterable: Iterable[_T]) -> Iterator[_T]: ...
|
|
def filterfalse(predicate: Optional[Predicate[_T]], iterable: Iterable[_T]) -> Iterator[_T]: ...
|
|
@overload
|
|
def groupby(iterable: Iterable[_T], key: None = ...) -> Iterator[Tuple[_T, Iterator[_T]]]: ...
|
|
@overload
|
|
def groupby(iterable: Iterable[_T], key: Callable[[_T], _S]) -> Iterator[Tuple[_S, Iterator[_T]]]: ...
|
|
@overload
|
|
def islice(iterable: Iterable[_T], stop: Optional[int]) -> Iterator[_T]: ...
|
|
@overload
|
|
def islice(iterable: Iterable[_T], start: Optional[int], stop: Optional[int], step: Optional[int] = ...) -> Iterator[_T]: ...
|
|
def starmap(func: Callable[..., _S], iterable: Iterable[Iterable[Any]]) -> Iterator[_S]: ...
|
|
def takewhile(predicate: Predicate[_T], iterable: Iterable[_T]) -> Iterator[_T]: ...
|
|
def tee(iterable: Iterable[_T], n: int = ...) -> Tuple[Iterator[_T], ...]: ...
|
|
def zip_longest(*p: Iterable[Any], fillvalue: Any = ...) -> Iterator[Any]: ...
|
|
|
|
_T1 = TypeVar("_T1")
|
|
_T2 = TypeVar("_T2")
|
|
_T3 = TypeVar("_T3")
|
|
_T4 = TypeVar("_T4")
|
|
_T5 = TypeVar("_T5")
|
|
_T6 = TypeVar("_T6")
|
|
@overload
|
|
def product(iter1: Iterable[_T1]) -> Iterator[Tuple[_T1]]: ...
|
|
@overload
|
|
def product(iter1: Iterable[_T1], iter2: Iterable[_T2]) -> Iterator[Tuple[_T1, _T2]]: ...
|
|
@overload
|
|
def product(iter1: Iterable[_T1], iter2: Iterable[_T2], iter3: Iterable[_T3]) -> Iterator[Tuple[_T1, _T2, _T3]]: ...
|
|
@overload
|
|
def product(
|
|
iter1: Iterable[_T1], iter2: Iterable[_T2], iter3: Iterable[_T3], iter4: Iterable[_T4]
|
|
) -> Iterator[Tuple[_T1, _T2, _T3, _T4]]: ...
|
|
@overload
|
|
def product(
|
|
iter1: Iterable[_T1], iter2: Iterable[_T2], iter3: Iterable[_T3], iter4: Iterable[_T4], iter5: Iterable[_T5]
|
|
) -> Iterator[Tuple[_T1, _T2, _T3, _T4, _T5]]: ...
|
|
@overload
|
|
def product(
|
|
iter1: Iterable[_T1],
|
|
iter2: Iterable[_T2],
|
|
iter3: Iterable[_T3],
|
|
iter4: Iterable[_T4],
|
|
iter5: Iterable[_T5],
|
|
iter6: Iterable[_T6],
|
|
) -> Iterator[Tuple[_T1, _T2, _T3, _T4, _T5, _T6]]: ...
|
|
@overload
|
|
def product(
|
|
iter1: Iterable[Any],
|
|
iter2: Iterable[Any],
|
|
iter3: Iterable[Any],
|
|
iter4: Iterable[Any],
|
|
iter5: Iterable[Any],
|
|
iter6: Iterable[Any],
|
|
iter7: Iterable[Any],
|
|
*iterables: Iterable[Any],
|
|
) -> Iterator[Tuple[Any, ...]]: ...
|
|
@overload
|
|
def product(*iterables: Iterable[Any], repeat: int = ...) -> Iterator[Tuple[Any, ...]]: ...
|
|
def permutations(iterable: Iterable[_T], r: Optional[int] = ...) -> Iterator[Tuple[_T, ...]]: ...
|
|
def combinations(iterable: Iterable[_T], r: int) -> Iterator[Tuple[_T, ...]]: ...
|
|
def combinations_with_replacement(iterable: Iterable[_T], r: int) -> Iterator[Tuple[_T, ...]]: ...
|