如何为Python数据类字段传入None时设置默认值？

I get it—writing repetitive checks in __post_init__ for every field gets old fast when you have a lot of parameters. Here's a more scalable solution that avoids boilerplate code while achieving exactly what you want:

Use a Reusable Decorator #

We can create a decorator that automatically adds the "replace None with default" logic to any dataclass. This way, you don't have to manually write checks for each field.

Step 1: Define the Decorator #

import dataclasses
from dataclasses import dataclass

def replace_none_with_default(cls):
    # Save any existing __post_init__ method to call later
    original_post_init = getattr(cls, '__post_init__', None)
    
    def new_post_init(self):
        # Iterate over all fields in the dataclass
        for field in dataclasses.fields(self):
            current_value = getattr(self, field.name)
            if current_value is None:
                # Use the field's default value if available
                if field.default is not dataclasses.MISSING:
                    setattr(self, field.name, field.default)
                # Optional: Handle fields with default_factory (e.g., mutable defaults)
                elif field.default_factory is not dataclasses.MISSING:
                    setattr(self, field.name, field.default_factory())
        
        # Call the original __post_init__ if it exists
        if original_post_init:
            original_post_init(self)
    
    cls.__post_init__ = new_post_init
    return cls

Step 2: Apply the Decorator to Your Dataclass #

Now just add the decorator above your dataclass definition—no more manual field checks:

@replace_none_with_default
@dataclass
class Specs3:
    a: str
    b: str = 'Bravo'
    c: str = 'Charlie'

Step 3: Test It Out #

This behaves exactly as you expect:

# Example 1: Replace None for b with default
r1 = Specs3('Apple', None, 'Cherry')
print(r1)  # Output: Specs3(a='Apple', b='Bravo', c='Cherry')

# Example 2: Replace None for c with default
r2 = Specs3('Apple', 'Beta', None)
print(r2)  # Output: Specs3(a='Apple', b='Beta', c='Charlie')

# Example 3: Use all defaults for omitted fields
r3 = Specs3('Apple')
print(r3)  # Output: Specs3(a='Apple', b='Bravo', c='Charlie')

How This Works #

• The decorator inspects all fields in your dataclass using dataclasses.fields().
• For each field where the value is None, it replaces it with the field's predefined default (or calls default_factory for mutable defaults like lists/dicts).
• It preserves any existing __post_init__ logic you might have, so you can still add custom initialization code if needed.

Why This Is Better Than Your Original Approach #

• No redundancy: You don't have to write a line of code for each field—even if you have 10+ fields, the decorator handles all of them automatically.
• Cleaner field definitions: You don't need ClassVar for default values; the dataclass's native field defaults work as expected.
• Reusable: You can apply this decorator to any dataclass in your project that needs this behavior.

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