From 446b3b50856d42a01cf06127ce3f734a59f63074 Mon Sep 17 00:00:00 2001
From: Conatum <cona@thewisewolf.dev>
Date: Thu, 8 Jun 2023 17:31:30 +0300
Subject: [PATCH] rewrite: More thorough debug mode.

---
 scripts/start_leo_debug.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/scripts/start_leo_debug.py b/scripts/start_leo_debug.py
index 258212b6..ce58e1c9 100755
--- a/scripts/start_leo_debug.py
+++ b/scripts/start_leo_debug.py
@@ -3,12 +3,30 @@
 import sys
 import os
 import tracemalloc
+import asyncio
+import logging
 
-tracemalloc.start()
 
 sys.path.insert(0, os.path.join(os.getcwd()))
 sys.path.insert(0, os.path.join(os.getcwd(), "src"))
 
+tracemalloc.start()
+event_loop = asyncio.get_event_loop()
+
+
+def loop_exception_handler(loop, context):
+    print(context)
+    task: asyncio.Task = context.get('task', None)
+    if task is not None:
+        addendum = f"<Task name='{task.name}' stack='{task.get_stack()}'>"
+        message = context.get('message', '')
+        context['message'] = ' '.join((message, addendum))
+    loop.default_exception_handler(context)
+
+
+event_loop.set_exception_handler(loop_exception_handler)
+event_loop.set_debug(enabled=True)
+
 
 if __name__ == '__main__':
     from bot import _main