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

Add invalid response object error handling

Browse Source
This commit is contained in:
Eli Uriegas 2016-12-25 10:37:23 -08:00
parent 6cf3754051
commit 68f6553181
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sanic/exceptions.py
View File

@ -21,6 +21,10 @@ class ServerError(SanicException):
status_code = 500 status_code = 500
class InvalidResponseObject(SanicException):
status_code = 500
class FileNotFound(NotFound): class FileNotFound(NotFound):
status_code = 404 status_code = 404

13
sanic/server.py
View File

@ -6,7 +6,7 @@ from signal import SIGINT, SIGTERM
from time import time from time import time
from httptools import HttpRequestParser from httptools import HttpRequestParser
from httptools.parser.errors import HttpParserError from httptools.parser.errors import HttpParserError
from .exceptions import ServerError from .exceptions import ServerError, InvalidResponseObject
try: try:
import uvloop as async_loop import uvloop as async_loop
@ -150,9 +150,14 @@ class HttpProtocol(asyncio.Protocol):
try: try:
keep_alive = self.parser.should_keep_alive() \ keep_alive = self.parser.should_keep_alive() \
and not self.signal.stopped and not self.signal.stopped
self.transport.write( try:
response.output( self.transport.write(
self.request.version, keep_alive, self.request_timeout)) response.output(
self.request.version, keep_alive,
self.request_timeout))
except AttributeError:
raise InvalidResponseObject(
'Invalid response object, must be of type HTTPResponse')
if not keep_alive: if not keep_alive:
self.transport.close() self.transport.close()
else: else: