mirror of
https://github.com/davidhalter/typeshed.git
synced 2026-05-05 04:55:48 +08:00
[multiprocessing.queues] Add missing pickle methods (#15424)
This commit is contained in:
@@ -1,15 +1,21 @@
|
||||
import sys
|
||||
from types import GenericAlias
|
||||
from typing import Any, Generic, TypeVar
|
||||
from typing import Any, Generic, NewType, TypeVar
|
||||
|
||||
__all__ = ["Queue", "SimpleQueue", "JoinableQueue"]
|
||||
|
||||
_T = TypeVar("_T")
|
||||
|
||||
_QueueState = NewType("_QueueState", object)
|
||||
_JoinableQueueState = NewType("_JoinableQueueState", object)
|
||||
_SimpleQueueState = NewType("_SimpleQueueState", object)
|
||||
|
||||
class Queue(Generic[_T]):
|
||||
# FIXME: `ctx` is a circular dependency and it's not actually optional.
|
||||
# It's marked as such to be able to use the generic Queue in __init__.pyi.
|
||||
def __init__(self, maxsize: int = 0, *, ctx: Any = ...) -> None: ...
|
||||
def __getstate__(self) -> _QueueState: ...
|
||||
def __setstate__(self, state: _QueueState) -> None: ...
|
||||
def put(self, obj: _T, block: bool = True, timeout: float | None = None) -> None: ...
|
||||
def get(self, block: bool = True, timeout: float | None = None) -> _T: ...
|
||||
def qsize(self) -> int: ...
|
||||
@@ -24,6 +30,8 @@ class Queue(Generic[_T]):
|
||||
def __class_getitem__(cls, item: Any, /) -> GenericAlias: ...
|
||||
|
||||
class JoinableQueue(Queue[_T]):
|
||||
def __getstate__(self) -> _JoinableQueueState: ... # type: ignore[override]
|
||||
def __setstate__(self, state: _JoinableQueueState) -> None: ... # type: ignore[override]
|
||||
def task_done(self) -> None: ...
|
||||
def join(self) -> None: ...
|
||||
|
||||
@@ -31,6 +39,8 @@ class SimpleQueue(Generic[_T]):
|
||||
def __init__(self, *, ctx: Any = ...) -> None: ...
|
||||
def close(self) -> None: ...
|
||||
def empty(self) -> bool: ...
|
||||
def __getstate__(self) -> _SimpleQueueState: ...
|
||||
def __setstate__(self, state: _SimpleQueueState) -> None: ...
|
||||
def get(self) -> _T: ...
|
||||
def put(self, obj: _T) -> None: ...
|
||||
def __class_getitem__(cls, item: Any, /) -> GenericAlias: ...
|
||||
|
||||
Reference in New Issue
Block a user