aboutsummaryrefslogtreecommitdiff
path: root/markov/errors.py
diff options
context:
space:
mode:
Diffstat (limited to 'markov/errors.py')
-rw-r--r--markov/errors.py40
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
+ )