獲取Python方法的描述、參數(shù)、返回值

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
# 獲取方法參數(shù)
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)

    # 遍歷抽象語(yǔ)法樹(shù)
    visitor = ReturnValueVisitor()
    visitor.visit(tree)

    # 獲取返回值的變量名稱(chēng)
    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))

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容