From 880948829aeafb546f4d4aa1cfbeab533e835b5c Mon Sep 17 00:00:00 2001
From: BoYanZh <bomingzh@sjtu.edu.cn>
Date: Thu, 16 Sep 2021 00:45:43 +0800
Subject: [PATCH] feat: teapot: for all methods log exception in loguru

---
 joint_teapot/teapot.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/joint_teapot/teapot.py b/joint_teapot/teapot.py
index bcf54ff..feaeafe 100644
--- a/joint_teapot/teapot.py
+++ b/joint_teapot/teapot.py
@@ -1,5 +1,6 @@
+import functools
 from datetime import datetime
-from typing import List
+from typing import Any, Callable, List
 
 from loguru import logger
 
@@ -8,6 +9,29 @@ from joint_teapot.utils import first
 from joint_teapot.workers import Canvas, Git, Gitea
 
 
+def for_all_methods(decorator: Callable[..., Any]) -> Callable[..., Any]:
+    def decorate(cls: Any) -> Any:
+        print(type(cls))
+        for attr in cls.__dict__:  # there's propably a better way to do this
+            if callable(getattr(cls, attr)):
+                setattr(cls, attr, decorator(getattr(cls, attr)))
+        return cls
+
+    return decorate
+
+
+def log_exception_in_loguru(func: Callable[..., Any]) -> Callable[..., Any]:
+    @functools.wraps(func)
+    def decorator(*args: Any, **kwargs: Any) -> Any:
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            logger.exception(e)
+
+    return decorator
+
+
+@for_all_methods(log_exception_in_loguru)
 class Teapot:
     _canvas = None
     _gitea = None