Eu tenho um script python que funciona exatamente como deveria, mas preciso escrever o tempo de execução. Pesquisei no Google que deveria usartimeit mas não consigo fazê-lo funcionar.

Meu script Python fica assim:

import sys
import getopt
import timeit
import random
import os
import re
import ibm_db
import time
from string import maketrans
myfile = open("results_update.txt", "a")

for r in range(100):
    rannumber = random.randint(0, 100)

    update = "update TABLE set val = %i where MyCount >= '2010' and MyCount < '2012' and number = '250'" % rannumber
    #print rannumber

    conn = ibm_db.pconnect("dsn=myDB","usrname","secretPWD")

for r in range(5):
    print "Run %s\n" % r        
    ibm_db.execute(query_stmt)
 query_stmt = ibm_db.prepare(conn, update)

myfile.close()
ibm_db.close(conn)

O que eu preciso é do tempo que leva para executar a consulta e gravá-la no arquivoresults_update.txt. O objetivo é testar uma instrução de atualização para meu banco de dados com diferentes índices e mecanismos de ajuste.