diff --git a/jedi/parsing_representation.py b/jedi/parsing_representation.py index 44a78344..a4a5fa9d 100644 --- a/jedi/parsing_representation.py +++ b/jedi/parsing_representation.py @@ -659,7 +659,8 @@ class Statement(Simple): :type start_pos: tuple(int, int) """ __slots__ = ('token_list', 'used_vars', - 'set_vars', '_commands', '_assignment_details') + 'set_vars', '_commands', '_assignment_details', + 'docstr') def __init__(self, module, set_vars, used_vars, token_list, start_pos, end_pos, parent=None): @@ -670,12 +671,17 @@ class Statement(Simple): s.parent = self.use_as_parent self.set_vars = self._remove_executions_from_set_vars(set_vars) self.parent = parent + self.docstr = '' # cache self._commands = None self._assignment_details = [] # this is important for other scripts + def add_docstr(self, string): + """ Clean up a docstring """ + self.docstr = cleandoc(literal_eval(string)) + def _remove_executions_from_set_vars(self, set_vars): """ Important mainly for assosiative arrays: