Merge pull request #98 from apollo13/patch-1

Adjust typing stubs to match type hints in trio itself.

Author: tjstum

trio-stubs/__init__.pyi

@@ -336,7 +336,7 @@ class StapledStream(trio.abc.HalfCloseableStream):
     send_stream: trio.abc.SendStream = attr.ib()
     receive_stream: trio.abc.ReceiveStream = attr.ib()
     async def aclose(self) -> None: ...
-    async def send_all(self, data: Union[bytes, memoryview]) -> None: ...
+    async def send_all(self, data: Union[bytes, bytearray, memoryview]) -> None: ...
     async def wait_send_all_might_not_block(self) -> None: ...
     async def receive_some(self, max_bytes: Optional[int] = ...) -> bytes: ...
     async def send_eof(self) -> None: ...
@@ -413,7 +413,7 @@ class SocketStream(trio.abc.HalfCloseableStream):
     @overload
     def getsockopt(self, level: int, option: int, buffersize: int) -> bytes: ...
     async def aclose(self) -> None: ...
-    async def send_all(self, data: Union[bytes, memoryview]) -> None: ...
+    async def send_all(self, data: Union[bytes, bytearray, memoryview]) -> None: ...
     async def wait_send_all_might_not_block(self) -> None: ...
     async def receive_some(self, max_bytes: Optional[int] = ...) -> bytes: ...
     async def send_eof(self) -> None: ...
@@ -829,7 +829,7 @@ class SSLStream(trio.abc.Stream):
     async def do_handshake(self) -> None: ...
     async def unwrap(self) -> Tuple[trio.abc.Stream, bytes]: ...
     async def aclose(self) -> None: ...
-    async def send_all(self, data: Union[bytes, memoryview]) -> None: ...
+    async def send_all(self, data: Union[bytes, bytearray, memoryview]) -> None: ...
     async def wait_send_all_might_not_block(self) -> None: ...
     async def receive_some(self, max_bytes: Optional[int] = ...) -> bytes: ...
 

trio-stubs/abc.pyi

@@ -63,7 +63,7 @@ class AsyncResource(metaclass=ABCMeta):
 
 class SendStream(AsyncResource):
     @abstractmethod
-    async def send_all(self, data: Union[bytes, memoryview]) -> None: ...
+    async def send_all(self, data: Union[bytes, bytearray, memoryview]) -> None: ...
     @abstractmethod
     async def wait_send_all_might_not_block(self) -> None: ...
 

trio-stubs/lowlevel.pyi

@@ -307,7 +307,7 @@ else:
 @final
 class FdStream(trio.abc.Stream):
     def __init__(self, fd: int): ...
-    async def send_all(self, data: Union[bytes, memoryview]) -> None: ...
+    async def send_all(self, data: Union[bytes, bytearray, memoryview]) -> None: ...
     async def wait_send_all_might_not_block(self) -> None: ...
     async def receive_some(self, max_bytes: Optional[int] = ...) -> bytes: ...
     async def aclose(self) -> None: ...

trio-stubs/socket.pyi

@@ -441,7 +441,7 @@ class SocketType:
     async def sendmsg(
         self,
         buffers: Iterable[ReadableBuffer],
-        ancdata: Iterable[Tuple[int, int, Union[bytes, memoryview]]] = ...,
+        ancdata: Iterable[Tuple[int, int, Union[bytes, bytearray, memoryview]]] = ...,
         flags: int = ...,
         address: Union[Tuple[Any, ...], str] = ...,
     ) -> int: ...