Improve http

stdlib/3/http/client.pyi

@@ -1,101 +1,213 @@
 # Stubs for http.client (Python 3.4)
 
-from typing import Any, Dict
+from typing import (
+ Any, Dict, IO, Iterable, List, Iterator, Mapping, Optional, Tuple, TypeVar,
+ Union,
+ overload,
+)
 import email.message
 import io
+import sys
+import ssl
+import types
+
+_DataType = Union[bytes, IO[Any], Iterable[bytes], str]
+_T = TypeVar('_T')
+
+HTTP_PORT = ... # type: int
+HTTPS_PORT = ... # type: int
+
+CONTINUE = ... # type: int
+SWITCHING_PROTOCOLS = ... # type: int
+PROCESSING = ... # type: int
+
+OK = ... # type: int
+CREATED = ... # type: int
+ACCEPTED = ... # type: int
+NON_AUTHORITATIVE_INFORMATION = ... # type: int
+NO_CONTENT = ... # type: int
+RESET_CONTENT = ... # type: int
+PARTIAL_CONTENT = ... # type: int
+MULTI_STATUS = ... # type: int
+IM_USED = ... # type: int
+
+MULTIPLE_CHOICES = ... # type: int
+MOVED_PERMANENTLY = ... # type: int
+FOUND = ... # type: int
+SEE_OTHER = ... # type: int
+NOT_MODIFIED = ... # type: int
+USE_PROXY = ... # type: int
+TEMPORARY_REDIRECT = ... # type: int
+
+BAD_REQUEST = ... # type: int
+UNAUTHORIZED = ... # type: int
+PAYMENT_REQUIRED = ... # type: int
+FORBIDDEN = ... # type: int
+NOT_FOUND = ... # type: int
+METHOD_NOT_ALLOWED = ... # type: int
+NOT_ACCEPTABLE = ... # type: int
+PROXY_AUTHENTICATION_REQUIRED = ... # type: int
+REQUEST_TIMEOUT = ... # type: int
+CONFLICT = ... # type: int
+GONE = ... # type: int
+LENGTH_REQUIRED = ... # type: int
+PRECONDITION_FAILED = ... # type: int
+REQUEST_ENTITY_TOO_LARGE = ... # type: int
+REQUEST_URI_TOO_LONG = ... # type: int
+UNSUPPORTED_MEDIA_TYPE = ... # type: int
+REQUESTED_RANGE_NOT_SATISFIABLE = ... # type: int
+EXPECTATION_FAILED = ... # type: int
+UNPROCESSABLE_ENTITY = ... # type: int
+LOCKED = ... # type: int
+FAILED_DEPENDENCY = ... # type: int
+UPGRADE_REQUIRED = ... # type: int
+PRECONDITION_REQUIRED = ... # type: int
+TOO_MANY_REQUESTS = ... # type: int
+REQUEST_HEADER_FIELDS_TOO_LARGE = ... # type: int
+
+INTERNAL_SERVER_ERROR = ... # type: int
+NOT_IMPLEMENTED = ... # type: int
+BAD_GATEWAY = ... # type: int
+SERVICE_UNAVAILABLE = ... # type: int
+GATEWAY_TIMEOUT = ... # type: int
+HTTP_VERSION_NOT_SUPPORTED = ... # type: int
+INSUFFICIENT_STORAGE = ... # type: int
+NOT_EXTENDED = ... # type: int
+NETWORK_AUTHENTICATION_REQUIRED = ... # type: int
 
 responses = ... # type: Dict[int, str]
 
