fix: escape backslashes in basic strings

frostming · frostming · commit b277474d3fe6 · 2022-10-27T15:50:53.000+08:00
Resolve #245
diff --git a/tests/test_items.py b/tests/test_items.py
@@ -887,3 +887,11 @@ def test_copy_copy():
     classifiers = result["tool"]["poetry"]["classifiers"]
     new = copy.copy(classifiers)
     assert new == classifiers
+
+
+@pytest.mark.parametrize(
+    "key_str,escaped",
+    [("\\", '"\\\\"'), ('"', '"\\""'), ("\t", '"\\t"'), ("\x10", '"\\u0010"')],
+)
+def test_escape_key(key_str, escaped):
+    assert api.key(key_str).as_string() == escaped
diff --git a/tomlkit/_utils.py b/tomlkit/_utils.py
@@ -113,7 +113,7 @@ def parse_rfc3339(string: str) -> Union[datetime, date, time]:
     **{v: f"\\{k}" for k, v in _escaped.items()},
     '"""': '""\\"',
 }
-_basic_escapes = CONTROL_CHARS | {'"'}
+_basic_escapes = CONTROL_CHARS | {'"', "\\"}
 
 
 def _unicode_escape(seq: str) -> str:

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ def parse_rfc3339(string: str) -> Union[datetime, date, time]:`
`113`	`113`	`**{v: f"\\{k}" for k, v in _escaped.items()},`
`114`	`114`	`'"""': '""\\"',`
`115`	`115`	`}`
`116`		`-_basic_escapes = CONTROL_CHARS \| {'"'}`
	`116`	`+_basic_escapes = CONTROL_CHARS \| {'"', "\\"}`
`117`	`117`
`118`	`118`
`119`	`119`	`def _unicode_escape(seq: str) -> str:`