def _check_variable_usage(self, node: ast.Name) -> None:
context = cast(ast.AST, get_context(node))
blocks = self._block_variables[context][node.id]
if all(is_contained_by(node, block) for block in blocks):
return
self.add_violation(
ControlVarUsedAfterBlockViolation(node, text=node.id), )
def _check_variable_usage(self, node: ast.Name) -> None:
context = cast(ast.AST, get_context(node))
blocks = self._block_variables[context][node.id]
is_contained_block_var = any(
is_contained_by(node, block) for block in blocks)
# Restrict the use of block variables with the same name to
# the same type of block - either `for` or `with`.
is_same_type_block = all(
isinstance(block, ForNodes) for block in blocks) or all(
isinstance(block, WithNodes) for block in blocks)
# Return if not a block variable or a contained block variable.
if not blocks or (is_contained_block_var and is_same_type_block):
return
self.add_violation(
ControlVarUsedAfterBlockViolation(node, text=node.id), )