76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
# DRF Rack
|
|
# Библиотека классов расширений для Django REST Framework
|
|
#
|
|
# Автор: Николай В. Анохин <n.anokhin@xeaf.net>
|
|
# Все права защищены. Лицензия: MIT
|
|
|
|
"""
|
|
Описание класса CoverageStructureVisitor
|
|
"""
|
|
|
|
import ast
|
|
|
|
|
|
class CoverageStructureVisitor(ast.NodeVisitor):
|
|
"""
|
|
Обходчик синтаксического дерева кода проекта
|
|
"""
|
|
|
|
def __init__(self, module_dot_path: str):
|
|
"""
|
|
Инициализация
|
|
|
|
:param module_dot_path: Путь к модулю
|
|
"""
|
|
|
|
super().__init__()
|
|
self.module_dot_path = module_dot_path
|
|
self.class_stack = []
|
|
self.results = []
|
|
|
|
def visit_ClassDef(self, class_node) -> None:
|
|
"""
|
|
Обход класса
|
|
|
|
:param class_node: Узел класса
|
|
"""
|
|
|
|
self.class_stack.append(class_node.name)
|
|
self.generic_visit(class_node)
|
|
self.class_stack.pop()
|
|
|
|
def visit_FunctionDef(self, func_node) -> None:
|
|
"""
|
|
Обход функции
|
|
|
|
:param func_node: Узел функции
|
|
"""
|
|
|
|
self._handle_function(func_node)
|
|
|
|
def visit_AsyncFunctionDef(self, async_func_node) -> None:
|
|
"""
|
|
Обход асинхронной функции
|
|
|
|
:param async_func_node: Узел асинхронной функции
|
|
"""
|
|
self._handle_function(async_func_node)
|
|
|
|
def _handle_function(self, func_node) -> None:
|
|
"""
|
|
Общий класс обработки функции
|
|
|
|
:param func_node: Узел функции
|
|
"""
|
|
|
|
func_name = func_node.name
|
|
class_name = self.class_stack[-1] if self.class_stack else None
|
|
|
|
if self.class_stack:
|
|
class_chain = ".".join(self.class_stack)
|
|
full_path = f"{self.module_dot_path}.{class_chain}.{func_name}"
|
|
else:
|
|
full_path = f"{self.module_dot_path}.{func_name}"
|
|
|
|
self.results.append((func_name, class_name, full_path))
|