From 9024e4e911448c15fe8fc34c374c7b61d7fde4e9 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Sat, 6 Mar 2021 20:10:45 -0500
Subject: [PATCH] [utils] (timeit)
---
src/polystar/utils/time.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/src/polystar/utils/time.py b/src/polystar/utils/time.py
index 8d2073f..f855fe6 100644
--- a/src/polystar/utils/time.py
+++ b/src/polystar/utils/time.py
@@ -1,5 +1,18 @@
from datetime import datetime
+from functools import wraps
+from typing import Callable
def create_time_id() -> str:
return datetime.now().strftime("%Y%m%d_%H%M%S")
+
+
+def time_it(f: Callable):
+ @wraps(f)
+ def _f(*args, **kwargs):
+ t = datetime.now()
+ rv = f(*args, **kwargs)
+ print(f"{f.__name__} took {datetime.now() - t}")
+ return rv
+
+ return _f
--
GitLab