VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2026-08-07 00:18:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add a special case against augassign

Browse Source
This commit is contained in:
Batuhan Taskaya
2020-06-04 21:47:28 +03:00
parent 44cf64a5f7
commit 6daf91880b
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
parso/python/errors.py
+6 -6
View File
@@ -979,7 +979,7 @@ class _FStringRule(SyntaxRule):
class _CheckAssignmentRule(SyntaxRule): class _CheckAssignmentRule(SyntaxRule):
def _check_assignment(self, node, is_deletion=False, is_namedexpr=False): def _check_assignment(self, node, is_deletion=False, is_namedexpr=False, is_aug_assign=False):
error = None error = None
type_ = node.type type_ = node.type
if type_ == 'lambdef': if type_ == 'lambdef':
@@ -1021,9 +1021,9 @@ class _CheckAssignmentRule(SyntaxRule):
# This is not a comprehension, they were handled # This is not a comprehension, they were handled
# further above. # further above.
for child in second.children[::2]: for child in second.children[::2]:
self._check_assignment(child, is_deletion, is_namedexpr) self._check_assignment(child, is_deletion, is_namedexpr, is_aug_assign)
else: # Everything handled, must be useless brackets. else: # Everything handled, must be useless brackets.
self._check_assignment(second, is_deletion, is_namedexpr) self._check_assignment(second, is_deletion, is_namedexpr, is_aug_assign)
elif type_ == 'keyword': elif type_ == 'keyword':
if node.value == "yield": if node.value == "yield":
error = "yield expression" error = "yield expression"
@@ -1066,13 +1066,13 @@ class _CheckAssignmentRule(SyntaxRule):
error = "f-string expression" error = "f-string expression"
elif type_ in ('testlist_star_expr', 'exprlist', 'testlist'): elif type_ in ('testlist_star_expr', 'exprlist', 'testlist'):
for child in node.children[::2]: for child in node.children[::2]:
self._check_assignment(child, is_deletion, is_namedexpr) self._check_assignment(child, is_deletion, is_namedexpr, is_aug_assign)
elif ('expr' in type_ and type_ != 'star_expr' # is a substring elif ('expr' in type_ and type_ != 'star_expr' # is a substring
or '_test' in type_ or '_test' in type_
or type_ in ('term', 'factor')): or type_ in ('term', 'factor')):
error = 'operator' error = 'operator'
elif type_ == "star_expr": elif type_ == "star_expr":
if not search_ancestor(node, *_STAR_EXPR_PARENTS): if not search_ancestor(node, *_STAR_EXPR_PARENTS) and not is_aug_assign:
self.add_issue(node, message="starred assignment target must be in a list or tuple") self.add_issue(node, message="starred assignment target must be in a list or tuple")
self._check_assignment(node.children[1]) self._check_assignment(node.children[1])
@@ -1109,7 +1109,7 @@ class _ExprStmtRule(_CheckAssignmentRule):
if self._normalizer.version <= (3, 8) or not is_aug_assign: if self._normalizer.version <= (3, 8) or not is_aug_assign:
for before_equal in node.children[:-2:2]: for before_equal in node.children[:-2:2]:
self._check_assignment(before_equal) self._check_assignment(before_equal, is_aug_assign=is_aug_assign)
if is_aug_assign: if is_aug_assign:
target = _remove_parens(node.children[0]) target = _remove_parens(node.children[0])