Skip to content

Commit 5366937

Browse files
committed
Use proper naming convention for Generic types
1 parent cbb9aab commit 5366937

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

trinity/protocol/common/managers.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@
2525
from .requests import BaseRequest
2626

2727

28-
PeerClass = TypeVar('PeerClass', bound=BasePeer)
29-
RequestType = TypeVar('RequestType', bound=BaseRequest)
30-
ResponseType = TypeVar('ResponseType')
31-
ReturnType = TypeVar('ReturnType')
28+
TPeer = TypeVar('TPeer', bound=BasePeer)
29+
TRequest = TypeVar('TRequest', bound=BaseRequest)
30+
TResponse = TypeVar('TResponse')
31+
TReturn = TypeVar('TReturn')
3232

3333

34-
class BaseRequestManager(PeerSubscriber, BaseService, Generic[PeerClass, RequestType, ResponseType, ReturnType]): # noqa: E501
34+
class BaseRequestManager(PeerSubscriber, BaseService, Generic[TPeer, TRequest, TResponse, TReturn]): # noqa: E501
3535
#
3636
# PeerSubscriber
3737
#
@@ -43,11 +43,11 @@ def subscription_msg_types(self) -> Set[Type[Command]]:
4343

4444
response_timout: int = 60
4545

46-
pending_request: Tuple[RequestType, 'asyncio.Future[ReturnType]'] = None
46+
pending_request: Tuple[TRequest, 'asyncio.Future[TReturn]'] = None
4747

48-
_peer: PeerClass
48+
_peer: TPeer
4949

50-
def __init__(self, peer: PeerClass, token: CancelToken) -> None:
50+
def __init__(self, peer: TPeer, token: CancelToken) -> None:
5151
self._peer = peer
5252
super().__init__(token)
5353

@@ -65,14 +65,14 @@ async def _run(self) -> None:
6565
self.logger.error("Unexpected peer: %s expected: %s", peer, self._peer)
6666
continue
6767
elif isinstance(cmd, self._response_msg_type):
68-
self._handle_msg(cast(ResponseType, msg))
68+
self._handle_msg(cast(TResponse, msg))
6969
else:
7070
self.logger.warning("Unexpected message type: %s", cmd.__class__.__name__)
7171

7272
async def _cleanup(self) -> None:
7373
pass
7474

75-
def _handle_msg(self, msg: ResponseType) -> None:
75+
def _handle_msg(self, msg: TResponse) -> None:
7676
if self.pending_request is None:
7777
self.logger.debug(
7878
"Got unexpected %s message from %", self.response_msg_name, self._peer
@@ -95,11 +95,11 @@ def _handle_msg(self, msg: ResponseType) -> None:
9595
self.pending_request = None
9696

9797
@abstractmethod
98-
def _normalize_response(self, msg: ResponseType) -> ReturnType:
98+
def _normalize_response(self, msg: TResponse) -> TReturn:
9999
pass
100100

101101
@abstractmethod
102-
def __call__(self) -> ReturnType:
102+
def __call__(self) -> TReturn:
103103
"""
104104
Subclasses must both implement this method and override the call
105105
signature to properly construct the `Request` object and pass it into
@@ -122,12 +122,12 @@ def response_msg_name(self) -> str:
122122
return self._response_msg_type.__name__
123123

124124
@abstractmethod
125-
def _send_sub_proto_request(self, request: RequestType) -> None:
125+
def _send_sub_proto_request(self, request: TRequest) -> None:
126126
pass
127127

128128
async def _wait_for_response(self,
129-
request: RequestType,
130-
timeout: int = None) -> ReturnType:
129+
request: TRequest,
130+
timeout: int = None) -> TReturn:
131131
if self.pending_request is not None:
132132
self.logger.error(
133133
"Already waiting for response to %s for peer: %s",
@@ -141,7 +141,7 @@ async def _wait_for_response(self,
141141
)
142142
)
143143

144-
future: 'asyncio.Future[ReturnType]' = asyncio.Future()
144+
future: 'asyncio.Future[TReturn]' = asyncio.Future()
145145
self.pending_request = (request, future)
146146

147147
try:
@@ -152,7 +152,7 @@ async def _wait_for_response(self,
152152

153153
return response
154154

155-
async def _request_and_wait(self, request: RequestType, timeout: int=None) -> ReturnType:
155+
async def _request_and_wait(self, request: TRequest, timeout: int=None) -> TReturn:
156156
if timeout is None:
157157
timeout = self.response_timout
158158
self._send_sub_proto_request(request)

0 commit comments

Comments
 (0)