Use a dictionary to store escaped characters and their

corresponding escape forms. If you need to add new escape characters later, just add them to the dictionary Signed-off-by: changxuqing <[email protected]>
avocado-framework · Jun 25, 2024 · 478126b · 478126b
1 parent 679d622
commit 478126b
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/avocado/utils/astring.py b/avocado/utils/astring.py
@@ -89,10 +89,14 @@ def shell_escape(command):
 
     See also: http://www.tldp.org/LDP/abs/html/escapingsection.html
     """
-    command = command.replace("\\", "\\\\")
-    command = command.replace("$", r"\$")
-    command = command.replace('"', r"\"")
-    command = command.replace("`", r"\`")
+    escape_chars = {
+        "\\": "\\\\",
+        "$": r"\$",
+        '"': r'\"',
+        "`": r"\`"
+    }
+    for char, escaped_char in escape_chars.items():
+        command = command.replace(char, escaped_char)
     return command