LBRYCommunity/aioupnp
1
0
Fork 0
Code Issues 7 Pull requests 1 Projects Releases 1 Packages Wiki Activity

fix socket type for python 3.8

Browse source
This commit is contained in:
Jack Robison 2020-10-30 12:51:13 -04:00
parent d8098d8315
commit a33a3a3a83
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aioupnp/protocols/multicast.py
View file

@ -3,6 +3,7 @@ import socket
import typing import typing
from asyncio.protocols import DatagramProtocol from asyncio.protocols import DatagramProtocol
from asyncio.transports import DatagramTransport from asyncio.transports import DatagramTransport
from asyncio.trsock import TransportSocket
from unittest import mock from unittest import mock
@ -10,7 +11,7 @@ def _get_sock(transport: typing.Optional[DatagramTransport]) -> typing.Optional[
if transport is None or not hasattr(transport, "_extra"): if transport is None or not hasattr(transport, "_extra"):
return None return None
sock: typing.Optional[socket.socket] = transport.get_extra_info('socket', None) sock: typing.Optional[socket.socket] = transport.get_extra_info('socket', None)
assert sock is None or isinstance(sock, (socket.SocketType, mock.MagicMock)) assert sock is None or isinstance(sock, (socket.SocketType, TransportSocket, mock.MagicMock))
return sock return sock