diff options
author | Claudiu Popa <pcmanticore@gmail.com> | 2018-07-24 10:22:22 +0200 |
---|---|---|
committer | Claudiu Popa <pcmanticore@gmail.com> | 2018-07-24 10:22:22 +0200 |
commit | fcbf25697d613c37bcc9d38004d576ce97f2cd78 (patch) | |
tree | 3bc5a31b9c0f32b63fc729ea0116659dc021a0be /astroid/inference.py | |
parent | 32c96a687ad0e5c59f7c1800bc2e00af26b16d0b (diff) | |
download | astroid-git-fcbf25697d613c37bcc9d38004d576ce97f2cd78.tar.gz |
Rename method for making it clearer
Diffstat (limited to 'astroid/inference.py')
-rw-r--r-- | astroid/inference.py | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/astroid/inference.py b/astroid/inference.py index 3b00f689..609b7948 100644 --- a/astroid/inference.py +++ b/astroid/inference.py @@ -57,17 +57,7 @@ nodes.Const._infer = infer_end nodes.Slice._infer = infer_end -def infer_seq(self, context=None): - if not any(isinstance(e, nodes.Starred) for e in self.elts): - yield self - else: - values = _infer_seq(self, context) - new_seq = type(self)(self.lineno, self.col_offset, self.parent) - new_seq.postinit(values) - yield new_seq - - -def _infer_seq(node, context=None): +def _infer_sequence_helper(node, context=None): """Infer all values based on _BaseContainer.elts""" values = [] @@ -80,15 +70,25 @@ def _infer_seq(node, context=None): if not hasattr(starred, 'elts'): raise exceptions.InferenceError(node=node, context=context) - values.extend(_infer_seq(starred)) + values.extend(_infer_sequence_helper(starred)) else: values.append(elt) return values -nodes.List._infer = infer_seq -nodes.Tuple._infer = infer_seq -nodes.Set._infer = infer_seq +def infer_sequence(self, context=None): + if not any(isinstance(e, nodes.Starred) for e in self.elts): + yield self + else: + values = _infer_sequence_helper(self, context) + new_seq = type(self)(self.lineno, self.col_offset, self.parent) + new_seq.postinit(values) + yield new_seq + + +nodes.List._infer = infer_sequence +nodes.Tuple._infer = infer_sequence +nodes.Set._infer = infer_sequence def infer_map(self, context=None): |