Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 1 | #!/usr/bin/python2.4 |
| 2 | # |
Joe Gregorio | 20a5aa9 | 2011-04-01 17:44:25 -0400 | [diff] [blame] | 3 | # Copyright (C) 2010 Google Inc. |
| 4 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | # you may not use this file except in compliance with the License. |
| 7 | # You may obtain a copy of the License at |
| 8 | # |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | # |
| 11 | # Unless required by applicable law or agreed to in writing, software |
| 12 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 16 | |
| 17 | """Errors for the library. |
| 18 | |
| 19 | All exceptions defined by the library |
| 20 | should be defined in this file. |
| 21 | """ |
| 22 | |
| 23 | __author__ = 'jcgregorio@google.com (Joe Gregorio)' |
| 24 | |
| 25 | |
Ali Afshar | 2dcc652 | 2010-12-16 10:11:53 +0100 | [diff] [blame] | 26 | from anyjson import simplejson |
| 27 | |
| 28 | |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 29 | class Error(Exception): |
| 30 | """Base error for this module.""" |
| 31 | pass |
| 32 | |
| 33 | |
| 34 | class HttpError(Error): |
| 35 | """HTTP data was invalid or unexpected.""" |
| 36 | |
Joe Gregorio | 4939655 | 2011-03-08 10:39:00 -0500 | [diff] [blame] | 37 | def __init__(self, resp, content, uri=None): |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 38 | self.resp = resp |
Ali Afshar | 2dcc652 | 2010-12-16 10:11:53 +0100 | [diff] [blame] | 39 | self.content = content |
Joe Gregorio | 4939655 | 2011-03-08 10:39:00 -0500 | [diff] [blame] | 40 | self.uri = uri |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 41 | |
Ali Afshar | 2dcc652 | 2010-12-16 10:11:53 +0100 | [diff] [blame] | 42 | def _get_reason(self): |
Joe Gregorio | fdf7c80 | 2011-06-30 12:33:38 -0400 | [diff] [blame] | 43 | """Calculate the reason for the error from the response content.""" |
Ali Afshar | 2dcc652 | 2010-12-16 10:11:53 +0100 | [diff] [blame] | 44 | if self.resp.get('content-type', '').startswith('application/json'): |
| 45 | try: |
| 46 | data = simplejson.loads(self.content) |
| 47 | reason = data['error']['message'] |
| 48 | except (ValueError, KeyError): |
| 49 | reason = self.content |
| 50 | else: |
| 51 | reason = self.resp.reason |
| 52 | return reason |
| 53 | |
| 54 | def __repr__(self): |
Joe Gregorio | 4939655 | 2011-03-08 10:39:00 -0500 | [diff] [blame] | 55 | if self.uri: |
| 56 | return '<HttpError %s when requesting %s returned "%s">' % ( |
| 57 | self.resp.status, self.uri, self._get_reason()) |
| 58 | else: |
| 59 | return '<HttpError %s "%s">' % (self.resp.status, self._get_reason()) |
Ali Afshar | 2dcc652 | 2010-12-16 10:11:53 +0100 | [diff] [blame] | 60 | |
| 61 | __str__ = __repr__ |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 62 | |
| 63 | |
Joe Gregorio | 4939655 | 2011-03-08 10:39:00 -0500 | [diff] [blame] | 64 | class InvalidJsonError(Error): |
| 65 | """The JSON returned could not be parsed.""" |
| 66 | pass |
| 67 | |
| 68 | |
Joe Gregorio | 3ad5e9a | 2010-12-09 15:01:04 -0500 | [diff] [blame] | 69 | class UnknownLinkType(Error): |
| 70 | """Link type unknown or unexpected.""" |
| 71 | pass |
Joe Gregorio | fdf7c80 | 2011-06-30 12:33:38 -0400 | [diff] [blame] | 72 | |
Joe Gregorio | a388ce3 | 2011-09-09 17:19:13 -0400 | [diff] [blame] | 73 | |
Joe Gregorio | fdf7c80 | 2011-06-30 12:33:38 -0400 | [diff] [blame] | 74 | class UnacceptableMimeTypeError(Error): |
| 75 | """That is an unacceptable mimetype for this operation.""" |
| 76 | pass |
| 77 | |
Joe Gregorio | a388ce3 | 2011-09-09 17:19:13 -0400 | [diff] [blame] | 78 | |
Joe Gregorio | fdf7c80 | 2011-06-30 12:33:38 -0400 | [diff] [blame] | 79 | class MediaUploadSizeError(Error): |
| 80 | """Media is larger than the method can accept.""" |
| 81 | pass |
Joe Gregorio | a388ce3 | 2011-09-09 17:19:13 -0400 | [diff] [blame] | 82 | |
| 83 | |
| 84 | class UnexpectedMethodError(Error): |
| 85 | """Exception raised by RequestMockBuilder on unexpected calls.""" |
| 86 | |
| 87 | def __init__(self, methodId=None): |
| 88 | """Constructor for an UnexpectedMethodError.""" |
| 89 | super(UnexpectedMethodError, self).__init__( |
| 90 | 'Received unexpected call %s' % methodId) |
| 91 | |
| 92 | |
| 93 | class UnexpectedBodyError(Error): |
| 94 | """Exception raised by RequestMockBuilder on unexpected bodies.""" |
| 95 | |
| 96 | def __init__(self, expected, provided): |
| 97 | """Constructor for an UnexpectedMethodError.""" |
| 98 | super(UnexpectedBodyError, self).__init__( |
| 99 | 'Expected: [%s] - Provided: [%s]' % (expected, provided)) |