枚举中的腌制物体？

Question

我正在从事一个工作项目，并且在腌制方面有问题。

这是我要做的事情的简化：

import pickle
from enum import Enum


class SiteInformation:
    def __init__(self, name, location):
        self.name = name
        self.location = location

        # Would actually be doing a number of calcs with name and location and storing
        # as members.
        self.time_zone = 'UTC'


class Site(Enum):
    GN = SiteInformation('North Site', 'Location String1')
    GS = SiteInformation('South Site', 'Location String2')


if __name__ == '__main__':
    print('Pickling...')
    with open('site_test', 'wb') as f:
        pickle.dump(Site.GN, f)
        pickle.dump(Site.GS, f)

    print('Unpickling...')
    with open('site_test', 'rb') as f:
        data = pickle.load(f)
    print(data)

当我运行此操作时，它在未挑剔的过程中失败：

Traceback (most recent call last):
  File "/Users/seb/Development/Scheduler/scripts/pickle_test.py", line 27, in <module>
    data = pickle.load(f)
  File "/Applications/Anaconda/anaconda3/envs/Scheduler/lib/python3.10/enum.py", line 385, in __call__
    return cls.__new__(cls, value)
  File "/Applications/Anaconda/anaconda3/envs/Scheduler/lib/python3.10/enum.py", line 710, in __new__
    raise ve_exc
ValueError: <__main__.SiteInformation object at 0x7fd77007ed40> is not a valid Site

我认为这与__ INT __ INT __ INT __在SiteInformation中调用 ，但是我不太确定如何解决此问题，因为我在腌制方面几乎没有经验。我尝试在siteInformation中实现__ epr __方法，但这无非是使valueerror更可读。

任何帮助将不胜感激。

Answer 1

在Python 3.11中，这将按照书面形式工作。在较早的Pythons中，您需要添加__ eq __（加上__ Hash __，如果您想要set> set> set> set（）和dict（dict）支持 ¹ ）启用siteInformation的不同实例，其值相同的值相等：

class SiteInformation:
    def __init__(self, name, location):
        self.name = name
        self.location = location

        # Would actually be doing a number of calcs with name and location and storing
        # as members.
        self.time_zone = 'UTC'

    def __eq__(self, other):
        return self.name == other.name and self.location == other.location

---

注意：__ eq __ eq __仅显示的目的。一个“真实” __ eq __看起来像：

def __eq__(self, other):
    if not isinstance(other, self.__class__):
        return NotImplemented
    return self.name == other.name and self.location == other.location

__ hash __看起来像：

def __hash__(self):
    return hash(self.name + self.location)

---

1：__ hash ____ ____，将对象放在set（），或用作dict（）中的键。

Answer 2

看来这已经解决了问题，并使代码基库更加清洁（基于Planet enum python文档中的示例）：

class Site(Enum):
    GN = ('North Site', 'Location String1')
    GS = ('South Site', 'Location String2')
    
    def __init__(self, site_name: str, location: str):
        self.site_name = site_name
        self.location = location
        self.time_zone = 'UTC'
        # Insert other member calculations here