From ef99ba4c95ca2d29189cc5f03e39ea775943a7ff Mon Sep 17 00:00:00 2001 From: reshinthadithyan Date: Sat, 19 Dec 2020 19:47:38 +0530 Subject: [PATCH 1/2] Addition of get_info() under class --- .../core/parser/python/python_parser.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tree_hugger/core/parser/python/python_parser.py b/tree_hugger/core/parser/python/python_parser.py index 226a0c7..7faa7b8 100644 --- a/tree_hugger/core/parser/python/python_parser.py +++ b/tree_hugger/core/parser/python/python_parser.py @@ -337,3 +337,24 @@ def get_all_class_documentations(self, strip_quotes: bool=False) -> Dict[str, st Returns the docstring of all classes """ return self.get_all_class_docstrings() + def get_info(self): + """ + Returns an info_dict which would be a + complete information of a given file. + """ + self.info_dict = { + "class":[], + "functions":[] + } + #Function related Mapping + all_func_bodies = self.get_all_function_bodies() + all_func_doc_string = self.get_all_function_docstrings() + all_funcs = {i:{"doc_string":[all_func_doc_string[i]],"body":[all_func_bodies[i]]} for i in self.get_all_function_names()} + #Class related Mapping + all_class_bodies = self.get_all_class_docstrings() + all_classes = {i:{"doc_string":[all_class_bodies[i]]} for i in self.get_all_class_names()} + #Appending to info_dict + self.info_dict["class"] = all_classes + self.info_dict["functions"] = all_funcs + + return self.info_dict#self.info_dict#self.info_dict["class"] \ No newline at end of file From 4a544a4f74c4620f3405cb0266b63bfecc44e45d Mon Sep 17 00:00:00 2001 From: reshinthadithyan Date: Sat, 19 Dec 2020 20:12:07 +0530 Subject: [PATCH 2/2] Making stage --- tree_hugger/core/parser/python/python_parser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tree_hugger/core/parser/python/python_parser.py b/tree_hugger/core/parser/python/python_parser.py index 7faa7b8..2ea791d 100644 --- a/tree_hugger/core/parser/python/python_parser.py +++ b/tree_hugger/core/parser/python/python_parser.py @@ -351,8 +351,8 @@ def get_info(self): all_func_doc_string = self.get_all_function_docstrings() all_funcs = {i:{"doc_string":[all_func_doc_string[i]],"body":[all_func_bodies[i]]} for i in self.get_all_function_names()} #Class related Mapping - all_class_bodies = self.get_all_class_docstrings() - all_classes = {i:{"doc_string":[all_class_bodies[i]]} for i in self.get_all_class_names()} + all_class_doc_strings = self.get_all_class_docstrings() + all_classes = {i:{"doc_string":[all_class_doc_strings[i]]} for i in self.get_all_class_names()} #Appending to info_dict self.info_dict["class"] = all_classes self.info_dict["functions"] = all_funcs