diff options
Diffstat (limited to 'markov/errors.py')
-rw-r--r-- | markov/errors.py | 40 |
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 + ) |