From 076eea12bdef3d6b48866348ea0be7ae61e63151 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 30 Apr 2017 02:56:44 +0200 Subject: [PATCH] Some minor refactoring of the python tree. --- jedi/parser/python/tree.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/jedi/parser/python/tree.py b/jedi/parser/python/tree.py index 4dd94436..75b9834c 100644 --- a/jedi/parser/python/tree.py +++ b/jedi/parser/python/tree.py @@ -459,9 +459,12 @@ class Function(ClassOrFunc): parameters = self.children[2] # After `def foo` parameters.children[1:-1] = _create_params(parameters, parameters.children[1:-1]) + def _get_param_nodes(self): + return self.children[2].children + @property def params(self): - return [p for p in self.children[2].children if p.type == 'param'] + return [p for p in self._get_param_nodes() if p.type == 'param'] @property def name(self): @@ -518,8 +521,7 @@ class Lambda(Function): def _get_paramlist_code(self): return '(' + ''.join(param.get_code() for param in self.params).strip() + ')' - @property - def params(self): + def _get_param_nodes(self): return self.children[1:-2] def is_generator(self):