qgis插件的Timeit模塊

Question

我想使用python模塊timeit來爲我的QGIS插件中的某些功能定時。

在這裏，我調用了函數內的函數，我在最後一個函數結束時調用它。不過，看起來這個插件的運行時間比平常要長，我想知道我是否在錯誤的地方調用了定時器。有沒有更好的方法來設置它？

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

OK，我不知道您的具體情況，但我會設置它雖然裝飾：

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