summaryrefslogtreecommitdiff
path: root/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils.py')
-rw-r--r--utils.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/utils.py b/utils.py
index cc20da3e..bbdc2c62 100644
--- a/utils.py
+++ b/utils.py
@@ -205,9 +205,11 @@ def _try_except_from_branch(node, stmt):
"""returns values to compare to nodes"""
if stmt in node.body:
return 'body', 1
- if stmt is node.orelse:
+ if stmt in node.orelse:
return 'else', 1
for i, handler in enumerate(node.handlers):
+ if stmt is handler:
+ return 'except', i
if stmt in handler.body:
return 'except', i
raise ASTNGError, "comparing try_except nodes '%s' '%s'" % (node, stmt)