python/qmp: return generic type from context manager
__enter__ can be invoked from a subclass, so it needs a more flexible type. Signed-off-by: John Snow <jsnow@redhat.com> Message-id: 20210607200649.1840382-31-jsnow@redhat.com Signed-off-by: John Snow <jsnow@redhat.com>
This commit is contained in:
parent
6e24a7edb8
commit
eac8aabc92
|
@ -30,6 +30,7 @@ from typing import (
|
||||||
TextIO,
|
TextIO,
|
||||||
Tuple,
|
Tuple,
|
||||||
Type,
|
Type,
|
||||||
|
TypeVar,
|
||||||
Union,
|
Union,
|
||||||
cast,
|
cast,
|
||||||
)
|
)
|
||||||
|
@ -220,7 +221,9 @@ class QEMUMonitorProtocol:
|
||||||
if ret is None:
|
if ret is None:
|
||||||
raise QMPConnectError("Error while reading from socket")
|
raise QMPConnectError("Error while reading from socket")
|
||||||
|
|
||||||
def __enter__(self) -> 'QEMUMonitorProtocol':
|
T = TypeVar('T')
|
||||||
|
|
||||||
|
def __enter__(self: T) -> T:
|
||||||
# Implement context manager enter function.
|
# Implement context manager enter function.
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue