在 python 模块中收集一些特定的类

Question

我有一个包含 classes C1, C2, CInvalid3, C4 的包，它们在不同的子模块中定义。

my_package/
    __init__.py
    sub_package1/
        __init__.py
        sub_module1.py # contains C1, C2
    sub_package2/
        __init__.py
        sub_module2.py # contains CInvalid3, C4

给定一个包含一些class名字的字符串，我想初始化相应的class。例如，

some_valid_class_name = 'C1'
some_valid_class = all_valid_classes[some_valid_class_name]()

目前我在 my_package 下创建了一个模块 all_classes.py，其中包含

all_valid_classes = {} # I want the dict contain C1, C2 and C4

def this_is_a_valid_class(cls):
    all_valid_classes[cls.__name__] = cls
    return cls

而在 sub_module1.py 和 sub_module2.py 中，我使用

from my_package.all_classes import this_is_a_valid_class

@this_is_a_valid_class
class C1(object):
    pass

并添加

from .sub_package1 import sub_module1
from .sub_package2 import sub_module2

到my_package/__init__.py.

有没有更好的方法来处理这个问题？我在 my_package/__init__.py 中导入所有内容。当我添加新的 sub_package3.sub_module3.C99 时，我需要记住将 from .sub_package3 import sub_module3 添加到 my_package/__init__.py。我不认为这些是好主意。

Answer 1

要获得您要查找的内容，您可以做的是标记具有 class 属性的有效 classes:

class C1(object):
  _valid = True

然后动态导入包中的所有模块，在此过程中将有效的 classes 单独存储在 all_valid_classes 字典中。

关键是无论如何你都必须导入你想使用的所有 classes 的模块，所以你可以手动（就像你现在所做的那样）或动态地（在包中找到模块的文件名并导入它们）。

这个片段展示了这个概念（请注意，它真的很难看，只是作为概念验证，我是匆忙组装的！）。它将进入您的 my_module/__init__.py 文件：

import os
import fnmatch
from importlib import import_module
from types import TypeType

# dictionary with all valid classes
all_valid_classes = {}

# walk directory and load modules dynamically
this_path = os.path.dirname(__file__)
modules = []
for root, dirnames, filenames in os.walk(this_path):
    for filename in fnmatch.filter(filenames, '*.py'):

        # avoid this file importing itself
        full_path = os.path.join(root, filename)
        if full_path == os.path.join(this_path, '__init__.py'):
            continue

        # convert the filename to the module name
        trimmed_path = full_path[len(this_path) + 1:-3]
        module_name = trimmed_path.replace('/', '.')

        # import the module
        module = import_module('%s.%s' % (__name__, module_name))

        # find valid classes (classes with a '_valid = True' attribute
        for item_name in dir(module):
            item = getattr(module, item_name)
            if type(item) == TypeType:
                try:
                    valid_attr = getattr(item, '_valid')
                    if valid_attr is True:
                        all_valid_classes[item_name] = item
                except AttributeError:
                    pass

导入my_module后，all_valid_classes会包含C1、C2和C4（前提是标有_valid = True） .

ps。但是请注意，我个人会像您已经在做的那样手动进行导入，在我看来它更像 pythonic 和干净。

在 python 模块中收集一些特定的类

collect some specific classes in python module

python

oop

python-2.7

python-3.x

python-3.4

在 python 模块中收集一些特定的 类

collect some specific classes in python module

python

oop

python-2.7

python-3.x

python-3.4

在 python 模块中收集一些特定的类