fix(#283): Wrong result stringifying subtables and nested arrays of tables, in arrays of tables

Close #283

Signed-off-by: Frost Ming <[email protected]>

Author: frostming

tests/test_write.py

@@ -24,3 +24,14 @@ def test_write_inline_table_in_nested_arrays():
     expected = "foo = [[{a = 1}]]\n"
     assert expected == dumps(d)
     assert loads(dumps(d))["foo"] == [[{"a": 1}]]
+
+
+def test_serialize_aot_with_nested_tables():
+    doc = {"a": [{"b": {"c": 1}}]}
+    expected = """\
+[[a]]
+[a.b]
+c = 1
+"""
+    assert dumps(doc) == expected
+    assert loads(expected) == doc

tomlkit/container.py

@@ -549,21 +549,18 @@ def _render_aot(self, key, aot, prefix=None):
 
     def _render_aot_table(self, table: Table, prefix: str | None = None) -> str:
         cur = ""
-
         _key = prefix or ""
+        open_, close = "[[", "]]"
 
-        if not table.is_super_table():
-            open_, close = "[[", "]]"
-
-            cur += (
-                f"{table.trivia.indent}"
-                f"{open_}"
-                f"{decode(_key)}"
-                f"{close}"
-                f"{table.trivia.comment_ws}"
-                f"{decode(table.trivia.comment)}"
-                f"{table.trivia.trail}"
-            )
+        cur += (
+            f"{table.trivia.indent}"
+            f"{open_}"
+            f"{decode(_key)}"
+            f"{close}"
+            f"{table.trivia.comment_ws}"
+            f"{decode(table.trivia.comment)}"
+            f"{table.trivia.trail}"
+        )
 
         for k, v in table.value.body:
             if isinstance(v, Table):