diff options
author | Bryce Guinta <bryce.paul.guinta@gmail.com> | 2018-02-26 17:18:50 -0700 |
---|---|---|
committer | Bryce Guinta <bryce.paul.guinta@gmail.com> | 2018-03-03 01:10:02 -0700 |
commit | fbb93327a3e85a3c4661261c0bd6a0735a53acf7 (patch) | |
tree | 66c2bb216790483956fb53c2f26b2eff5d661c96 | |
parent | c71433d13a7f884cbda48da7b812da42078761fb (diff) | |
download | pylint-git-fbb93327a3e85a3c4661261c0bd6a0735a53acf7.tar.gz |
Add in_comprehension utility function
-rw-r--r-- | pylint/checkers/utils.py | 10 |
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 |