daemon: Mypy fixes.

This commit is contained in:
Damir Jelić 2019-04-10 14:46:11 +02:00
parent 25a36a6d47
commit 78041c7a6a

View File

@ -43,7 +43,7 @@ class ProxyDaemon:
default_session = attr.ib(init=False, default=None) default_session = attr.ib(init=False, default=None)
def get_access_token(self, request): def get_access_token(self, request):
# type: (aiohttp.BaseRequest) -> str # type: (aiohttp.web.BaseRequest) -> str
"""Extract the access token from the request. """Extract the access token from the request.
This method extracts the access token either from the query string or This method extracts the access token either from the query string or
@ -63,11 +63,11 @@ class ProxyDaemon:
async def forward_request( async def forward_request(
self, self,
request, request, # type: aiohttp.web.BaseRequest
params=None, params=None, # type: CIMultiDict
session=None session=None # type: aiohttp.ClientSession
): ):
# type: (aiohttp.BaseRequest, aiohttp.ClientSession) -> str # type: (...) -> aiohttp.ClientResponse
"""Forward the given request to our configured homeserver. """Forward the given request to our configured homeserver.
Args: Args:
@ -82,6 +82,8 @@ class ProxyDaemon:
self.default_session = ClientSession() self.default_session = ClientSession()
session = self.default_session session = self.default_session
assert session
path = request.path path = request.path
method = request.method method = request.method