More robustly escape input character to cope with special regexp inpu…

…ts like brackets
PierreMarchand20 · Jun 10, 2023 · 54a30fa · 54a30fa
1 parent eb5a913
commit 54a30fa
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/asciinema_automation/instruction.py b/asciinema_automation/instruction.py
@@ -1,5 +1,6 @@
 import logging
 import random
+import re
 import time
 
 logger = logging.getLogger(__name__)
@@ -55,11 +56,7 @@ def run(self, script):
         self.receive_value = self.send_value
 
         # Check for special character
-        if "\\" in self.send_value:
-            self.receive_value = [
-                character if character != "\\" else character + "\\"
-                for character in list(self.send_value)
-            ]
+        self.receive_value = [re.escape(c) for c in list(self.send_value)]
 
         # Write intruction
         for send_character, receive_character in zip(

diff --git a/examples/special_character.sh b/examples/special_character.sh
@@ -0,0 +1,2 @@
+#$ sendcharacter #
+This is a test with URL (https://github.com)
diff --git a/tests/reference_output/special_character.cast b/tests/reference_output/special_character.cast
@@ -0,0 +1,3 @@
+[?2004h$ #This is a test with URL (https://github.com)
+[?2004l[?2004h$ [?2004l
+exit
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		#$ sendcharacter #
		This is a test with URL (https://github.com)