python - 如何在对象创建过程中设置字段值？

Question

我想将id初始化为phrase和type的串联

from enum import Enum
from typing import NamedTuple

class WORD_TYPE(str, Enum):
    APPROVED = 'approved'
    FORBIDDEN = 'forbidden'
    RISKY = 'risky'

class WordItem(NamedTuple):
    phrase: str
    type: WORD_TYPE

    id: str = f'{phrase}_{type.name.lower()}'

因此，每次当我指定phrase和type时，我都希望自动获得一个id。

word_item = WordItem(phrase='phrase', type=WORD_TYPE.FORBIDDEN)
asssert word_item.id == 'phrase_forbidden'

最好的方法是什么？也许有某种方法可以使用 https://github.com/ericvsmith/dataclasses 来做到这一点？

Answer 1

这就是特殊方法 __new__ 应该解决的问题。不幸的是，NamedTuple 阻止其在其定义内进行覆盖。

但是您可以使用装饰器在定义之外覆盖它:

def set_default(attr, func):
    def set_new(typ):
        old = typ.__new__
        sig = inspect.signature(old)
        def _new(cls, *args, **kwargs):
                bound = sig.bind_partial(cls, *args, **kwargs).arguments
                if not attr in bound:
                    bound[attr] = func(*args, **kwargs)
                # print(bound)           # uncomment for debug traces
                return old(**bound)
        typ.__new__ = _new
        return typ
    return set_new

@set_default('id', lambda phrase, typ: f'{phrase}_{typ.name.lower()}')
class WordItem(NamedTuple):
    phrase: str
    type: WORD_TYPE
    id: str = None

然后您可以使用 WordItem:

>>> w = WordItem('a', WORD_TYPE.APPROVED)
>>> w
WordItem(phrase='a', type=<WORD_TYPE.APPROVED: 'approved'>, id='a_approved')
>>> w2 = WordItem('b', WORD_TYPE.APPROVED, 'c')
>>> w2
WordItem(phrase='b', type=<WORD_TYPE.APPROVED: 'approved'>, id='c')

但是:

• NamedTuple 出于某种原因阻止覆盖 __new__
• 这试图实现明确禁止的功能

如果可以的话，你应该避免这个技巧。它在我的 3.6 版本中工作正常，但它可能会在 NamedTuple 的 future 版本中崩溃，或者被 native 实现取代......