fix patch

This commit is contained in:
frostming
2018-11-09 11:23:02 +08:00
parent 076252836c
commit 3490fc85ef
2 changed files with 6 additions and 8 deletions
+1 -2
View File
@@ -196,8 +196,7 @@ class Container(dict):
self._body[idx] = (None, Comment(Trivia(comment_ws="", comment=trivia.comment)))
else:
self._body[idx] = (None, Null())
super(Container, self).__delitem__(key.key)
super(Container, self).__delitem__(key.key)
return self
+5 -6
View File
@@ -9,11 +9,11 @@ index 37014921..5ddd72e7 100644
+from .items import Trivia
from .items import Whitespace
from .items import item as _item
@@ -189,9 +190,14 @@ class Container(dict):
if idx is None:
raise NonExistentKey(key)
- self._body[idx] = (None, Null())
+ old_data = self._body[idx][1]
+ trivia = getattr(old_data, "trivia", None)
@@ -21,9 +21,8 @@ index 37014921..5ddd72e7 100644
+ self._body[idx] = (None, Comment(Trivia(comment_ws="", comment=trivia.comment)))
+ else:
+ self._body[idx] = (None, Null())
+ super(Container, self).__delitem__(key.key)
+ super(Container, self).__delitem__(key.key)
- super(Container, self).__delitem__(key.key)
return self