|
76 | 76 | "snowflake_time",
|
77 | 77 | "find",
|
78 | 78 | "get_or_fetch",
|
79 |
| - "sleep_until ", |
80 | 79 | "utcnow",
|
81 | 80 | "remove_markdown",
|
82 | 81 | "escape_markdown",
|
@@ -547,68 +546,3 @@ def raw_role_mentions(text: str) -> list[int]:
|
547 | 546 | A list of role IDs found in the string.
|
548 | 547 | """
|
549 | 548 | return [int(x) for x in re.findall(r"<@&([0-9]+)>", text)]
|
550 |
| - |
551 |
| - |
552 |
| -def _chunk(iterator: Iterator[T], max_size: int) -> Iterator[list[T]]: |
553 |
| - ret = [] |
554 |
| - n = 0 |
555 |
| - for item in iterator: |
556 |
| - ret.append(item) |
557 |
| - n += 1 |
558 |
| - if n == max_size: |
559 |
| - yield ret |
560 |
| - ret = [] |
561 |
| - n = 0 |
562 |
| - if ret: |
563 |
| - yield ret |
564 |
| - |
565 |
| - |
566 |
| -async def _achunk(iterator: AsyncIterator[T], max_size: int) -> AsyncIterator[list[T]]: |
567 |
| - ret = [] |
568 |
| - n = 0 |
569 |
| - async for item in iterator: |
570 |
| - ret.append(item) |
571 |
| - n += 1 |
572 |
| - if n == max_size: |
573 |
| - yield ret |
574 |
| - ret = [] |
575 |
| - n = 0 |
576 |
| - if ret: |
577 |
| - yield ret |
578 |
| - |
579 |
| - |
580 |
| -@overload |
581 |
| -def as_chunks(iterator: Iterator[T], max_size: int) -> Iterator[list[T]]: ... |
582 |
| - |
583 |
| - |
584 |
| -@overload |
585 |
| -def as_chunks(iterator: AsyncIterator[T], max_size: int) -> AsyncIterator[list[T]]: ... |
586 |
| - |
587 |
| - |
588 |
| -def as_chunks(iterator: _Iter[T], max_size: int) -> _Iter[list[T]]: |
589 |
| - """A helper function that collects an iterator into chunks of a given size. |
590 |
| -
|
591 |
| - .. versionadded:: 2.0 |
592 |
| -
|
593 |
| - .. warning:: |
594 |
| -
|
595 |
| - The last chunk collected may not be as large as ``max_size``. |
596 |
| -
|
597 |
| - Parameters |
598 |
| - ---------- |
599 |
| - iterator: Union[:class:`collections.abc.Iterator`, :class:`collections.abc.AsyncIterator`] |
600 |
| - The iterator to chunk, can be sync or async. |
601 |
| - max_size: :class:`int` |
602 |
| - The maximum chunk size. |
603 |
| -
|
604 |
| - Returns |
605 |
| - ------- |
606 |
| - Union[:class:`collections.abc.Iterator`, :class:`collections.abc.AsyncIterator`] |
607 |
| - A new iterator which yields chunks of a given size. |
608 |
| - """ |
609 |
| - if max_size <= 0: |
610 |
| - raise ValueError("Chunk sizes must be greater than 0.") |
611 |
| - |
612 |
| - if isinstance(iterator, AsyncIterator): |
613 |
| - return _achunk(iterator, max_size) |
614 |
| - return _chunk(iterator, max_size) |
0 commit comments