-class HTTPMessage(email.message.Message):
- def getallmatchingheaders(self, name): ...
-
-class HTTPResponse(io.RawIOBase):
- fp = ... # type: Any
- debuglevel = ... # type: Any
- headers = ... # type: Any
- version = ... # type: Any
- status = ... # type: Any
- reason = ... # type: Any
- chunked = ... # type: Any
- chunk_left = ... # type: Any
- length = ... # type: Any
- will_close = ... # type: Any
- def __init__(self, sock, debuglevel=..., method=..., url=...) -> None: ...
- code = ... # type: Any
- def begin(self): ...
- def close(self): ...
- def flush(self): ...
- def readable(self): ...
- def isclosed(self): ...
- def read(self, amt=...): ...
- def readinto(self, b): ...
- def fileno(self): ...
- def getheader(self, name, default=...): ...
- def getheaders(self): ...
- def __iter__(self): ...
- def info(self): ...
- def geturl(self): ...
- def getcode(self): ...
+class HTTPMessage(email.message.Message): ...
+
+# TODO uncomment when mypy handle conditionals
+#if sys.version_info >= (3, 5):
+# class HTTPResponse(io.BufferedIOBase):
+# msg = ... # type: HTTPMessage
+# version = ... # type: int
+# debuglevel = ... # type: int
+# closed = ... # type: bool
+# status = ... # type: int
+# reason = ... # type: str
+# def read(self, amt: Optional[int] = ...) -> bytes: ...
+# def readinto(self, b: bytearray) -> int: ...
+# @overload
+# def getheader(self, name: str) -> Optional[str]: ...
+# @overload
+# def getheader(self, name: str, default: _T) -> Union[str, _T]: ...
+# def getheaders(self) -> List[Tuple[str, str]]: ...
+# def fileno(self) -> int: ...
+# def __iter__(self) -> Iterator[bytes]: ...
+# def __enter__(self) -> 'HTTPResponse': ...
+# def __exit__(self, exc_type: Optional[type],
+# exc_val: Optional[Exception],
+# exc_tb: Optional[types.TracebackType]) -> bool: ...
+#else:
+# class HTTPResponse:
+# msg = ... # type: HTTPMessage
+# version = ... # type: int
+# debuglevel = ... # type: int
+# closed = ... # type: bool
+# status = ... # type: int
+# reason = ... # type: str
+# def read(self, amt: Optional[int] = ...) -> bytes: ...
+# if sys.version_info >= (3, 3):
+# def readinto(self, b: bytearray) -> int: ...
+# @overload
+# def getheader(self, name: str) -> Optional[str]: ...
+# @overload
+# def getheader(self, name: str, default: _T) -> Union[str, _T]: ...
+# def getheaders(self) -> List[Tuple[str, str]]: ...
+# def fileno(self) -> int: ...
+# def __iter__(self) -> Iterator[bytes]: ...
+# def __enter__(self) -> 'HTTPResponse': ...
+# def __exit__(self, exc_type: Optional[type],
+# exc_val: Optional[Exception],
+# exc_tb: Optional[types.TracebackType]) -> bool: ...
+class HTTPResponse(io.BufferedIOBase):
+ msg = ... # type: HTTPMessage
+ version = ... # type: int
+ debuglevel = ... # type: int
+ closed = ... # type: bool
+ status = ... # type: int
+ reason = ... # type: str
+ def read(self, amt: Optional[int] = ...) -> bytes: ...
+ def readinto(self, b: bytearray) -> int: ...
+ @overload
+ def getheader(self, name: str) -> Optional[str]: ...
+ @overload
+ def getheader(self, name: str, default: _T) -> Union[str, _T]: ...
+ def getheaders(self) -> List[Tuple[str, str]]: ...
+ def fileno(self) -> int: ...
+ def __iter__(self) -> Iterator[bytes]: ...
+ def __enter__(self) -> 'HTTPResponse': ...
+ def __exit__(self, exc_type: Optional[type],
+ exc_val: Optional[Exception],
+ exc_tb: Optional[types.TracebackType]) -> bool: ...
 
 class HTTPConnection:
