如何简化Python两个变量之间类型同步的实现代码

核心原理 #

Python 中的类型本身是可调用对象，你不需要逐个判断类型分支，直接获取参考变量的类型构造器，即可直接完成转换，大幅简化代码。

基础通用实现 #

适配int、float、str、bool等绝大多数内置类型的默认转换规则：

def sync_type(original_var, new_var):
    target_type = type(original_var)
    if isinstance(new_var, target_type):
        return new_var
    return target_type(new_var)

调用方式和你原来的函数完全一致，不需要修改业务代码。

保留自定义转换规则的扩展实现 #

如果你需要保留原来对bytes类型的特殊处理（整数转2字节大端格式），只需要对特殊规则单独做判断即可：

def sync_type(original_var, new_var):
    target_type = type(original_var)
    if isinstance(new_var, target_type):
        return new_var
    # 仅特殊处理自定义转换逻辑的类型
    if target_type is bytes:
        return new_var.to_bytes(2, "big")
    return target_type(new_var)

类似astype的便捷调用方案 #

你不需要额外定义函数也可以实现单行转换：

# 一行代码完成类型同步，效果和调用sync_type完全一致
new_var = type(original_var)(new_var) if not isinstance(new_var, type(original_var)) else new_var

如果你想要完全实现new_var.astype(original_var)的调用语法，需要对内置类型做猴子补丁，不推荐在生产环境使用，上述单行写法已经足够简洁。

注意：如果目标类型的默认构造方法不支持传入当前new_var类型作为参数，会抛出TypeError，该行为和你原有分支判断实现的报错逻辑完全一致。

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