import time from time import perf_counter, sleep from functools import wraps from typing import Callable, Any # ---- get time decorator ------------------------------------------- def get_time(func: Callable) -> Callable: @wraps(func) def wrapper(*args, **kwargs) -> Any: # Note that timing your code once isn't the most reliable # option for timing your code. Look into the timeit module # for more accurate timing. start_time: float = perf_counter() result: Any = func(*args, **kwargs) end_time: float = perf_counter() print(f'"{func.__name__}()" ' +\ f'took {end_time - start_time:.3f} seconds to execute') return result return wrapper # ---- Sample function 1 -------------------------------------------- @get_time def connect() -> None: print('Connecting...') sleep(2) print('Connected!') # ---- Sample function 2 -------------------------------------------- @get_time def fifty_million_loops() -> None: fifty_million: int = int(5e7) print('Looping...') for _ in range(fifty_million): pass print('Done looping!') # ---- main function ------------------------------------------------ def main() -> None: fifty_million_loops() connect() # ---- main --------------------------------------------------------- if __name__ == '__main__': main()