From 26527d88ddc886b5369fcfa9a134a660e0b77ef3 Mon Sep 17 00:00:00 2001
From: Gabriel Gerlero <gerlero@users.noreply.github.com>
Date: Sat, 2 Nov 2024 14:41:36 -0300
Subject: [PATCH] Improve file writing

---
 foamlib/_files/_files.py | 31 +++++++++++++++----------------
 1 file changed, 15 insertions(+), 16 deletions(-)

diff --git a/foamlib/_files/_files.py b/foamlib/_files/_files.py
index 503956f..da87be4 100644
--- a/foamlib/_files/_files.py
+++ b/foamlib/_files/_files.py
@@ -250,21 +250,20 @@ def __setitem__(
 
             start, end = parsed.entry_location(keywords, missing_ok=True)
 
-            before = b""
-            if parsed.contents[:start] and not parsed.contents[:start].endswith(b"\n"):
-                before = b"\n"
-            if (
-                parsed.contents[:start]
-                and len(keywords) <= 1
-                and not parsed.contents[:start].endswith(b"\n\n")
-            ):
-                before = b"\n\n"
+            if start and not parsed.contents[:start].endswith(b"\n\n"):
+                if parsed.contents[:start].endswith(b"\n"):
+                    before = b"\n" if len(keywords) <= 1 else b""
+                else:
+                    before = b"\n\n" if len(keywords) <= 1 else b"\n"
+            else:
+                before = b""
 
-            after = b""
-            if parsed.contents[end:].startswith(b"}"):
-                after = b"    " * (len(keywords) - 2)
-            if not parsed.contents[end:] or not parsed.contents[end:].startswith(b"\n"):
-                after = b"\n" + after
+            if not parsed.contents[end:].strip() or parsed.contents[end:].startswith(
+                b"}"
+            ):
+                after = b"\n" + b"    " * (len(keywords) - 2)
+            else:
+                after = b""
 
             indentation = b"    " * (len(keywords) - 1)
 
@@ -292,7 +291,7 @@ def __setitem__(
             elif keywords:
                 parsed.put(
                     keywords,
-                    data,
+                    deepcopy(data),
                     before
                     + indentation
                     + dumps(keywords[-1])
@@ -303,7 +302,7 @@ def __setitem__(
                 )
 
             else:
-                parsed.put(keywords, data, before + dumps(data, kind=kind) + after)
+                parsed.put((), deepcopy(data), before + dumps(data, kind=kind) + after)
 
     def __delitem__(self, keywords: str | tuple[str, ...] | None) -> None:
         if not keywords: