仅导入类静态方法

时间:2012-09-26 14:03:32

标签: python inheritance import

我在基类中有以下装饰器:

class BaseTests(TestCase):
    @staticmethod
    def check_time(self, fn):
        @wraps(fn)
        def test_wrapper(*args,**kwargs):
            # do checks ...
        return test_wrapper

以下类继承自BaseTests:

from path.base_posting import BaseTests
from path.base_posting.BaseTests import check_time  # THIS LINE DOES NOT WORK!

class SpecificTest(BaseTests):

    @check_time # use the decorator
    def test_post(self):
        # do testing ...

我想在上面的SpecificTest中使用装饰器,而不必使用BaseTests.check_time,因为在原始代码中他们有很长的名字,我必须在很多地方使用它。有什么想法吗?

编辑: 我决定在BaseTests文件中使check_time成为一个独立的函数,然后只需导入

from path.base_posting import BaseTests, check_time

1 个答案:

答案 0 :(得分:14)

简单地说

check_time = BaseTests.check_time

在你的第二个模块中:


from module_paths.base_posting import BaseTests
check_time = BaseTests.check_time

class SpecificTest(BaseTests):

    @check_time # use the decorator
    def test_post(self):
        # do testing ...

您可能还想重新考虑使check_time成为静态方法,因为看起来您的用例更多地将其用作独立函数而不是静态方法。