sanic/sanic/exceptions.py

244 lines
5.3 KiB
Python
Raw Normal View History

2017-05-19 22:40:01 +01:00
from sanic.response import ALL_STATUS_CODES, COMMON_STATUS_CODES
2017-05-19 21:35:04 +01:00
2017-01-13 00:54:34 +00:00
TRACEBACK_STYLE = '''
<style>
body {
padding: 20px;
font-family: Arial, sans-serif;
}
p {
margin: 0;
}
.summary {
padding: 10px;
}
h1 {
margin-bottom: 0;
}
h3 {
margin-top: 10px;
}
h3 code {
font-size: 24px;
}
.frame-line > * {
padding: 5px 10px;
}
.frame-line {
margin-bottom: 5px;
}
.frame-code {
font-size: 16px;
padding-left: 30px;
}
.tb-wrapper {
border: 1px solid #f3f3f3;
}
.tb-header {
background-color: #f3f3f3;
padding: 5px 10px;
}
.tb-border {
padding-top: 20px;
}
2017-01-13 00:54:34 +00:00
.frame-descriptor {
background-color: #e2eafb;
}
.frame-descriptor {
font-size: 14px;
}
</style>
'''
TRACEBACK_WRAPPER_HTML = '''
<html>
<head>
{style}
</head>
<body>
{inner_html}
<div class="summary">
<p>
2017-01-13 00:54:34 +00:00
<b>{exc_name}: {exc_value}</b>
2017-03-03 16:49:35 +00:00
while handling path <code>{path}</code>
2017-01-13 00:54:34 +00:00
</p>
</div>
</body>
</html>
'''
TRACEBACK_WRAPPER_INNER_HTML = '''
<h1>{exc_name}</h1>
<h3><code>{exc_value}</code></h3>
<div class="tb-wrapper">
<p class="tb-header">Traceback (most recent call last):</p>
{frame_html}
</div>
'''
TRACEBACK_BORDER = '''
<div class="tb-border">
<b><i>
The above exception was the direct cause of the
following exception:
</i></b>
</div>
'''
2017-01-13 00:54:34 +00:00
TRACEBACK_LINE_HTML = '''
<div class="frame-line">
<p class="frame-descriptor">
File {0.filename}, line <i>{0.lineno}</i>,
in <code><b>{0.name}</b></code>
</p>
<p class="frame-code"><code>{0.line}</code></p>
</div>
'''
INTERNAL_SERVER_ERROR_HTML = '''
<h1>Internal Server Error</h1>
<p>
The server encountered an internal error and cannot complete
your request.
</p>
'''
2017-05-20 18:44:09 +01:00
_sanic_exceptions = {}
2017-05-20 09:21:09 +01:00
def add_status_code(code):
"""
2017-05-20 18:29:00 +01:00
Decorator used for adding exceptions to _sanic_exceptions.
2017-05-20 09:21:09 +01:00
"""
def class_decorator(cls):
cls.status_code = code
2017-05-20 18:29:00 +01:00
_sanic_exceptions[code] = cls
2017-05-20 09:21:09 +01:00
return cls
return class_decorator
2016-10-03 05:11:38 +01:00
class SanicException(Exception):
2017-02-09 01:59:34 +00:00
2017-05-19 22:40:01 +01:00
def __init__(self, message, status_code=None):
super().__init__(message)
if status_code is not None:
self.status_code = status_code
2016-10-03 05:11:38 +01:00
2017-05-20 09:21:09 +01:00
@add_status_code(404)
2016-10-03 05:11:38 +01:00
class NotFound(SanicException):
2017-05-21 03:30:08 +01:00
pass
2017-05-20 09:21:09 +01:00
@add_status_code(400)
2016-10-03 05:11:38 +01:00
class InvalidUsage(SanicException):
2017-05-21 03:30:08 +01:00
pass
2017-05-20 09:21:09 +01:00
@add_status_code(500)
2016-10-03 05:11:38 +01:00
class ServerError(SanicException):
2017-05-21 03:30:08 +01:00
pass
2017-05-21 03:30:08 +01:00
class URLBuildError(ServerError):
pass
2017-02-02 17:21:14 +00:00
class FileNotFound(NotFound):
2017-05-21 03:30:08 +01:00
pass
def __init__(self, message, path, relative_url):
super().__init__(message)
self.path = path
self.relative_url = relative_url
2017-05-20 09:21:09 +01:00
@add_status_code(408)
class RequestTimeout(SanicException):
2017-05-21 03:30:08 +01:00
pass
2017-05-20 09:21:09 +01:00
@add_status_code(413)
class PayloadTooLarge(SanicException):
2017-05-21 03:30:08 +01:00
pass
2017-02-09 01:59:34 +00:00
2017-05-21 03:30:08 +01:00
class HeaderNotFound(InvalidUsage):
pass
2017-05-20 09:21:09 +01:00
@add_status_code(416)
class ContentRangeError(SanicException):
2017-05-21 03:30:08 +01:00
pass
def __init__(self, message, content_range):
super().__init__(message)
self.headers = {
'Content-Type': 'text/plain',
"Content-Range": "bytes */%s" % (content_range.total,)
}
class InvalidRangeType(ContentRangeError):
pass
2017-05-19 22:40:01 +01:00
@add_status_code(401)
class Unauthorized(SanicException):
"""
Unauthorized exception (401 HTTP status code).
:param scheme: Name of the authentication scheme to be used.
:param realm: Description of the protected area. (optional)
:param others: A dict containing values to add to the WWW-Authenticate
header that is generated. This is especially useful when dealing with the
Digest scheme. (optional)
"""
pass
def __init__(self, message, scheme, realm="", others=None):
super().__init__(message)
adds = ""
if others is not None:
values = ["{!s}={!r}".format(k, v) for k, v in others.items()]
adds = ', '.join(values)
adds = ', {}'.format(adds)
self.headers = {
"WWW-Authenticate": "{} realm='{}'{}".format(scheme, realm, adds)
}
2017-05-19 22:40:01 +01:00
def abort(status_code, message=None):
"""
Raise an exception based on SanicException. Returns the HTTP response
message appropriate for the given status code, unless provided.
:param status_code: The HTTP status code to return.
:param message: The HTTP response body. Defaults to the messages
in response.py for the given status code.
"""
if message is None:
message = COMMON_STATUS_CODES.get(status_code,
ALL_STATUS_CODES.get(status_code))
2017-05-21 03:30:08 +01:00
# These are stored as bytes in the STATUS_CODES dict
message = message.decode('utf8')
2017-05-20 18:29:00 +01:00
sanic_exception = _sanic_exceptions.get(status_code, SanicException)
2017-05-19 22:40:01 +01:00
raise sanic_exception(message=message, status_code=status_code)