From 782fc9ac8318fb4459c36e8db47db9be5143782d Mon Sep 17 00:00:00 2001 From: Philipp Hossner Date: Thu, 8 Nov 2018 21:17:19 +0100 Subject: [PATCH] Fix potential issue with line number recognition on non Str/Expr nodes --- astmonkey/visitors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/astmonkey/visitors.py b/astmonkey/visitors.py index 419f964..9d18c70 100644 --- 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: