Python如何实现input()接收含变量引用的非字符串字面量输入

Python 内置的input()函数设计上永远返回字符串类型，没有原生参数可以让它直接返回数值、表达式对象等其他类型。你需要的能力本质是「把输入得到的字符串表达式，放到程序当前的变量上下文里做计算」，不需要修改input本身的行为，有两类成熟实现方案，根据使用场景选择即可。

方案1：本地自用/信任输入场景（实现最简单） #

如果脚本是个人本地使用，不存在恶意输入风险，可以直接用eval()函数执行字符串表达式，传入指定的变量作用域限制访问范围即可。
修正后的可运行代码：

def main():
    x = 2
    input_expr = input("Please enter your Function: ")
    # 限定表达式仅能访问传入的x变量，缩小权限范围
    fx = eval(input_expr, globals(), {"x": x})
    print(fx)

main()

运行时输入x*x会直接输出计算结果4。

使用eval()的注意事项：

• 仅适合本地个人脚本、完全信任输入来源的场景
• 禁止在面向公网、接收陌生用户输入的服务中使用，恶意输入可以直接执行任意系统命令，造成数据泄露、系统损坏等严重后果

方案2：生产环境/不可信输入场景（安全可控） #

如果需要接收外部不可信用户输入，不能使用eval()，可以借助Python标准库ast模块做表达式语法解析，通过白名单机制仅允许合法的数学运算、指定变量访问，拦截所有恶意操作。
示例代码：

import ast
import operator

# 配置允许使用的运算符白名单
ALLOWED_OPERATORS = {
    ast.Add: operator.add,
    ast.Sub: operator.sub,
    ast.Mult: operator.mul,
    ast.Div: operator.truediv,
    ast.Pow: operator.pow,
    ast.USub: operator.neg
}

def safe_calculate(expr, valid_vars):
    def parse_node(node):
        if isinstance(node, ast.Constant):
            if isinstance(node.value, (int, float)):
                return node.value
            raise ValueError(f"非法常量类型：{type(node.value)}")
        elif isinstance(node, ast.Name):
            if node.id in valid_vars:
                return valid_vars[node.id]
            raise ValueError(f"未定义变量：{node.id}")
        elif isinstance(node, ast.BinOp):
            if type(node.op) not in ALLOWED_OPERATORS:
                raise ValueError(f"非法运算符：{type(node.op)}")
            return ALLOWED_OPERATORS[type(node.op)](parse_node(node.left), parse_node(node.right))
        elif isinstance(node, ast.UnaryOp):
            if type(node.op) not in ALLOWED_OPERATORS:
                raise ValueError(f"非法一元运算符：{type(node.op)}")
            return ALLOWED_OPERATORS[type(node.op)](parse_node(node.operand))
        else:
            raise ValueError(f"非法语法：{type(node)}")
    ast_tree = ast.parse(expr, mode='eval')
    return parse_node(ast_tree.body)

def main():
    x = 2
    input_expr = input("Please enter your Function: ")
    fx = safe_calculate(input_expr, {"x": x})
    print(fx)

main()

这个方案会自动拦截所有非白名单的操作，比如导入模块、执行系统命令、访问未授权变量的行为都会直接抛出错误，安全性有保障。如果需要支持更多数学函数（比如三角函数、开方），只需要把对应函数加入valid_vars字典、按需补充运算符白名单即可。

不要用exec()实现这个需求，exec()执行权限更高，安全风险比eval()更大，且没有直接返回值，不适合做表达式计算场景。

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