Open source

libraries/stripe/error.py

@@ -0,0 +1,74 @@
+# Exceptions
+import sys
+
+
+class StripeError(Exception):
+
+    def __init__(self, message=None, http_body=None, http_status=None,
+                 json_body=None, headers=None):
+        super(StripeError, self).__init__(message)
+
+        if http_body and hasattr(http_body, 'decode'):
+            try:
+                http_body = http_body.decode('utf-8')
+            except:
+                http_body = ('<Could not decode body as utf-8. '
+                             'Please report to support@stripe.com>')
+
+        self._message = message
+        self.http_body = http_body
+        self.http_status = http_status
+        self.json_body = json_body
+        self.headers = headers or {}
+        self.request_id = self.headers.get('request-id', None)
+
+    def __unicode__(self):
+        if self.request_id is not None:
+            msg = self._message or "<empty message>"
+            return u"Request {0}: {1}".format(self.request_id, msg)
+        else:
+            return self._message
+
+    if sys.version_info > (3, 0):
+        def __str__(self):
+            return self.__unicode__()
+    else:
+        def __str__(self):
+            return unicode(self).encode('utf-8')
+
+
+class APIError(StripeError):
+    pass
+
+
+class APIConnectionError(StripeError):
+    pass
+
+
+class CardError(StripeError):
+
+    def __init__(self, message, param, code, http_body=None,
+                 http_status=None, json_body=None, headers=None):
+        super(CardError, self).__init__(
+            message, http_body, http_status, json_body,
+            headers)
+        self.param = param
+        self.code = code
+
+
+class InvalidRequestError(StripeError):
+
+    def __init__(self, message, param, http_body=None,
+                 http_status=None, json_body=None, headers=None):
+        super(InvalidRequestError, self).__init__(
+            message, http_body, http_status, json_body,
+            headers)
+        self.param = param
+
+
+class AuthenticationError(StripeError):
+    pass
+
+
+class RateLimitError(StripeError):
+    pass