我可以创建一个可以解包的类吗？

Question

例如：

x = (1, 2)
a,b = x

现在我想在x 是某个不是列表或元组的类的实例的情况下完成此操作。简单地覆盖 __getitem__ 或 __getslice__ 不起作用：

class Test(object):
    def __getitem__(self, key):
        return 1

a,b = Test()

产生ValueError: too many values to unpack。我可以在不继承list 或tuple（或它们各自的UserX 类）的情况下完成这项工作吗？或者这只是一些我无法使用的底层魔法？

Answer 1

您需要覆盖__iter__ 或__getitem__。这是一个使用__iter__的示例：

class Test(object):
    def __iter__(self):
        return iter([1, 2])

a,b = Test()

小心，__iter__ 需要返回一个迭代器。最简单的方法是将数据结构包装在 iter 中，或者让它从生成器中生成。

有关使用 __getitem__ 的示例，请参阅 jonrsharpe 的出色回答。

Answer 2

根据PEP-234，它定义了迭代器（强调我的）：

如果实现了一个对象，则可以用“for”对其进行迭代 __iter__() 或__getitem__()。

您可以使用__getitem__ 执行此操作，但它必须抛出IndexError 以指示您何时到达终点。您会看到 too many values to unpack，因为 Python 无法判断它已完成对您的对象的迭代。

这可行：

class Test(object):
    def __getitem__(self, key):
        if key > 1:
            raise IndexError
        return 1 

使用中：

>>> t = Test()
>>> a, b = t
>>> a
1
>>> b
1
>>> t[2]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 4, in __getitem__
IndexError