Fix to allow escaping of the <> tags in f-string injections. (#463)

Authored-by: Nabeel Ansari <[email protected]>
nojanath · Aug 13, 2024 · b73d32c · b73d32c
1 parent d06e8e8
commit b73d32c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compiler/ksp_compiler.py b/compiler/ksp_compiler.py
@@ -375,7 +375,7 @@ def process_f_string(line):
 
         for i, c in enumerate(line):
             if record_arg == True:
-                if c == '>':
+                if c == '>' and not escaping:
                     record_arg = False
                     escaping = False
                     all_args.append(arg_content)
@@ -416,7 +416,7 @@ def process_f_string(line):
             deleted += 1
 
         for a in all_args:
-            new_line = new_line.replace("<{}>".format(a), "\' & {} & \'".format(a))
+            new_line = new_line.replace("<{}>".format(a), "\' & {} & \'".format(a.replace('\\>', '>').replace(('\\<', '<'))))
 
         return new_line