qgis 插件的 Timeit 模块

Question

我想使用 python 模块 timeit 来为我的 QGIS 插件中的一些功能计时。

在这里，我在最后一个函数末尾调用的函数中调用了 time it 函数。不过，该插件的运行时间似乎比平时更长，我想知道我是否在错误的地方调用了计时器。有没有更好的设置方法？

class myPluginName:

    def firstFunction(self):
        ...
        self.secondFunction()

    def secondFunction(self):
        ...
        self.timeThings()

    def run(self):
        self.firstFunction()

    def timeThings(self):
        QMessageBox.information(None, 'First Function', 'Time : %s' % timeit.timeit(self.firstFunction,number=1)
        QMessageBox.information(None, 'Second Function', 'Time : %s' % timeit.timeit(self.secondFunction,number=1)

更新：遵循一些建议后，我尝试以下列方式实现包装器。但是，我得到了TypeError: firstFunction() takes exactly 1 argument (2 given) on ret = func(**args, **kwargs)

def time_func(func):
    try:
        name = func.__name__
    except:
        name = func.f__name
    def tf_wrapper(*args, **kwargs):
        t = time.time()
        ret = func(*args, **kwargs)
        QMessageLog.logMessage("{}: {}".format(name, time.time() - t))
        return ret
    return tf_wrapper

class myPlugin:
    def initGui(self):
        QObject.connect(self.dlg.ui.comboBox,SIGNAL("currentIndexChanged(int)"), self.firstFunction)
    @time_func
    def firstFunc(self):
        registry = QgsMapLayerRegistry.instance()
        firstID = str(self.dlg.ui.firstCombo.itemData(self.dlg.ui.firstCombo.currentIndex()))
        secondID = str(self.dlg.ui.secondCombo.itemData(self.dlg.ui.secondCombo.currentIndex()))
        self.firstLayer = registry.mapLayer(firstID)
        self.secondLayer = registry.mapLayer(secondID)

    @time_func
    def secondFunc(self):
        ...
        self.thirdFunc()
    def thirdFunct(self):
        ...
    def run(self):
        self.dlg.ui.firstCombo.clear()
        self.dlg.ui.secondCombo.clear()
        for layer in self.iface.legendInterface().layers():
            if layer.type() == QgsMapLayer.VectorLayer:
                self.dlg.ui.firstCombo.addItem(layer.name(), layer.id())
                self.dlg.ui.secondCombo.addItem(layer.name(), layer.id())
        result = self.dlg.exec_()
        if result == 1:
            self.secondFunction()

Answer 1

好的，我不知道你的确切情况，但我会通过装饰器进行设置：

import time

def time_func(func):
    try:
        name = func.__name__
    except:
        name = func.f_name
    def tf_wrapper(*args, **kwargs):
        t = time.time()
        ret = func(*args, **kwargs)
        print("{}: {}".format(name, time.time() - t))  # Or use QMessageBox
        return ret
    return tf_wrapper

class myPluginName:
    @time_func
    def firstFunction(self):
        pass

    @time_func
    def secondFunction(self):
        pass

    def run(self):
        self.firstFunction()
myPluginName().firstFunction()

使用此代码，任何包含在time_func 中的函数都将有时间来执行返回时打印的函数及其名称。例如。运行它将打印：

firstFunction: 1.430511474609375e-06

对于您的TypeError，您需要更改；

    def firstFunction(self):
        pass

收件人：

    def firstFunction(self, index):
        pass