mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-15 10:19:59 +00:00
Code Issues Projects Releases Wiki Activity

Fix typing for utils.as_chunks

Browse Source
This commit is contained in:
Rapptz 2022-02-17 09:37:58 -05:00
parent b389596f26
commit 3113036a54
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
discord/utils.py
View File

@ -28,6 +28,7 @@ import asyncio
import collections.abc import collections.abc
from typing import ( from typing import (
Any, Any,
AsyncIterable,
AsyncIterator, AsyncIterator,
Callable, Callable,
Dict, Dict,
@ -131,7 +132,6 @@ if TYPE_CHECKING:
class _RequestLike(Protocol): class _RequestLike(Protocol):
headers: Mapping[str, Any] headers: Mapping[str, Any]
P = ParamSpec('P') P = ParamSpec('P')
else: else:
@ -140,7 +140,7 @@ else:
T = TypeVar('T') T = TypeVar('T')
T_co = TypeVar('T_co', covariant=True) T_co = TypeVar('T_co', covariant=True)
_Iter = Union[Iterator[T], AsyncIterator[T]] _Iter = Union[Iterable[T], AsyncIterable[T]]
class CachedSlotProperty(Generic[T, T_co]): class CachedSlotProperty(Generic[T, T_co]):
@ -809,7 +809,7 @@ def escape_mentions(text: str) -> str:
return re.sub(r'@(everyone|here|[!&]?[0-9]{17,20})', '@\u200b\\1', text) return re.sub(r'@(everyone|here|[!&]?[0-9]{17,20})', '@\u200b\\1', text)
def _chunk(iterator: Iterator[T], max_size: int) -> Iterator[List[T]]: def _chunk(iterator: Iterable[T], max_size: int) -> Iterator[List[T]]:
ret = [] ret = []
n = 0 n = 0
for item in iterator: for item in iterator:
@ -823,7 +823,7 @@ def _chunk(iterator: Iterator[T], max_size: int) -> Iterator[List[T]]:
yield ret yield ret
async def _achunk(iterator: AsyncIterator[T], max_size: int) -> AsyncIterator[List[T]]: async def _achunk(iterator: AsyncIterable[T], max_size: int) -> AsyncIterator[List[T]]:
ret = [] ret = []
n = 0 n = 0
async for item in iterator: async for item in iterator:
@ -838,12 +838,12 @@ async def _achunk(iterator: AsyncIterator[T], max_size: int) -> AsyncIterator[Li
@overload @overload
def as_chunks(iterator: Iterator[T], max_size: int) -> Iterator[List[T]]: def as_chunks(iterator: Iterable[T], max_size: int) -> Iterator[List[T]]:
... ...
@overload @overload
def as_chunks(iterator: AsyncIterator[T], max_size: int) -> AsyncIterator[List[T]]: def as_chunks(iterator: AsyncIterable[T], max_size: int) -> AsyncIterator[List[T]]:
... ...
@ -854,7 +854,7 @@ def as_chunks(iterator: _Iter[T], max_size: int) -> _Iter[List[T]]:
Parameters Parameters
---------- ----------
iterator: Union[:class:`collections.abc.Iterator`, :class:`collections.abc.AsyncIterator`] iterator: Union[:class:`collections.abc.Iterable`, :class:`collections.abc.AsyncIterable`]
The iterator to chunk, can be sync or async. The iterator to chunk, can be sync or async.
max_size: :class:`int` max_size: :class:`int`
The maximum chunk size. The maximum chunk size.
@ -872,7 +872,7 @@ def as_chunks(iterator: _Iter[T], max_size: int) -> _Iter[List[T]]:
if max_size <= 0: if max_size <= 0:
raise ValueError('Chunk sizes must be greater than 0.') raise ValueError('Chunk sizes must be greater than 0.')
if isinstance(iterator, AsyncIterator): if isinstance(iterator, AsyncIterable):
return _achunk(iterator, max_size) return _achunk(iterator, max_size)
return _chunk(iterator, max_size) return _chunk(iterator, max_size)