better cleanup_toml

Signed-off-by: Kenneth Reitz <me@kennethreitz.org>
2026-06-05 22:50:18 +00:00 · 2017-10-06 10:02:53 -04:00
parent 73574898de
commit a5d29418c0
1 changed files with 8 additions and 2 deletions
@@ -276,12 +276,18 @@ def cleanup_toml(tml):

    # Add newlines between TOML sections.
    for i, line in enumerate(toml.split('\n')):
+        after = False
        # Skip the first line.
-        if i > 0:
-            if line.startswith('['):
+        if line.startswith('['):
+            if i > 0:
+                # Insert a newline before the heading.
                new_toml.append('\n')
+            after = True

        new_toml.append(line)
+        # Insert a newline after the heading.
+        if after:
+            new_toml.append('')

    toml = '\n'.join(new_toml)
    return toml