Python中可散列类与Enum的交互

时间:2016-11-25 14:06:36

标签: python python-2.7 hash enums magic-methods

我已经定义了一个我试图制作的类。此外,还有一个枚举,它使用此类的对象作为其枚举成员的值。

from enum import Enum


class Dummy(object):
    def __init__(self, name, property_):
        self.name = name             # can only be a string
        self.property = property_    # can only be a string

    def __hash__(self):
        # print "Hash called for ", self
        # print("----")
        return hash(self.property)

    def __eq__(self, other):
        # print "Eq called for: "
        # print self
        # print other
        return (self.property == other.property)

    def __ne__ (self, other):
        return not (self == other)

    def __str__(self):
        return (self.name + "$" + self.property)


class Property(Enum):
    cool = Dummy("foo", "cool")
    hot = Dummy("bar", "hot")

虽然这很好用,但我注意到 - 通过取消评论print语句 - 为两个枚举成员值调用__hash____eq__魔术方法。为什么会这样?这些仅在散列和相等检查期间使用过吗?

此外,如果我将枚举类更改为以下内容,那么所有地狱都会崩溃。

class Property(Enum):
    cool = Dummy("foo", "cool")
    hot = [Dummy("bar-1", "hot-1"), Dummy("bar-2", "hot-2")]

__eq__魔术方法似乎是针对与Dummy对应的Property.cool对象和与Property.hot对应的列表调用的,从输出中可以看出:

Hash called for  foo$cool
----
Eq called for: 
foo$cool
[<__main__.Dummy object at 0x7fd36633f2d0>, <__main__.Dummy object at 0x7fd36633f310>]
----
Traceback (most recent call last):
  File "test.py", line 28, in <module>
    class Property(Enum):
  File "/blah/.local/lib/python2.7/site-packages/enum/__init__.py", line 237, in __new__
    if canonical_member.value == enum_member._value_:
  File "test.py", line 19, in __eq__
    return (self.property is other.property)
AttributeError: 'list' object has no attribute 'property'

为什么会这样?为什么首先调用魔术方法,为什么__eq__在类对象和列表上调用?

请注意,这只是一个有代表性的示例,真正的用例使得这个设计 - 枚举值作为可散列类对象的列表 - 看起来不那么奇怪。

1 个答案:

答案 0 :(得分:2)

Enum类正在比较其成员对象值,以查看是否有另一个别名。例如,在以下枚举中,ab都代表相同的值,因此只有a应显示在成员列表中(别名不是)

class A(Enum):
    a=1
    b=1

您可以通过查看执行相等性检查的行的源代码来验证这一点:source

对于散列,这样做是为了提供枚举成员的按值查找。同样,这可以在source code

中找到