Python2转Python3：如何让字符串与bytes比较触发错误？

相信很多从Python 2转过来的开发者都踩过这个坑：字符串（str）和字节（bytes）的相等比较在Python 2里会自动隐式转换，比如这段代码：

foo = b'foo'
if foo == 'foo':
    print("They match!")

在Python 2里会正常输出They match!，但到了Python 3里直接静默返回False，连个提示都没有。如果是这种简单场景还好排查，但要是变量是在其他模块或者深层逻辑里定义的，看不到明显的类型标记，这种静默失败就很容易埋下难以追踪的bug。

下面分享几种让Python 3在遇到这类比较时主动抛出错误的方法：

1. 过渡阶段：用unicode_literals提前在Python 2中暴露问题 #

如果你的代码还在Python 2和3的兼容过渡阶段，可以在每个Python文件的顶部加上：

from __future__ import unicode_literals

这会让Python 2里所有的字符串字面量默认变成unicode类型（对应Python 3的str），这时和bytes比较就会直接抛出错误，帮你提前在迁移阶段就把这类问题修正掉。不过这个方法只在Python 2中生效，Python 3里没有作用。

2. 自定义装饰器：在函数层面拦截字面量比较 #

可以写一个装饰器，通过分析函数的AST语法树，检查是否存在str和bytes字面量的相等比较，一旦发现就抛出TypeError。代码示例：

import ast
import inspect
import functools

def block_str_bytes_compare(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        # 解析函数的源代码生成AST树
        func_source = inspect.getsource(func)
        tree = ast.parse(func_source)
        
        # 遍历所有节点，寻找相等比较操作
        for node in ast.walk(tree):
            if isinstance(node, ast.Compare) and isinstance(node.ops[0], ast.Eq):
                left_is_str = isinstance(node.left, ast.Str)
                left_is_bytes = isinstance(node.left, ast.Bytes)
                comp_is_str = isinstance(node.comparators[0], ast.Str)
                comp_is_bytes = isinstance(node.comparators[0], ast.Bytes)
                
                # 判断是否是str和bytes的跨类型比较
                if (left_is_str and comp_is_bytes) or (left_is_bytes and comp_is_str):
                    raise TypeError(f"禁止直接比较str和bytes！位置：{func.__name__}函数第{node.lineno}行")
        
        return func(*args, **kwargs)
    return wrapper

使用时给需要检查的函数加上装饰器即可：

@block_str_bytes_compare
def test_func():
    a = b'hello'
    b = 'hello'
    if a == b:
        print("这个比较会触发错误")

test_func()  # 执行后抛出TypeError

不过这个方法只能检测函数内部的字面量比较，如果是外部传入的变量类型不匹配，AST分析就抓不到了。

3. 静态检查：在代码提交阶段提前发现问题 #

借助静态代码分析工具，比如flake8搭配flake8-bugbear插件，其中的B008规则会专门检测str和bytes的相等比较，并给出警告。把这类检查加入你的CI流程，就能在代码提交阶段就把问题揪出来，不用等到运行时才发现。

4. 全局钩子：拦截所有运行时的跨类型比较 #

如果想要在全局层面彻底禁止str和bytes的相等比较，可以修改这两个类型的__eq__方法：

# 保存原来的__eq__方法
original_str_eq = str.__eq__
original_bytes_eq = bytes.__eq__

def patched_str_eq(self, other):
    if isinstance(other, bytes):
        raise TypeError("不允许直接比较str和bytes类型！")
    return original_str_eq(self, other)

def patched_bytes_eq(self, other):
    if isinstance(other, str):
        raise TypeError("不允许直接比较bytes和str类型！")
    return original_bytes_eq(self, other)

# 替换原有方法
str.__eq__ = patched_str_eq
bytes.__eq__ = patched_bytes_eq

这样之后，任何str == bytes或者bytes == str的操作都会直接抛出TypeError。不过要注意，这种全局修改可能会影响第三方库的行为——有些库可能为了兼容旧代码特意做了这类比较，所以使用前一定要充分测试你的整个项目。

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