markov: improve exceptions

1 files changed, 36 insertions, 4 deletions

diff --git a/markov/errors.py b/markov/errors.py
index 1af7de8..875d3b7 100644
--- a/markov/errors.py
+++ b/markov/errors.py
@@ -1,14 +1,46 @@
 class MarkovGenerationError(Exception):
-    pass
+    def __init__(self, guild_id: int, member_id: int | None):
+        self.guild_id = guild_id
+        self.member_id = member_id
+
+    def __repr__(self):
+        return f"MarkovGenerationError(guild_id={self.guild_id}, member_id={self.member_id})"
 
 
 class NoTotalCompletionCountError(MarkovGenerationError):
-    pass
+    def __init__(self, guild_id: int, member_id: int | None, token: str):
+        super().__init__(guild_id, member_id)
+        self.token = token
+
+    def __repr__(self):
+        return (
+            "NoTotalCompletionCountError(guild_id={}, member_id={}, token={})".format(
+                self.guild_id, self.member_id, repr(self.token)
+            )
+        )
 
 
 class NoNextTokenError(MarkovGenerationError):
-    pass
+    def __init__(self, guild_id: int, member_id: int | None, token: str, offset: int):
+        super().__init__(guild_id, member_id)
+        self.token = token
+        self.offset = offset
+
+    def __repr__(self):
+        return (
+            "NoNextTokenError(guild_id={}, member_id={}, token={}, offset={})".format(
+                self.guild_id, self.member_id, repr(self.token), self.offset
+            )
+        )
 
 
 class InvalidCompletionCountError(MarkovGenerationError):
-    pass
+    def __init__(self, guild_id: int, member_id: int | None, token: str, offset: int):
+        super().__init__(guild_id, member_id)
+        self.token = token
+        self.offset = offset
+
+    def __repr__(self):
+        return "InvalidCompletionCountError(guild_id={}, member_id={}, token={}, offset={})".format(
+            self.guild_id, self.member_id, repr(self.token), self.offset
+        )