如何无需逐个改写将Python dataclass所有属性设为可选？

问题背景 #

我有如下数据类：

@dataclass
class User:
    id: str
    name: str
    pwd: str
    picture: str
    url: str
    age: int
    # 还有25-30个其他属性

之前接收的数据中，值为None的属性仍会保留键；但现在数据格式变更，值为None的属性会直接省略对应键。

原调用方式（正常工作）：

attributes = {"id": "18ut2", "pwd": "qwerty", "name": "John Doe", "picture": None, "age": None, "url": "www.example.com"}
user = User(**attributes)

当前数据调用会报错：

attributes = {"id": "18ut2", "pwd": "qwerty", "name": "John Doe", "url": "www.example.com"} # 缺少picture和age键
user = User(**attributes)

报错信息：

TypeError: User.__init__() missing 2 required positional arguments: 'picture' and 'age'

我需要将所有属性改为Optional[原类型] = None的形式，但手动修改几十行太繁琐，且不想通过重写__init__或__init_subclass__实现，求高效解决方案。

可行方案 #

方案1：自定义装饰器自动转换 #

通过装饰器批量处理所有字段，将类型转为Optional并设置默认值None：

from dataclasses import dataclass, fields, make_dataclass
from typing import Optional, TypeVar

T = TypeVar('T')

def all_optional(cls: type[T]) -> type[T]:
    original_fields = fields(cls)
    # 转换每个字段的类型和默认值
    new_field_defs = [
        (field.name, Optional[field.type], None)
        for field in original_fields
    ]
    # 基于原类创建新的可选数据类
    return make_dataclass(
        cls.__name__,
        new_field_defs,
        bases=(cls,),
        namespace=cls.__dict__.copy(),
        # 保留原dataclass的默认配置
        init=True, repr=True, eq=True,
        order=False, unsafe_hash=False, frozen=False
    )

# 使用方式
@all_optional
@dataclass
class User:
    id: str
    name: str
    pwd: str
    picture: str
    url: str
    age: int
    # 其他属性

测试调用：

attributes = {"id": "18ut2", "pwd": "qwerty", "name": "John Doe", "url": "www.example.com"}
user = User(**attributes)
print(user.picture)  # 输出: None
print(user.age)      # 输出: None

方案2：利用Python 3.10+的kw_only批量修改字段 #

如果你的Python版本≥3.10，可结合kw_only=True直接修改dataclass的字段属性（偏hack但快速）：

from dataclasses import dataclass, fields
from typing import Optional

@dataclass(kw_only=True)
class User:
    id: str
    name: str
    pwd: str
    picture: str
    url: str
    age: int
    # 其他属性

# 批量更新所有字段的类型和默认值
for field in fields(User):
    field.type = Optional[field.type]
    field.default = None

# 正常调用
attributes = {"id": "18ut2", "pwd": "qwerty", "name": "John Doe", "url": "www.example.com"}
user = User(**attributes)

方案3：元类自动处理（更灵活） #

通过自定义元类在类创建阶段完成字段转换：

from dataclasses import dataclass, fields, make_dataclass
from typing import Optional, TypeVar
import inspect

T = TypeVar('T')

class AllOptionalMeta(type):
    def __new__(cls, name: str, bases: tuple[type, ...], namespace: dict) -> type:
        # 先创建原类实例
        original_cls = super().__new__(cls, name, bases, namespace)
        # 仅处理dataclass
        if inspect.isdataclass(original_cls):
            original_fields = fields(original_cls)
            new_field_defs = [
                (f.name, Optional[f.type], None) for f in original_fields
            ]
            # 生成新的可选数据类
            return make_dataclass(
                name, new_field_defs, bases=bases, namespace=namespace,
                init=True, repr=True, eq=True
            )
        return original_cls

# 使用方式
@dataclass
class User(metaclass=AllOptionalMeta):
    id: str
    name: str
    pwd: str
    picture: str
    url: str
    age: int
    # 其他属性

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