Add in_comprehension utility function

1 files changed, 10 insertions, 0 deletions

diff --git a/pylint/checkers/utils.py b/pylint/checkers/utils.py
index 5b9525dfe..c73d14704 100644
--- a/pylint/checkers/utils.py
+++ b/pylint/checkers/utils.py
@@ -928,3 +928,13 @@ def get_node_last_lineno(node):
         return get_node_last_lineno(node.body[-1])
     # Not a compound statement
     return node.lineno
+
+
+def in_comprehension(node):
+    """Return True if the given node is in a comprehension"""
+    curnode = node
+    while curnode.parent:
+        curnode = curnode.parent
+        if is_comprehension(curnode):
+            return True
+    return False