Fix potential issue with line number recognition on non Str/Expr nodes

mutpy · Nov 8, 2018 · 782fc9a · 782fc9a
1 parent e893b8b
commit 782fc9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/astmonkey/visitors.py b/astmonkey/visitors.py
@@ -159,7 +159,7 @@ def _get_current_line_no(self):
 
     @classmethod
     def _get_actual_lineno(cls, node):
-        if node.col_offset == -1 and isinstance(node, (ast.Expr, ast.Str)):
+        if isinstance(node, (ast.Expr, ast.Str)) and node.col_offset == -1:
             str_content = cls._get_string_content(node)
             node_lineno = node.lineno - str_content.count('\n')
         else: