diff options
author | Arjun Satarkar <me@arjunsatarkar.net> | 2024-03-20 03:14:38 +0000 |
---|---|---|
committer | Arjun Satarkar <me@arjunsatarkar.net> | 2024-03-20 03:14:38 +0000 |
commit | 08069e8d35e10a85087c1938fffb4f5a8c625e14 (patch) | |
tree | 7856852dd96d356aafcfa4536218d4df897ecb67 | |
parent | 16aeaa50db79e3c17194128a93d22370bf5daa33 (diff) | |
download | aps-cogs-08069e8d35e10a85087c1938fffb4f5a8c625e14.tar aps-cogs-08069e8d35e10a85087c1938fffb4f5a8c625e14.tar.gz aps-cogs-08069e8d35e10a85087c1938fffb4f5a8c625e14.zip |
markov: fix bug where append_token could clobber previous token
Specifically if we had an open paren followed by some other
punctuation, the open paren would disappear. Probably this didn't exist
before the addition of parens as valid tokens.
-rw-r--r-- | markov/markov.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/markov/markov.py b/markov/markov.py index 73ac08e..d7cf14b 100644 --- a/markov/markov.py +++ b/markov/markov.py @@ -164,12 +164,12 @@ class Markov(commands.Cog): def append_token(self, text, token): # NOTE: if changing PUNCTUATION, also change the regex in process_message() with the corresponding note PUNCTUATION = r".,!?/;()" - if token == "/": - text = text[:-1] + token - elif token == "(": + if token == "(": text += token + elif token == "/": + text = text.removesuffix(" ") + token elif token in PUNCTUATION: - text = text[:-1] + token + " " + text = text.removesuffix(" ") + token + " " else: text += token + " " return text |