Python代码重构：如何追踪变量读写判断代码块依赖关系

要实现代码块间的依赖判断（通过读写变量的交集），可以从静态分析和动态分析两个方向入手，以下是具体可行的方案：

静态分析方案 #

静态分析无需运行代码，通过解析代码的语法结构识别变量读写操作，适合快速批量排查：

• 基于Python内置ast模块自定义分析逻辑
  编写继承自ast.NodeVisitor的遍历器，重写对应节点的处理方法：
  • 捕获写变量：处理Assign、AugAssign、AnnAssign等赋值节点，记录被赋值的变量名；对于函数调用，若传入的是可变对象（如列表、字典），可通过递归分析函数AST判断是否存在修改。
  • 捕获读变量：处理Name节点，记录被引用的变量名（需排除赋值操作中的目标变量）。
    示例代码片段：

  import ast
  
  class VariableTracker(ast.NodeVisitor):
      def __init__(self):
          self.writes = set()
          self.reads = set()
  
      def visit_Assign(self, node):
          for target in node.targets:
              if isinstance(target, ast.Name):
                  self.writes.add(target.id)
          self.generic_visit(node)
  
      def visit_Name(self, node):
          if isinstance(node.ctx, ast.Load):
              self.reads.add(node.id)
          self.generic_visit(node)
  

• 基于现有静态分析工具扩展
  利用pycparser或pylint底层的astroid库，提取变量读写信息。这类工具已封装了成熟的AST遍历逻辑，可直接用于变量依赖分析。

动态分析方案 #

动态分析通过运行代码捕捉实际变量修改，能精准处理函数内部修改传入可变对象这类静态分析难以覆盖的场景：

• 利用sys.settrace()实现运行时追踪
  设置全局追踪函数，监控代码执行过程中变量的赋值、修改操作：
  • 在代码块执行前后，对比局部/全局变量的状态（可变对象需通过深拷贝或哈希值判断是否被修改）。
  • 记录所有被修改的变量，以及被读取的变量。
    可封装为上下文管理器，方便包裹需要分析的代码块：

  import sys
  import copy
  
  class RuntimeTracker:
      def __init__(self):
          self.writes = set()
          self.reads = set()
          self.prev_state = {}
  
      def trace_func(self, frame, event, arg):
          if event == 'line':
              current_state = copy.deepcopy(frame.f_locals)
              # 对比变量变化，记录被修改的变量
              for var, val in current_state.items():
                  if var in self.prev_state and self.prev_state[var] != val:
                      self.writes.add(var)
              self.prev_state = current_state
          return self.trace_func
  
      def __enter__(self):
          self.prev_state = copy.deepcopy(sys._getframe(1).f_locals)
          sys.settrace(self.trace_func)
          return self
  
      def __exit__(self, exc_type, exc_val, exc_tb):
          sys.settrace(None)
  

• 利用tracemalloc追踪内存变化
  通过追踪内存块的分配和修改，关联到对应的变量名，适合精准定位可变对象的修改，但实现复杂度较高，需结合变量的内存地址映射分析。

关键注意事项 #

• 可变对象处理：静态分析需额外识别函数对可变参数的修改逻辑，动态分析则能直接捕捉这类运行时修改。
• 跨作用域变量：需注意global/nonlocal声明的变量，静态分析要处理作用域链，动态分析要监控全局变量的变化。
• 性能开销：动态分析会显著增加代码运行时间，仅适合在测试环境中使用。

内容的提问来源于stack exchange，提问作者root