diff --git a/stdlib/2.7/cPickle.pyi b/stdlib/2.7/cPickle.pyi index 583cb8be2..fb129d1b5 100644 --- a/stdlib/2.7/cPickle.pyi +++ b/stdlib/2.7/cPickle.pyi @@ -7,10 +7,19 @@ format_version = ... # type: str class Pickler: def __init__(self, file: IO[str], protocol: int = ...) -> None: ... + def dump(self, obj: Any) -> None: ... + + def clear_memo(self) -> None: ... + class Unpickler: def __init__(self, file: IO[str]) -> None: ... + def load(self) -> Any: ... + + def noload(self) -> Any: ... + + def dump(obj: Any, file: IO[str], protocol: int = ...) -> None: ... def dumps(obj: Any, protocol: int = ...) -> str: ... def load(file: IO[str]) -> Any: ... diff --git a/stdlib/2.7/pickle.pyi b/stdlib/2.7/pickle.pyi index 1e47e0219..1b5b1d6f3 100644 --- a/stdlib/2.7/pickle.pyi +++ b/stdlib/2.7/pickle.pyi @@ -1,8 +1,38 @@ # Stubs for pickle (Python 2) -from typing import Any, IO +from typing import Any, BinaryIO -def dump(obj: Any, file: IO[str], protocol: int = ...) -> None: ... -def dumps(obj: Any, protocol: int = ...) -> str: ... -def load(file: IO[str]) -> Any: ... -def loads(str: str) -> Any: ... + +HIGHEST_PROTOCOL = ... # type: int + + +def dump(obj: Any, file: BinaryIO, protocol: int = None) -> None: ... +def dumps(obj: Any, protocol: int = ...) -> bytes: ... +def load(file: BinaryIO) -> Any: ... +def loads(string: bytes) -> Any: ... + + +class PickleError(Exception): + pass + + +class PicklingError(PickleError): + pass + + +class UnpicklingError(PickleError): + pass + + +class Pickler: + def __init__(self, file: BinaryIO, protocol: int = None) -> None: ... + + def dump(self, obj: Any) -> None: ... + + def clear_memo(self) -> None: ... + + +class Unpickler: + def __init__(self, file: BinaryIO) -> None: ... + + def load(self) -> Any: ...