encode/httpx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor to use isinstance(..., typing.Iterator) (#1282)

Browse Source
This commit is contained in:
Tom Christie 2020-09-11 14:37:51 +01:00 committed by GitHub
parent 5ee6135256
commit 8a5050ea41
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
httpx/_content_streams.py
View File

@ -380,11 +380,9 @@ def encode(
return URLEncodedStream(data=data)
elif isinstance(data, (str, bytes)):
return ByteStream(body=data)
elif hasattr(data, "__aiter__"):
data = typing.cast(typing.AsyncIterator[bytes], data)
elif isinstance(data, typing.AsyncIterator):
return AsyncIteratorStream(aiterator=data)
elif hasattr(data, "__iter__"):
data = typing.cast(typing.Iterator[bytes], data)
elif isinstance(data, typing.Iterator):
return IteratorStream(iterator=data)
raise TypeError(f"Unexpected type for 'data', {type(data)!r}")
@ -395,11 +393,9 @@ def encode_response(content: ResponseContent = None) -> ContentStream:
return ByteStream(b"")
elif isinstance(content, bytes):
return ByteStream(body=content)
elif hasattr(content, "__aiter__"):
content = typing.cast(typing.AsyncIterator[bytes], content)
elif isinstance(content, typing.AsyncIterator):
return AsyncIteratorStream(aiterator=content)
elif hasattr(content, "__iter__"):
content = typing.cast(typing.Iterator[bytes], content)
elif isinstance(content, typing.Iterator):
return IteratorStream(iterator=content)
raise TypeError(f"Unexpected type for 'content', {type(content)!r}")