Возврат генератора питона для анализа времени

Question

Я написал фрагмент, который использует генераторы для возврата прошедшего времени с момента последнего вызова.

def time_gen(start_time): 
    start = start_time 
    while True: 
     elapsed = time.time() - start 
     start = time.time() 
     yield elapsed 

def test(): 
    a = time_gen(time.time()) 
    for i in range(4): 
     time.sleep(1) 
     print(a.next()) #prints about 1 second 

Кажется, что хорошо работает. Но мне не нравится, как пользователь должен пройти в начале. Есть ли способ написать функцию, которая возвращает генератор с временем начала, уже прошедшим в качестве параметра?

#Broken Code 
def time_closure(): 
    def time_gen(start_time): 
     start = start_time 
     while True: 
      elapsed = time.time() - start 
      start = time.time() 
      yield elapsed 

    a = time_gen(time.time()) 
    return a 

#Usage of Broken Code 
a = time_closure() 
a.next() 

Answer 1

Вам не обязательно передавать время старта генератору. Вы можете просто сделать start = time.time() в своей первой строке, а остальные оставить как есть