33

需要一种方法来检查对象是否是某个特定模块中任何类的实例。

我知道我可以通过从该模块显式导入每个类并使用元组检查来做到这一点：

from my_module import ClassOne, ClassTwo

>>> isinstance(my_obj, (ClassOne, ClassTwo))
True

但实际上，我从中导入的模块中有大量的类，并且显式地导入它们似乎是不必要的冗长，使用它们来构建一个巨大的元组，并对它进行类型检查。我尝试了一些方法来避免这种情况：

import my_module

# Test my_obj against the module itself
>>> isinstance(my_obj, my_module)
TypeError: isinstance() arg 2 must be a class, type, or tuple of classes and types

# Try to test against a wildcard attribute on my_module
>>> isinstance(my_obj, my_module.*)
SyntaxError: invalid syntax

#Try to build a tuple of clases with iteration to check against
>>> for klass in my_module:
TypeError: 'module' object is not iterable

有没有办法对 my_module 中的所有类进行类型检查，而无需在元组中明确命名它们？

可选背景信息：
我可能忽略了解决问题的更好方法——如果您想知道，情况如下：

我们正在将数据从 Google App Engine 应用程序导出到我们在 Rackspace 上托管的应用程序。我们使用 HTTP 请求序列化数据，pickle然后将其发送到我们的 Rackspace 服务器。

Google App Engine 数据库中的一些数据是 GAE 特定的数据类型，从 google.appengine.api.datastore_types 导入。如果这些数据类型中的任何一个通过网络连接到我们的 Rackspace 服务器，它们将引发错误 depickling，因为我们的 Rackspace 应用程序没有所需的 GAE 库。因此，在退出 GAE 的过程中，我正在检查是否有任何传出对象具有来自 google.appengine.api.datastore_types 的类型。如果是这样，我要么将它们转换为内置数据类型，要么从对象中删除该字段。

4

2 回答 2

21

您可以使用inspect.getmembers获取模块中的所有类：

inspect.getmembers(my_module,inspect.isclass)

这将返回名称-类对的列表。您只需要以下课程：

my_module_classes = tuple(x[1] for x in inspect.getmembers(my_module,inspect.isclass))

当我最初写这个答案时，我设法忽略的一件事是检查类__module__属性的能力。您可能可以检查是否__module__是您期望的那样：

from somewhere import module

if getattr(obj, '__module__', None) == module.__name__:
    # obj is from module.

这可能比isinstance检查大量的类名更便宜。

于 2013-01-28T20:41:57.927 回答
2

我使用了@mgilson 简单的解决方案：

from somewhere import module

if getattr(obj, '__module__', None) == module.__name__:
    # obj is from module.

并注意到如果你有一个模块树，并且你想确保它来自基本模块，你必须这样做：

from somewhere import module

if getattr(obj, '__module__', None).split('.')[0] == module.__name__:
    # obj is from module.

但是，如果您object来自内置它会引发异常，所以我会：

from somewhere import module

module_tree = getattr(obj, '__module__', None)
parent = module_tree.split('.')[0] if module_tree else None

if parent == module.__name__:
    # obj is from module.
于 2018-10-12T16:04:38.443 回答