覆盖 Python 元类 __getitem__

Question

鉴于以下代码，我希望返回值'overridden'而不是'value1' ：

class MyMetaclass(type):
  def __new__(cls, name, bases, attrs):
    attrs.update({'_my_dict': {'key1': 'value1', 'key2': 'value2'}})
    return super().__new__(cls, name, bases, attrs)
  def __getitem__(cls, value):
    return cls._my_dict[str(value)]


class MyBaseClass(metaclass=MyMetaclass):
  pass


class MyClass(MyBaseClass):
  @classmethod
  def __getitem__(cls, value):
    return 'overridden'


>>> MyClass['key1']
'value1'  # I expect: 'overridden'

我需要改变什么？

Answer 1

__getitem__指的是定义它的类的实例的项获取行为。

因此，如果您在元类上定义__getitem__ ，它将适用于类 - 如果您在类本身上编写__getitem__ ，它将仅适用于该类的实例。

但是，您可以编写元类__getitem__以便它调用类' __getitem__如果存在） - 然后模拟此行为：

class MyMetaclass(type):
  def __new__(cls, name, bases, attrs):
    attrs.update({'_my_dict': {'key1': 'value1', 'key2': 'value2'}})
    return super().__new__(cls, name, bases, attrs)
  def __getitem__(cls, value):
    for supercls in cls.__mro__:
         if "__getitem__" in supercls.__dict__:
               # getting the value straight from "__dict__" will prevent Python's mechanisms of converting it to a bound instance method:
               return supercls.__dict__["__getitem__"](cls, value)
    return cls._my_dict[str(value)]


class MyBaseClass(metaclass=MyMetaclass):
  pass


class MyClass(MyBaseClass):
  def __getitem__(cls, value):
    return 'overridden'

请注意，通过这种方式，您甚至可以在我们的层次结构中为“class_getitem”使用任何其他方法名称，并且仍然保留通常的__getitem__可供实例使用。

Answer 2

您需要子类化元类，在那里覆盖__getitem__ ，并在新的常规类中设置该元类。