Use regex for matching placeholders

rdnetto · Aug 1, 2019 · 6959741 · 6959741
1 parent 590302c
commit 6959741
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/config_gen.py b/config_gen.py
@@ -453,14 +453,18 @@ def generate_ycm_conf(flags, config_file, template_file):
         with open(config_file, "w") as output:
             output.write("# Generated by YCM Generator at {}\n\n".format(str(datetime.datetime.today())))
 
+            placeholder_regex = re.compile(r"^(\s*)# INSERT FLAGS HERE$")
+
             for line in template:
-                if(line == "    # INSERT FLAGS HERE\n"):
+                match = placeholder_regex.match(line)
+                if(match):
                     # insert generated code
+                    indent = match.group(1)
                     for flag in flags:
                         if(isinstance(flag, basestring)):
-                            output.write("    '{}',\n".format(flag))
+                            output.write("{}'{}',\n".format(indent, flag))
                         else: # is tuple
-                            output.write("    '{}', '{}',\n".format(*flag))
+                            output.write("{}'{}', '{}',\n".format(indent, *flag))
 
                 else:
                     # copy template