временной блок кода в Python, не помещая его в функцию

Question

Я хотел бы выделить блок кода, не помещая его в отдельную функцию. например:

def myfunc: 
    # some code here 
    t1 = time.time() 
    # block of code to time here 
    t2 = time.time() 
    print "Code took %s seconds." %(str(t2-t1)) 

однако, я хотел бы сделать это с помощью модуля timeit в более чистой дороге, но я не хочу, чтобы сделать отдельную функцию для блока кода.

спасибо.

Answer 1

Вы можете сделать это с помощью инструкции with. Например:

import time  
from contextlib import contextmanager 

@contextmanager 
def measureTime(title): 
    t1 = time.clock() 
    yield 
    t2 = time.clock() 
    print '%s: %0.2f seconds elapsed' % (title, t2-t1) 

Чтобы использовать так:

def myFunc(): 
    #... 

    with measureTime('myFunc'): 
     #block of code to time here 

    #... 

Answer 2

Вы можете создать переменную для обозначения блока кода, который требуется время, поставив этот блок внутри тройные кавычки Python. Затем используйте эту переменную при создании экземпляра объекта timeit. Несколько следующий пример из Python timeit docs, я придумал следующее:

import timeit 
code_block = """\ 
total = 0 
for cnt in range(0, 1000): 
    total += cnt 
print total 
""" 
tmr = timeit.Timer(stmt=code_block) 
print tmr.timeit(number=1) 

Что для меня напечатал:

0,000341892242432

(Где 499500 является выходного блока времени, а 0.000341892242432 - время работы.)

Answer 3

Согласно Skilldrickanswer есть глупый модуль, который, я думаю, может помочь: BlockLogginInator.

import time 
from blocklogginginator import logblock # logblock is the alias 

with logblock(name='one second'): 
    """" 
    Our code block. 
    """ 
    time.sleep(1) 

---

>>> Block "one second" started 
>>> Block "one second" ended (1.001)