Procházet zdrojové kódy

Merge pull request #90 from SillyLossy/regexp

Escape regexp in message extraction
oobabooga před 3 roky
rodič
revize
c92f347c94
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      server.py

+ 2 - 2
server.py

@@ -559,8 +559,8 @@ def extract_message_from_reply(question, reply, current, other, check, extension
     next_character_found = False
     substring_found = False
 
-    previous_idx = [m.start() for m in re.finditer(f"(^|\n){current}:", question)]
-    idx = [m.start() for m in re.finditer(f"(^|\n){current}:", reply)]
+    previous_idx = [m.start() for m in re.finditer(f"(^|\n){re.escape(current)}:", question)]
+    idx = [m.start() for m in re.finditer(f"(^|\n){re.escape(current)}:", reply)]
     idx = idx[len(previous_idx)-1]
 
     if extensions: