Spaces:
Runtime error
Runtime error
File size: 3,000 Bytes
a9e7556 18f1b47 a9e7556 18f1b47 a9e7556 18f1b47 a9e7556 18f1b47 9ae6797 a9e7556 18f1b47 9ae6797 a9e7556 18f1b47 a9e7556 843b43f 18f1b47 a9e7556 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import time
import streamlit as st
from os import environ
class Timer():
total = 0
place_holder = None
expander = None
timer_list = []
def __init__(self, pt_name, en_name):
self.pt_name = pt_name
self.en_name = en_name
if environ['PORTUGUESE'] == 'true':
self.portuguese = True
elif environ['PORTUGUESE'] == 'false':
self.portuguese = False
else:
raise EnvironmentError
if not Timer.place_holder:
if self.portuguese:
Timer.place_holder = st.empty()
Timer.expander = Timer.place_holder.expander('Ver progresso')
else:
Timer.place_holder = st.empty()
Timer.expander = Timer.place_holder.expander('See progress')
self.display = Timer.expander.empty()
Timer.timer_list.append(self)
def __enter__(self):
if self.portuguese:
self.display.info(f'Executando "{self.pt_name}"...')
else:
self.display.info(f'Running "{self.en_name}"...')
self.start_time = time.time()
def __exit__(self, type, value, traceback):
end_time = time.time()
elapsed_time = end_time - self.start_time
self.display.empty()
if self.portuguese:
Timer.expander.info(f'"{self.pt_name}" terminou em {elapsed_time:.2f} s')
else:
Timer.expander.info(f'"{self.en_name}" finished in {elapsed_time:.2f} s')
# for manually starting the timer
def start(self):
if self.portuguese:
self.display.warning(f'Executando "{self.pt_name}"...')
else:
self.display.warning(f'Running "{self.en_name}"...')
self.start_time = time.time()
# for manually stopping the timer
def stop(self):
end_time = time.time()
elapsed_time = end_time - self.start_time
Timer.total += elapsed_time
self.display.empty()
if self.portuguese:
Timer.expander.warning(f'"{self.pt_name}" terminou em {elapsed_time:.2f} s')
else:
Timer.expander.warning(f'"{self.en_name}" finished in {elapsed_time:.2f} s')
def reset():
Timer.total = 0
if Timer.place_holder:
Timer.place_holder.empty()
Timer.place_holder = None
Timer.expander = None
def show_total():
if environ['PORTUGUESE'] == 'true':
Timer.expander.success(f'Tempo de execução total: {Timer.total:.2f} s')
elif environ['PORTUGUESE'] == 'false':
Timer.expander.success(f'Total elapsed time: {Timer.total:.2f} s')
def time_it(pt_name, en_name):
def decorator(func):
def wrapper(*args, **kwargs):
timer = Timer(pt_name, en_name)
timer.start()
result = func(*args, **kwargs)
timer.stop()
return result
return wrapper
return decorator |