获取Python方法的描述、参数、返回值

2023-08-21  本文已影响0人  木木111314
import inspect
from  Tool import Tool

tool1=   Tool()
methods = [method for method in dir(tool1) if callable(getattr(tool1, method))]

# 获取方法描述
def getmethoddes(method):
    docstring = inspect.getdoc(method)
    if docstring==None:
        return ""
    lines = docstring.strip().split('\n')

    first_paragraph = ""
    for line in lines:
        line = line.strip()
        if line:
            first_paragraph += line + " "
        else:
            break
    return first_paragraph
# 获取方法参数
def getmethodparams(method):
    parameters = inspect.signature(method).parameters
    #for name, parameter in parameters.items():
    #    print(name, parameter.default)
    return parameters

import ast
class ReturnValueVisitor(ast.NodeVisitor):
    def __init__(self):
        self.return_variables = []

    def visit_Return(self, node):
        if isinstance(node.value, ast.Tuple):
            for item in node.value.elts:
                if isinstance(item, ast.Name):
                    self.return_variables.append(item.id)
        elif isinstance(node.value, ast.Name):
            self.return_variables.append(node.value.id)

# 获取方法返回值
def getReturns(method):
    import textwrap
    source_code = inspect.getsource(method)
    dedented_code = textwrap.dedent(source_code)
    # 解析源代码
    tree = ast.parse(dedented_code)

    # 遍历抽象语法树
    visitor = ReturnValueVisitor()
    visitor.visit(tree)

    # 获取返回值的变量名称
    return_variables = visitor.return_variables
    return return_variables





for name in methods:
    if not name.startswith("__"):
        method = getattr(tool1, name)
        print(name)
        #des=getmethoddes(method)
        #print("\"{0}\",".format(des))
        parameters = getmethodparams(method)
        return_variables = getReturns(method)
        print(return_variables)
    # print(dir(method))

上一篇 下一篇

猜你喜欢

热点阅读