如何用Python实现支持直接输入表达式的计算器并处理无效输入

下面是符合需求的改进代码，支持直接输入完整数学表达式并处理无效输入：

while True:
    expression = input("请输入数学表达式（输入q退出）：").strip()
    
    # 退出逻辑
    if expression.lower() == 'q':
        print("计算器已退出")
        break
    
    # 空输入处理
    if not expression:
        print("输入不能为空，请重新输入")
        continue
    
    try:
        # 计算表达式结果
        result = eval(expression)
        print(f"计算结果：{expression} = {result}")
    except (SyntaxError, NameError, ZeroDivisionError, TypeError):
        print("输入无效，请输入合法的数学表达式（如123+456、789*101）")

关键改进说明： #

• 直接表达式输入：用户无需分步选择操作，直接输入完整算术表达式即可，匹配需求中的使用场景
• 多场景错误处理：覆盖多种无效输入情况：
  • 语法错误（如输入123++456）
  • 含非数字非法字符（如输入123a+456）
  • 除数为零
  • 类型不匹配的运算
• 循环交互：添加退出机制，输入q即可结束程序，提升使用体验
• 空输入拦截：避免用户直接回车导致的程序异常

安全增强版（避免恶意输入风险） #

如果需要面向不可信用户使用，可通过ast模块先验证表达式合法性，仅允许基础算术运算：

import ast

def check_valid_expression(expr):
    try:
        tree = ast.parse(expr, mode='eval')
        # 仅允许数字和加减乘除运算符
        allowed_nodes = (ast.Constant, ast.Add, ast.Sub, ast.Mult, ast.Div)
        for node in ast.walk(tree):
            if not isinstance(node, allowed_nodes):
                return False
        return True
    except SyntaxError:
        return False

while True:
    expression = input("请输入数学表达式（输入q退出）：").strip()
    
    if expression.lower() == 'q':
        print("计算器已退出")
        break
    
    if not expression:
        print("输入不能为空，请重新输入")
        continue
    
    if not check_valid_expression(expression):
        print("输入无效，请输入合法的数学表达式（如123+456、789*101）")
        continue
    
    try:
        result = eval(expression)
        print(f"计算结果：{expression} = {result}")
    except ZeroDivisionError:
        print("输入无效：除数不能为零")

这个版本通过语法树校验，严格限制可执行的表达式范围，彻底避免恶意代码执行风险。

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