Testing text2img command (will be replaced, only for tests); added exception handling, queue and database table with generated images

bot/utils/trace_exception.py

@@ -0,0 +1,47 @@
+import os
+import traceback
+import contextlib
+import re
+
+
+class PrettyException:
+    def __init__(self, e: Exception):
+        self.pretty_exception = f'❌ Error! Report it to admins: \n' \
+                                f'🐊 <code>{e.__traceback__.tb_frame.f_code.co_filename.replace(os.getcwd(), "")}' \
+                                f'</code>:{e.__traceback__.tb_frame.f_lineno} \n' \
+                                f'😍 {e.__class__.__name__} \n' \
+                                f'👉 {"".join(traceback.format_exception_only(e)).strip()} \n\n' \
+                                f'⬇️ Trace: \n' \
+                                f'{self.get_full_stack()}'
+
+    @staticmethod
+    def get_full_stack():
+        full_stack = traceback.format_exc().replace(
+            "Traceback (most recent call last):\n", ""
+        )
+
+        line_regex = r'  File "(.*?)", line ([0-9]+), in (.+)'
+
+        def format_line(line: str) -> str:
+            filename_, lineno_, name_ = re.search(line_regex, line).groups()
+            with contextlib.suppress(Exception):
+                filename_ = os.path.basename(filename_)
+
+            return (
+                f"🤯 <code>{filename_}:{lineno_}</code> (<b>in</b>"
+                f" <code>{name_}</code> call)"
+            )
+
+        full_stack = "\n".join(
+            [
+                format_line(line)
+                if re.search(line_regex, line)
+                else f"<code>{line}</code>"
+                for line in full_stack.splitlines()
+            ]
+        )
+
+        return full_stack
+
+    def __str__(self):
+        return self.pretty_exception