将txt文件中的所有函数体存储在python的字典中

Question

我正在尝试做代码分析器应用程序，我有一个包含 python 代码的 txt 文件，我现在的目标是将这个 txt 文件中的所有函数保存在类的字典中，但我不知道如何才能首先

，我创建名为 class CodeAnalyzer: 的类

def __init__(self, file):
    self.file = file
    self.file_string = ""
    self.file_func= {}
    self.errors = {}

，并且我想将函数保存在 self.file_func= {} 中

，这是流程步骤，每个方法都应该返回键以及属性的增值

def process_file(self):
        for i, line in enumerate(self.file):
            self.file_string += line
            self.check_divide_by_zero(i, line)
            self.check_parameters_num(i, line)

这是我试图做的但 ie 失败了：

def store_function(self,i,line):
            if(line.startswith('def')):
                self.file_func.setdefault(i,[]).append((self.file_string[file_string.index(':') ,:]))

有人对此有想法或帮助吗？

Answer 1

您只需使用 exec() ，并将其 globals() 字典设置为类实例的命名空间。

class CodeAnalyzer:
  def __init__(self,file):
    # Read a string from the file
    f=open(file)
    t=f.read()
    f.close()

    #Populate the namespace dictionary by executing the code in the file
    self.namespace={}#this includes all declarations of functions, variables and classes
    exec(t,self.namespace)#this means that when variables are declared, they use the class instance's attributes dictionary as their global namespace

    #Filter the namespace dict based on the contents
    self.functions={i:self.namespace[i] for i in self.namespace if isinstance(i,type(lambda:0))}#type(lambda:0) is just FunctionType
    self.classes={i:self.namespace[i] for i in self.namespace if isinstance(i,type)}#all classes are instances of type
    self.variables={i:self.namespace[i] for i in self.namespace if i not in self.functions|self.classes}#everything else, using dictionary merge

如果您还有其他问题，请随时对此答案发表评论。

Answer 2

我认为 isinstance 检查应该是

self.functions={i:self.namespace[i] for i in self.namespace if isinstance(self.namespace[i],type(lambda:0))}

我们想要测试 self.namespace 的第 i 个元素，而不是 i 的值。