Estava com problema para ver quanto de tempo estava no trabalho, essa ferramente diz quanto trabalhei e quanto falta

calcula_tempo_trabalho.py

from datetime import datetime, timedelta


def format_time(element):
    return str(element) if element >=10 else "0"+str(element)
    
def hours_to_str(time):
    hour = format_time(time.hour)
    minute = format_time(time.minute)
    return hour+":"+minute
    
JORNADA = datetime(1,1,1,8,0)
HORA_ALMOCAR = 12
print("hora de entrada")
_ = input()
entrada_hora, entrada_minuto = [int(x) for x in _.split(":")]
inicio = timedelta(hours=entrada_hora, minutes=entrada_minuto)

now = datetime.now()
tempo_almocado = timedelta(hours=1,minutes=0)
if(now.hour > HORA_ALMOCAR):
    print("tempo almocado")
    _ = input()
    almoco_hora, almoco_minuto = [int(x) for x in _.split(":")]
    tempo_almocado  = timedelta(hours=almoco_hora, minutes=almoco_minuto)
    inicio = inicio + tempo_almocado

corrente = timedelta(hours=now.hour, minutes=now.minute)    
tempo_passado  = corrente - inicio

print("tempo passado",tempo_passado)
resto = JORNADA-tempo_passado
print("restando {0} de {1}".format(hours_to_str(resto),JORNADA.hour))
saida = inicio + JORNADA + tempo_almocado
print("saida {0}".format(hours_to_str(saida)))
input()