如何区分函数的硬编码输入变量与本地变量？能否用LibCST检测参数值？

一、区分硬编码变量与本地定义变量 #

从代码语法结构上，可以通过参数对应的AST节点类型直接区分：

• 硬编码变量：传参时使用字面量（比如字符串、数字等直接写死的值），对应LibCST中的SimpleString/Literal等节点，例如my_func(x="xyz")里的"xyz"。
• 本地定义变量：传参时使用当前作用域内定义的变量名，对应LibCST中的Name节点，例如你代码里my_func(x=local_var)中的local_var。

二、用LibCST Visitor检测my_func的x参数值为"xyz"的可行性 #

LibCST本身是静态语法分析工具，仅做代码结构解析，默认不会追踪变量的赋值内容。但可以通过自定义作用域内的符号追踪逻辑实现需求，具体思路如下：

1. 在Visitor中维护作用域栈，记录每个函数作用域内的变量与对应赋值（仅处理直接赋值为字面量的场景）；
2. 识别my_func的调用节点，提取x参数对应的变量名；
3. 到当前作用域的符号表中查找该变量的赋值内容，判断是否为"xyz"。

示例实现代码 #

import libcst as cst

class MyFuncParamChecker(cst.CSTVisitor):
    def __init__(self):
        # 用栈管理嵌套作用域
        self.scopes = [{}]

    def visit_FunctionDef(self, node: cst.FunctionDef) -> None:
        # 进入新函数，创建新作用域
        self.scopes.append({})

    def leave_FunctionDef(self, node: cst.FunctionDef) -> None:
        # 离开函数，弹出当前作用域
        self.scopes.pop()

    def visit_Assign(self, node: cst.Assign) -> None:
        # 处理单变量直接赋值为字符串字面量的情况
        if len(node.targets) == 1 and isinstance(node.targets[0].target, cst.Name):
            var_name = node.targets[0].target.value
            if isinstance(node.value, cst.SimpleString):
                self.scopes[-1][var_name] = node.value.value

    def visit_Call(self, node: cst.Call) -> None:
        # 识别my_func的调用
        if isinstance(node.func, cst.Name) and node.func.value == "my_func":
            # 遍历关键字参数找x
            for arg in node.args:
                if isinstance(arg, cst.Arg) and arg.keyword and arg.keyword.value == "x":
                    arg_node = arg.value
                    if isinstance(arg_node, cst.Name):
                        var_name = arg_node.value
                        # 从当前作用域查找变量赋值
                        if var_name in self.scopes[-1]:
                            assigned_val = self.scopes[-1][var_name]
                            if assigned_val == "xyz":
                                print(f"检测到my_func的x参数值为'xyz'，位于第{node.lineno}行")

# 测试你的代码
test_code = """
def my_func(x: str) -> None:
    pass

def my_caller_func() -> None:
    local_var = "xyz"
    local_var_2 = "zzz"
    my_func(x=local_var)
"""

# 解析代码并运行检测
module = cst.parse_module(test_code)
checker = MyFuncParamChecker()
module.visit(checker)

注意事项 #

这个实现仅处理简单直接赋值为字符串字面量的场景，如果变量是通过表达式计算、函数返回值、或者跨作用域赋值，仅靠LibCST的基础Visitor无法完成复杂的数据流追踪，需要结合更专业的静态分析能力才能实现更全面的检测。

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