Merge pull request #697 from sinwoobang/master

Add new approved http status 451

Author: asvetlov

aiohttp/web_exceptions.py

@@ -45,6 +45,7 @@
     'HTTPPreconditionRequired',
     'HTTPTooManyRequests',
     'HTTPRequestHeaderFieldsTooLarge',
+    'HTTPUnavailableForLegalReasons',
     'HTTPServerError',
     'HTTPInternalServerError',
     'HTTPNotImplemented',
@@ -286,6 +287,18 @@ class HTTPRequestHeaderFieldsTooLarge(HTTPClientError):
     status_code = 431
 
 
+class HTTPUnavailableForLegalReasons(HTTPClientError):
+    status_code = 451
+
+    def __init__(self, link=None, *, headers=None, reason=None,
+                 body=None, text=None, content_type=None):
+        super().__init__(headers=headers, reason=reason,
+                         body=body, text=text, content_type=content_type)
+        if link:
+            self.headers['Link'] = '<%s>; rel="blocked-by"' % link
+            self.link = link
+
+
 ############################################################
 # 5xx Server Error
 ############################################################

tests/test_web_exceptions.py

@@ -163,3 +163,10 @@ def test_empty_body_205():
 def test_empty_body_304():
     resp = web.HTTPNoContent()
     resp.body is None
+
+
+def test_link_header_451(buf, request):
+    resp = web.HTTPUnavailableForLegalReasons(link='http://warning.or.kr/')
+
+    assert 'http://warning.or.kr/' == resp.link
+    assert '<http://warning.or.kr/>; rel="blocked-by"' == resp.headers['Link']