Início / coding / Perguntas / 79557806
Accepted
qq_rr_ii_mm
Asked: 2025-04-06 12:41:35 +0800 CST

como ignorar elegantemente argumentos de correspondência de palavras-chave não correspondentes em python dataclass

  • 772

Com classes normais você tem **kwargsargumentos __init__de palavras-chave não correspondentes que podem ser ignorados:

class MyClass:
    def __init__(self, a, **kwargs):
        self.a=a

my_class = MyClass(20, **{"kwarg1" : 1})

Existe um equivalente @dataclassque pode ignorar facilmente argumentos de palavras-chave não correspondentes sem precisar incluir um __init__?

from dataclasses import dataclass

@dataclass
class MyClass:
    a: int

my_class = MyClass(20, **{"kwarg1" : 1}) # TypeError: MyClass.__init__() got an unexpected keyword argument 'kwarg1'
python

2 respostas

  1. Best Answer

    TL;DR Não, é impossível

    Esse cenário foi até citado na introdução original dataclass.

    Acho que isso vai contra as suposições originais dataclass- que simplificariam a semântica em detrimento de tornar a funcionalidade menos flexível.

    vídeo pep-0557

    Às vezes, o método init gerado não é suficiente. Por exemplo, suponha que você queira ter um objeto para armazenar *args e **kwargs:

    @dataclass(init=False)
class ArgHolder:
    args: List[Any]
    kwargs: Mapping[Any, Any]

    def __init__(self, *args, **kwargs):
        self.args = args
        self.kwargs = kwargs

a = ArgHolder(1, 2, three=3)
    • 1

  2. Embora @dataclassnão suporte nativamente uma opção para ignorar argumentos de palavras-chave não correspondentes, você pode usar um decorador wrapper que corrige o __init__método gerado por @dataclasscom uma função wrapper que passa para o __init__método original apenas argumentos de palavras-chave que correspondem à sua assinatura :

    from dataclasses import dataclass
from inspect import signature, Parameter

def lax_dataclass(cls_or_None=None, **kwargs):
    def decorator(cls):
        def __init__(self, *args, **kwargs):
            orig_init(self, *args, **{
                name: value for name, value in kwargs.items()
                if name in accepted_keywords
            })
        cls = dataclass(**kwargs)(cls)
        orig_init = cls.__init__
        accepted_keywords = {
            param.name for param in signature(orig_init).parameters.values()
            if param.kind in (
                Parameter.POSITIONAL_OR_KEYWORD,
                Parameter.KEYWORD_ONLY
            )
        }
        cls.__init__ = __init__
        return cls
    if cls_or_None is None:
        return decorator
    return decorator(cls_or_None)

    para que:

    @lax_dataclass
class MyClass:
    a: int

print(MyClass(20, **{"kwarg1" : 1}))

    saídas:

    MyClass(a=20)

    Demonstração: https://ideone.com/OnZPbw

    • 0

relate perguntas