- response_class = ... # type: Any
- default_port = ... # type: Any
- auto_open = ... # type: Any
- debuglevel = ... # type: Any
- mss = ... # type: Any
- timeout = ... # type: Any
- source_address = ... # type: Any
- sock = ... # type: Any
- def __init__(self, host, port=..., timeout=..., source_address=...) -> None: ...
- def set_tunnel(self, host, port=..., headers=...): ...
- def set_debuglevel(self, level): ...
- def connect(self): ...
- def close(self): ...
- def send(self, data): ...
- def putrequest(self, method, url, skip_host=..., skip_accept_encoding=...): ...
- def putheader(self, header, *values): ...
- def endheaders(self, message_body=...): ...
- def request(self, method, url, body=..., headers=...): ...
- def getresponse(self): ...
+ if sys.version_info >= (3, 4):
+ def __init__(self, # type: ignore
+ host: str, port: Optional[int] = ...,
+ timeout: int = ...,
+ source_address: Optional[Tuple[str, int]] = ...) \
+ -> None: ...
+ else:
+ def __init__(self, # type: ignore
+ host: str, port: Optional[int] = ...,
+ strict: bool = ..., timeout: int = ...,
+ source_address: Optional[Tuple[str, int]] = ...) \
+ -> None: ...
+ def request(self, method: str, url: str,
+ body: Optional[_DataType] = ...,
+ headers: Mapping[str, str] = ...) -> None: ...
+ def getresponse(self) -> HTTPResponse: ...
+ def set_debuglevel(self, level: int) -> None: ...
+ def set_tunnel(self, host: str, port: Optional[int] = ...,
+ headers: Optional[Mapping[str, str]] = ...) -> None: ...
+ def connect(self) -> None: ...
+ def close(self) -> None: ...
+ def putrequest(self, request: str, selector: str, skip_host: bool = ...,
+ skip_accept_encoding: bool = ...) -> None: ...
+ def putheader(self, header: str, *argument: str) -> None: ...
+ def endheaders(self, message_body: Optional[_DataType] = ...) -> None: ...
+ def send(self, data: _DataType) -> None: ...
 
 class HTTPSConnection(HTTPConnection):
- default_port = ... # type: Any
- key_file = ... # type: Any
- cert_file = ... # type: Any
- def __init__(self, host, port=..., key_file=..., cert_file=..., timeout=...,
- source_address=..., *, context=..., check_hostname=...): ...
- sock = ... # type: Any
- def connect(self): ...
+ if sys.version_info >= (3, 4):
+ def __init__(self, # type: ignore
+ host: str, port: Optional[int] = ...,
+ key_file: Optional[str] = ...,
+ cert_file: Optional[str] = ...,
+ timeout: int = ...,
+ source_address: Optional[Tuple[str, int]] = ...,
+ *, context: Optional[ssl.SSLContext] = ...,
+ check_hostname: Optional[bool] = ...) -> None: ...
+ else:
+ def __init__(self, # type: ignore
+ host: str, port: Optional[int] = ...,
+ key_file: Optional[str] = ...,
+ cert_file: Optional[str] = ...,
+ strict: bool = ..., timeout: int = ...,
+ source_address: Optional[Tuple[str, int]] = ...,
+ *, context: Optional[ssl.SSLContext] = ...,
+ check_hostname: Optional[bool] = ...) -> None: ...
 
 class HTTPException(Exception): ...
+
 class NotConnected(HTTPException): ...
 class InvalidURL(HTTPException): ...
-
-class UnknownProtocol(HTTPException):
- args = ... # type: Any
- version = ... # type: Any
- def __init__(self, version) -> None: ...
-
+class UnknownProtocol(HTTPException): ...
 class UnknownTransferEncoding(HTTPException): ...
 class UnimplementedFileMode(HTTPException): ...
-
-class IncompleteRead(HTTPException):
- args = ... # type: Any
- partial = ... # type: Any
- expected = ... # type: Any
- def __init__(self, partial, expected=...) -> None: ...
+class IncompleteRead(HTTPException): ...
 
 class ImproperConnectionState(HTTPException): ...
 class CannotSendRequest(ImproperConnectionState): ...
 class CannotSendHeader(ImproperConnectionState): ...
 class ResponseNotReady(ImproperConnectionState): ...
 
-class BadStatusLine(HTTPException):
- args = ... # type: Any
- line = ... # type: Any
- def __init__(self, line) -> None: ...
-
-class LineTooLong(HTTPException):
- def __init__(self, line_type) -> None: ...
+class BadStatusLine(HTTPException): ...
+class LineTooLong(HTTPException): ...
 
-error = HTTPException
+if sys.version_info >= (3, 5):
+ class RemoteDisconnected(ConnectionResetError, BadStatusLine): ...