blob: c01791291f644ed107baffa4ac163e79194dc9a4 [file] [log] [blame]
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -05001#!/usr/bin/python2.4
2#
Joe Gregorio20a5aa92011-04-01 17:44:25 -04003# 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 Gregorio3ad5e9a2010-12-09 15:01:04 -050016
17"""Errors for the library.
18
19All exceptions defined by the library
20should be defined in this file.
21"""
22
23__author__ = 'jcgregorio@google.com (Joe Gregorio)'
24
25
Ali Afshar2dcc6522010-12-16 10:11:53 +010026from anyjson import simplejson
27
28
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -050029class Error(Exception):
30 """Base error for this module."""
31 pass
32
33
34class HttpError(Error):
35 """HTTP data was invalid or unexpected."""
36
Joe Gregorio49396552011-03-08 10:39:00 -050037 def __init__(self, resp, content, uri=None):
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -050038 self.resp = resp
Ali Afshar2dcc6522010-12-16 10:11:53 +010039 self.content = content
Joe Gregorio49396552011-03-08 10:39:00 -050040 self.uri = uri
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -050041
Ali Afshar2dcc6522010-12-16 10:11:53 +010042 def _get_reason(self):
Joe Gregoriofdf7c802011-06-30 12:33:38 -040043 """Calculate the reason for the error from the response content."""
Ali Afshar2dcc6522010-12-16 10:11:53 +010044 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 Gregorio49396552011-03-08 10:39:00 -050055 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 Afshar2dcc6522010-12-16 10:11:53 +010060
61 __str__ = __repr__
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -050062
63
Joe Gregorio49396552011-03-08 10:39:00 -050064class InvalidJsonError(Error):
65 """The JSON returned could not be parsed."""
66 pass
67
68
Joe Gregorio3ad5e9a2010-12-09 15:01:04 -050069class UnknownLinkType(Error):
70 """Link type unknown or unexpected."""
71 pass
Joe Gregoriofdf7c802011-06-30 12:33:38 -040072
Joe Gregorioa388ce32011-09-09 17:19:13 -040073
Joe Gregoriofdf7c802011-06-30 12:33:38 -040074class UnacceptableMimeTypeError(Error):
75 """That is an unacceptable mimetype for this operation."""
76 pass
77
Joe Gregorioa388ce32011-09-09 17:19:13 -040078
Joe Gregoriofdf7c802011-06-30 12:33:38 -040079class MediaUploadSizeError(Error):
80 """Media is larger than the method can accept."""
81 pass
Joe Gregorioa388ce32011-09-09 17:19:13 -040082
83
84class 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
93class 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))