From 8e7e694d81e674db63049145a35972df8ad2e3fa Mon Sep 17 00:00:00 2001
From: abc <98614666+xtekky@users.noreply.github.com>
Date: Fri, 20 Oct 2023 19:04:13 +0100
Subject: =?UTF-8?q?~=20|=C2=A0updated=20g4f.api?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

new api and requirements
---
 g4f/api/_logging.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 g4f/api/_logging.py

(limited to 'g4f/api/_logging.py')

diff --git a/g4f/api/_logging.py b/g4f/api/_logging.py
new file mode 100644
index 00000000..e91dff76
--- /dev/null
+++ b/g4f/api/_logging.py
@@ -0,0 +1,32 @@
+import sys,logging
+
+from loguru import logger
+
+def __exception_handle(e_type, e_value, e_traceback):
+    if issubclass(e_type, KeyboardInterrupt):
+        print('\nBye...')
+        sys.exit(0)
+
+    sys.__excepthook__(e_type, e_value, e_traceback)
+
+class __InterceptHandler(logging.Handler):
+    def emit(self, record):
+        try:
+            level = logger.level(record.levelname).name
+        except ValueError:
+            level = record.levelno
+
+        frame, depth = logging.currentframe(), 2
+        while frame.f_code.co_filename == logging.__file__:
+            frame = frame.f_back
+            depth += 1
+
+        logger.opt(depth=depth, exception=record.exc_info).log(
+            level, record.getMessage()
+        )
+
+def hook_except_handle():
+    sys.excepthook = __exception_handle
+
+def hook_logging(**kwargs):
+    logging.basicConfig(handlers=[__InterceptHandler()], **kwargs)
-- 
cgit v1.2.3