当类是字典的键时如何更改类表示？

Question

考虑以下代码：

class DummyClass:
    
    def __init__(self, name):
        self.name = name

obj_1 = DummyClass("name_1")
obj_2 = DummyClass("name_2")

my_dict = {
    obj_1: "value_1",
    obj_2: "value_2"
}

print(my_dict)

输出

{<__main__.DummyClass object at 0x7f8f6493a0b8>: 'value_1', <__main__.DummyClass object at 0x7f8f649869e8>: 'value_2'}

此代码创建了一个字典 my_dict，其中包含 DummyClass 的实例作为键。

我想知道是否有可以在DummyClass 中覆盖的魔术函数（例如__something__），以便上面的代码产生以下输出：

{'name_1': 'value_1', 'name_2': 'value_2'}

其中name_1 和name_2 都是字符串，即DummyClass 实例的name 属性。

我不想更改可视化或打印，我想实际更改内存中的键。

Answer 1

__repr__ 方法用于生成dict 用于显示键的字符串。在dict 创建或更新期间没有修改密钥的机制，也没有dict 特定的方式来修改密钥的显示方式。

默认情况下，DummyClass 从object 继承其__repr__ 方法，这就是产生<...> 字符串的原因。覆盖 __repr__ 以返回从 name 属性构建的字符串。

class DummyClass:

    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return f"'{self.name}'"

---

有人向create a key-transforming dictionary 提出过提案，但被拒绝了。它本来可以让你写类似的东西

d = TransformDict(lambda x: x.name)
d[obj_1] = ...
d[obj_2] = ...

由此产生的dict 的行为就像'name1' 和'name2' 是密钥一样（同时仍允许访问原始密钥obj_1 和obj_2，如有必要）。