Question

我正在尝试创建一个只接受可迭代参数的方法，例如list，tuple，set或dict。

这是我的代码：

class MjmMenuControl(MjmBaseMenu):
    def __init__(self, items=None):
        iterables = (dict, list, set, tuple)
        for iterable in iterables:
            if isinstance(items, iterable):
                ...
                break

然而，我想知道是否有更简单的方法，例如if isinstance(items, <iterable_base_class>):，但我找不到任何东西。

我已经尝试为list等找到基类，但它们似乎都来自object：

>>> inspect.getmro(list)
(<class 'list'>, <class 'object'>)
>>> inspect.getmro(tuple)
(<class 'tuple'>, <class 'object'>)

这是可能的，还是我必须坚持可怕的for循环？

Answer 1

是的，请使用collections.abc.Iterable：

>>> from collections import abc
>>> isinstance(set(), abc.Iterable)
True
>>> isinstance((), abc.Iterable)
True
>>> isinstance([], abc.Iterable)
True
>>> isinstance('', abc.Iterable)
True
>>> isinstance({}, abc.Iterable)
True
>>> isinstance(0, abc.Iterable)
False

但是，我只是让方法假设它被传递一个可迭代的，调用者的责任是确保传递正确的类型。

{3}}是Python 3.3中的新功能;它的内容以前只存在于collections.abc module中，尽管即使在Python 3.3及更高版本中，这些名称仍然可以在旧位置使用。

Answer 2

collections.abc.Iterable选项的一个弱点是它无法应对像这样的自定义可迭代类......

from collections.abc import Iterable

class MyIterable(object):
    def __getitem__(self, index):
        if index >= 10:
            raise IndexError
        return index

>>> myiter = MyIterable()
>>> isinstance(myiter, Iterable)
False
>>> [i for i in myiter]
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

经常声称“Pythonic”这样做的方法是使用duck typing，即尝试将对象视为可迭代，并在失败时处理异常。例如......

class MjmMenuControl(MjmBaseMenu):
    def __init__(self, items=None):
        try:
            for item in items:
                do_something_with(item)
        except TypeError:
            # If we get here, 'items' is not iterable
            do_something_else()

...但是，两种方法都有将字符串处理为与列表相同的弱点，这通常不是您想要的。通常你想测试一个可迭代的容器，这稍微不同，所以通常会看到将字符串视为特殊情况的代码......

class MjmMenuControl(MjmBaseMenu):
    def __init__(self, items=None):
        if isinstance(items, (str, bytes)):
            do_something_else()
        else:
            try:
                for item in items:
                    do_something_with(item)
            except TypeError:
                # If we get here, 'items' is not iterable
                do_something_else()

......这有点乱，但完成了工作。

迭代的常见基类？

2 个答案: