Fix parse errors when single quotes are present in a table name (pyth…

…on-poetry#71) Fixes python-poetry#70
capuanob · Feb 3, 2020 · cb6351e · cb6351e
1 parent d6674e7
commit cb6351e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/tests/examples/table_names_with_string_delimiters.toml b/tests/examples/table_names_with_string_delimiters.toml
@@ -0,0 +1,5 @@
+['Special "table"']
+foo = "bar"
+
+["BJ's Restaurant"]
+account = "dining"
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -53,6 +53,7 @@ def json_serial(obj):
         "newline_in_strings",
         "preserve_quotes_in_string",
         "string_slash_whitespace_newline",
+        "table_names_with_string_delimiters",
     ],
 )
 def test_parse_can_parse_valid_toml_files(example, example_name):

diff --git a/tomlkit/parser.py b/tomlkit/parser.py
@@ -219,7 +219,12 @@ def _split_table_name(self, name):  # type: (str) -> Generator[Key]
                 continue
             elif c in {"'", '"'}:
                 if in_name:
-                    if t == KeyType.Literal and c == '"':
+                    if (
+                        t == KeyType.Literal
+                        and c == '"'
+                        or t == KeyType.Basic
+                        and c == "'"
+                    ):
                         current += c
                         continue