Fixed string escaping in AstPrinter

3 years ago · 5fec9f5db9
parent 26acf2152d
commit 5fec9f5db9
1 changed files with 5 additions and 1 deletions
--- a/mesonbuild/ast/printer.py
+++ b/mesonbuild/ast/printer.py
@ -72,9 +72,13 @@ class AstPrinter(AstVisitor):
        self.append(str(node.value), node)
        node.lineno = self.curr_line or node.lineno

+    def escape(self, val: str) -> str:
+        return val.translate(str.maketrans({'\'': '\\\'',
+                                            '\\': '\\\\'}))
+
    def visit_StringNode(self, node: mparser.StringNode) -> None:
        assert isinstance(node.value, str)
-        self.append("'" + node.value + "'", node)
+        self.append("'" + self.escape(node.value) + "'", node)
        node.lineno = self.curr_line or node.lineno

    def visit_FormatStringNode(self, node: mparser.FormatStringNode) -> None: