1 files changed, 76 insertions, 0 deletions
diff --git a/bbb/exceptions.py b/bbb/exceptions.py
new file mode 100644
index 0000000..9460387
--- /dev/null
+++ b/bbb/exceptions.py
@@ -0,0 +1,76 @@
+
+import re
+
+class WikibaseException(Exception):
+    """Generic base class for Wikibase API errors"""
+
+    def __init__(self, err):
+        self.error = err
+
+    def __repr__(self):
+        return self.__unicode__()
+
+    def __str__(self):
+        return self.__unicode__()
+
+    def __unicode__(self):
+        return self.error
+
+class WikibaseAccountError(WikibaseException):
+
+    def __init__(self, user, error):
+        self.user = user
+        self.error = error
+
+    def __unicode__(self):
+        return "User '%s' had error: %s" % (self.user, self.error)
+
+class WikibaseAPIError(WikibaseException):
+
+    def __init__(self, code, info, action):
+        self.code = code
+        self.info = info
+        self.action = action
+
+    def __unicode__(self):
+        return "Wikibase server returned error for action '%s': %s" % (
+            self.action, self.code)
+
+class MissingEntityError(WikibaseException):
+
+    def __init__(self, id=None, title=None, info=None):
+        self.what = "Entity"
+        self.id = id
+        self.title = title
+        if info:
+            found_id = re.search("\(Invalid id: (.*)\)", info)
+            if found_id:
+                self.id = found_id.groups()[0]
+            found_title = re.search("\(Invalid title: (.*)\)", info)
+            if found_title:
+                self.title = found_title.groups()[0]
+
+    def __unicode__(self):
+        if self.id:
+            return "Couldn't find %s with id: %s" % (self.what, self.id)
+        elif self.title:
+            return "Couldn't find %s with title: %s" % (self.what, self.title)
+        else:
+            return "Couldn't find %s (unknown)" % self.what
+
+class MissingItemError(MissingEntityError):
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.what = "Item"
+
+class MissingPropertyError(MissingEntityError):
+    pass
+
+class APITimeoutError(WikibaseException):
+
+    def __init__(self, query):
+        self.query
+
+    def __unicode__(self):
+        return "HTTP (or HTTPS) request timed out: %s" % self.query