Tengo un script de Python que funciona como debería, pero necesito escribir el tiempo de ejecución. He buscado en Google que debería usartimeit pero parece que no puedo hacer que funcione.

Mi script de Python se ve así:

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)

Lo que necesito es el tiempo que lleva ejecutar la consulta y escribirla en el archivoresults_update.txt. El propósito es probar una declaración de actualización para mi base de datos con diferentes índices y mecanismos de ajuste.