如何在Python及Jupyter Notebook中强制标识符仅使用ASCII字符，同时允许UTF-8字符串？

这个需求挺贴合实际的——既要保留字符串里UTF-8内容的灵活性，又要规范标识符（变量、函数、类名等）只能用ASCII字符，避免团队协作时出现编码相关的混乱。我整理了几个实用的方案，分Python脚本和Jupyter Notebook两种场景来说：

一、普通Python脚本的解决方案 #

方案1：AST静态检查脚本（推荐，轻量无依赖） #

Python的抽象语法树（AST）可以帮我们轻松遍历代码里的所有标识符，同时完全不影响字符串内容。你可以写一个简单的检查脚本，在运行代码前先做校验：

import ast
import sys

def check_ascii_identifiers(filename):
    # 读取脚本文件（用UTF-8编码兼容所有字符串内容）
    with open(filename, 'r', encoding='utf-8') as f:
        tree = ast.parse(f.read())
    
    # 遍历AST节点，检查各类标识符
    for node in ast.walk(tree):
        # 检查函数名
        if isinstance(node, ast.FunctionDef):
            if not node.name.isascii():
                print(f"❌ 错误：第{node.lineno}行存在非ASCII函数名 '{node.name}'")
                return 1
        # 检查变量名、参数名等
        elif isinstance(node, ast.Name):
            if not node.id.isascii():
                print(f"❌ 错误：第{node.lineno}行存在非ASCII标识符 '{node.id}'")
                return 1
        # 检查类名
        elif isinstance(node, ast.ClassDef):
            if not node.name.isascii():
                print(f"❌ 错误：第{node.lineno}行存在非ASCII类名 '{node.name}'")
                return 1
    print("✅ 所有标识符均符合ASCII要求")
    return 0

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("使用方式：python check_identifiers.py 你的脚本文件.py")
        sys.exit(1)
    sys.exit(check_ascii_identifiers(sys.argv[1]))

使用方法：

• 把脚本保存为check_identifiers.py
• 运行命令：python check_identifiers.py your_script.py
• 如果有非ASCII标识符会直接报错，字符串里的UTF-8内容完全不受影响

你还可以把这个脚本集成到Git的pre-commit钩子中，每次提交代码自动做检查，从流程上规范编码。

方案2：用flake8自定义规则（适合已有代码检查流程的项目） #

如果你的项目已经在用flake8做代码检查，可以自定义一个规则来校验标识符。不过需要写简单的flake8插件，相对上面的脚本稍复杂一些，适合有一定工具链基础的团队。

二、Jupyter Notebook的解决方案 #

方案1：用nbqa配合静态检查脚本 #

nbqa工具可以让你用普通的Python代码检查工具来处理Notebook文件。步骤如下：

1. 安装nbqa：pip install nbqa
2. 用上文的check_identifiers.py脚本，运行命令：

   nbqa python check_identifiers.py 你的笔记本.ipynb
   

这样就能批量检查Notebook里所有代码cell的标识符，字符串内容不会被拦截。

方案2：自定义魔术命令（实时检查单个cell） #

如果你想在运行Notebook的cell时实时检查，可以写一个自定义的Jupyter魔术命令：

from IPython.core.magic import register_cell_magic
import ast

@register_cell_magic
def check_ascii(line, cell):
    try:
        # 解析cell代码为AST
        tree = ast.parse(cell)
        # 遍历检查标识符
        for node in ast.walk(tree):
            if isinstance(node, ast.FunctionDef) and not node.name.isascii():
                raise SyntaxError(f"非ASCII函数名：'{node.name}'")
            elif isinstance(node, ast.Name) and not node.id.isascii():
                raise SyntaxError(f"非ASCII标识符：'{node.id}'")
            elif isinstance(node, ast.ClassDef) and not node.name.isascii():
                raise SyntaxError(f"非ASCII类名：'{node.name}'")
        # 检查通过则执行cell代码
        exec(cell, globals())
    except Exception as e:
        print(f"❌ 检查失败：{e}")

使用方法：

• 先运行上面的cell注册魔术命令
• 在需要检查的代码cell开头加上%%check_ascii，比如：

  %%check_ascii
  # 这个会触发报错
  def тест():
      pass
  
  # 这个能正常运行
  text = "Привет, мир!"
  

这样每次运行cell时都会先做标识符检查，有问题直接提示，不影响合法的字符串内容。

备注：内容来源于stack exchange，提问作者Филя Усков