Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Let black f*ckup the layout

Browse Source
This commit is contained in:
L. Kärkkäinen
2020-02-26 19:27:22 +02:00
parent 29c6f3c49f
commit 6d8f5988d3
1 changed files with 29 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
sanic/server.py
View File

@@ -52,6 +52,7 @@ class Status(enum.Enum):
HANDLER = 3 # Headers done, handler running
RESPONSE = 4 # Response headers sent
class HttpProtocol(asyncio.Protocol):
"""
This class provides a basic HTTP implementation of the sanic framework.
@@ -195,18 +196,20 @@ class HttpProtocol(asyncio.Protocol):
"""Runs itself once a second to enforce any expired timeouts."""
duration = current_time() - self._time
status = self._status
if (status == Status.IDLE and duration > self.keep_alive_timeout):
if status == Status.IDLE and duration > self.keep_alive_timeout:
logger.debug("KeepAlive Timeout. Closing connection.")
elif (status == Status.REQUEST and duration > self.request_timeout):
elif status == Status.REQUEST and duration > self.request_timeout:
self._exception = RequestTimeout("Request Timeout")
elif (status.value > Status.REQUEST.value and duration > self.response_timeout):
elif (
status.value > Status.REQUEST.value
and duration > self.response_timeout
):
self._exception = ServiceUnavailable("Response Timeout")
else:
self.loop.call_later(.1, self.check_timeouts)
self.loop.call_later(0.1, self.check_timeouts)
return
self._task.cancel()
async def http1(self):
"""HTTP 1.1 connection handler"""
try:
@@ -256,9 +259,8 @@ class HttpProtocol(asyncio.Protocol):
if headers.get("connection", "").lower() == "close":
self.keep_alive = False
# Prepare for request body
body = (
headers.get("transfer-encoding") == "chunked"
or int(headers.get("content-length", 0))
body = headers.get("transfer-encoding") == "chunked" or int(
headers.get("content-length", 0)
)
self._request_chunked = False
self._request_bytes_left = 0
@@ -281,13 +283,16 @@ class HttpProtocol(asyncio.Protocol):
)
# Consume any remaining request body
if self._request_bytes_left or self._request_chunked:
logger.error(f"Handler of {method} {self.url} did not consume request body.")
while await self.stream_body(): pass
logger.error(
f"Handler of {method} {self.url} did not consume request body."
)
while await self.stream_body():
pass
self._status, self._time = Status.IDLE, current_time()
except asyncio.CancelledError:
self.write_error(
self._exception or
ServiceUnavailable("Request handler cancelled")
self._exception
or ServiceUnavailable("Request handler cancelled")
)
except SanicException as e:
self.write_error(e)
@@ -391,7 +396,9 @@ class HttpProtocol(asyncio.Protocol):
self._status, self._time = Status.RESPONSE, current_time()
self._last_response_time = self._time
self.transport.write(
response.output("1.1", self.keep_alive, self.keep_alive_timeout)
response.output(
"1.1", self.keep_alive, self.keep_alive_timeout
)
)
self.log_response(response)
except AttributeError:
@@ -436,7 +443,9 @@ class HttpProtocol(asyncio.Protocol):
try:
self._status, self._time = Status.RESPONSE, current_time()
response.protocol = self
await response.stream("1.1", self.keep_alive, self.keep_alive_timeout)
await response.stream(
"1.1", self.keep_alive, self.keep_alive_timeout
)
self.log_response(response)
except AttributeError:
logger.error(