如何优化分离实例创建/更新/组合的Python代码更具Pythonic风格

所有方案均不改动原有「实例创建-属性更新-实例组合运算」的分层业务逻辑，仅优化对象传递与访问的实现方式：

方案1：使用dataclass实现固定结构的组件容器 #

• 适用场景：组件集合的字段固定（如示例中固定包含a/b两个实例，无动态增减字段的需求），是替换硬编码字典的首选方案
• 优势：自带类型提示支持，IDE可自动补全属性，不会因为键名拼写错误抛出KeyError，代码可读性远高于裸字典
• 实现示例：

from dataclasses import dataclass

# 原有业务类完全不需要改动
class A:
    def __init__(self, variable):
        self.variable = variable

class B:
    def __init__(self, variable):
        self.variable = variable

# 定义固定结构的组件容器，无需写冗余的__init__逻辑
@dataclass
class Components:
    a: A
    b: B

def create_components():
    a = A(variable=1)
    b = B(variable=2)
    return Components(a=a, b=b)

def add_number(components: Components, number):
    # 直接通过属性访问实例，无需重复写字典取值逻辑
    components.a.variable += number
    components.b.variable += number
    # 自定义对象为引用传递，无需重新构造容器返回
    return components

def operation(components: Components):
    return components.a.variable + components.b.variable

def model():
    components = create_components()
    new_components = add_number(components, 1)
    return operation(new_components)

if __name__ == "__main__":
    result = model()
    print(result) # 输出5，与原逻辑完全一致

方案2：使用SimpleNamespace实现轻量动态容器 #

• 适用场景：不想提前定义容器类，存在少量动态增减组件字段的需求，追求最简代码
• 优势：Python标准库内置，无需额外定义类，支持属性访问，代码量极小
• 实现示例：

from types import SimpleNamespace

class A:
    def __init__(self, variable):
        self.variable = variable

class B:
    def __init__(self, variable):
        self.variable = variable

def create_components():
    a = A(variable=1)
    b = B(variable=2)
    # 直接构造属性容器，无需提前定义类结构
    return SimpleNamespace(a=a, b=b)

def add_number(components, number):
    components.a.variable += number
    components.b.variable += number
    return components

def operation(components):
    return components.a.variable + components.b.variable

# model层逻辑与原逻辑完全一致

• 缺点：没有固定字段约束，IDE自动补全、类型检查的支持弱于dataclass，适合快速开发场景。

方案3：自定义AttrDict兼顾字典灵活性与属性访问简洁性 #

• 适用场景：确实需要字典的动态键能力（如运行时才确定组件列表），不想每次写components["xxx"]的冗余取值代码
• 优势：完全兼容原有字典的所有操作，同时支持属性访问，旧代码改造成本极低
• 实现示例：

class AttrDict(dict):
    def __getattr__(self, key):
        try:
            return self[key]
        except KeyError:
            raise AttributeError(f"不存在名为{key}的组件")
    def __setattr__(self, key, value):
        self[key] = value

# 业务代码中既可以用属性访问，也保留字典的所有操作能力
def create_components():
    a = A(variable=1)
    b = B(variable=2)
    return AttrDict(a=a, b=b)

额外优化细节 #

• 原代码中每次操作后都重新拼接字典返回属于冗余逻辑：无论是字典、dataclass还是SimpleNamespace，在Python中都是引用传递，只要不需要生成不可变副本，直接修改原对象返回即可，不需要重复构造容器。
• 如果业务类（如示例中的A、B）本身仅用来存储属性、没有复杂自定义方法，也可以直接将其定义为dataclass，省去手动编写__init__的冗余代码：

from dataclasses import dataclass

@dataclass
class A:
    variable: int

@dataclass
class B:
    variable: int

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