sft-ext 0.5.0__py3-none-any.whl

sft_ext/errors/err_handling.py

@@ -0,0 +1,89 @@
+import sys
+from asyncio import CancelledError
+from functools import wraps
+from io import StringIO
+from logging import getLogger
+
+from aiologic import Event
+from rich.console import Console
+from sft_ext.errors.send_alert_email import send_alert_email
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+  from collections.abc import Callable, Coroutine
+
+main_module = sys.modules["__main__"]
+RICH_CONSOLE = getattr(main_module, "RICH_CONSOLE", None)
+
+logger = getLogger(__name__)
+
+
+FATAL_EVENT = Event()
+
+
+def handle_fatal_exc_sync[**TP, TR](func: Callable[TP, TR]) -> Callable[TP, TR | None]:
+  @wraps(func)
+  def wrapper(*args: TP.args, **kwargs: TP.kwargs) -> TR | None:
+    try:
+      return func(*args, **kwargs)
+    except CancelledError:
+      pass
+      raise  # raise whatever to make the type checker happy about return values
+    except BaseException as e:
+      if isinstance(e, CancelledError):
+        raise
+      logger.critical(f"Fatal exception in {func.__qualname__}: {e}", exc_info=True)
+
+      strio = StringIO()
+
+      tmp = Console(force_terminal=False, force_interactive=False, color_system=None, markup=False, file=strio, no_color=True)
+
+      with tmp.capture() as capture:
+        tmp.print_exception(show_locals=True)
+      content = capture.get()
+
+      send_alert_email(f"Fatal exception in {func.__qualname__}", f"{e}:\n\n{content}")
+      FATAL_EVENT.set()
+      return None
+
+  return func if __debug__ and __name__ != "__main__" else wrapper
+
+
+def handle_fatal_exc_async[**TP, TR](func: Callable[TP, Coroutine[None, None, TR]]) -> Callable[TP, Coroutine[None, None, TR | None]]:
+  @wraps(func)
+  async def wrapper(*args: TP.args, **kwargs: TP.kwargs) -> TR | None:
+    try:
+      return await func(*args, **kwargs)
+    except CancelledError:
+      pass
+      raise  # raise whatever to make the type checker happy about return values
+    except GeneratorExit:
+      pass
+      return None  # if a GeneratorExit is caught, that means a coroutine is being cancelled for a graceful shutdown.
+    except BaseException as e:
+      if isinstance(e, CancelledError):
+        raise
+      logger.critical(f"Fatal exception in {func.__qualname__}: {e}", exc_info=True)
+
+      strio = StringIO()
+
+      tmp = Console(force_terminal=False, force_interactive=False, color_system=None, markup=False, file=strio, no_color=True)
+
+      with tmp.capture() as capture:
+        tmp.print_exception(show_locals=True)
+      content = capture.get()
+
+      send_alert_email(f"Fatal exception in {func.__qualname__}", f"{e}:\n\n{content}")
+      FATAL_EVENT.set()
+      return None
+
+  return func if __debug__ and __name__ != "__main__" else wrapper
+
+
+if __name__ == "__main__":
+
+  @handle_fatal_exc_sync
+  def test_func():
+    raise ValueError("This is a test exception.")
+
+  test_func()

sft_ext/errors/send_alert_email.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+import smtplib
+import ssl
+import sys
+from email.message import EmailMessage
+from logging import getLogger
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+  from sft_ext.settings import BaseSettings
+
+main_module = sys.modules["__main__"]
+RICH_CONSOLE = getattr(main_module, "RICH_CONSOLE", None)
+
+logger = getLogger(__name__)
+
+
+try:
+  import sys
+
+  try:
+    settings_module = sys.modules["environment_init_vars"]
+    SETTINGS: BaseSettings = settings_module.SETTINGS
+  except KeyError:
+    settings_module = sys.modules["environment_settings"]
+    SETTINGS: BaseSettings = settings_module.SETTINGS()  # type: ignore
+except (KeyError, AttributeError):
+  from sft_ext.settings import BaseSettings
+
+  SETTINGS: BaseSettings = BaseSettings()  # type: ignore
+
+
+ALERTS_EMAIL = SETTINGS.alerts_email
+ALERTS_EMAIL_PWD = SETTINGS.alerts_email_pwd
+ALERTS_RECIPIENTS = SETTINGS.alerts_recipients
+
+
+def send_alert_email(subject: str, content: str) -> None:
+  if not ALERTS_RECIPIENTS:
+    logger.warning("Skipping alert email because no recipients are configured.")
+    return
+
+  msg = EmailMessage()
+  msg.set_content("View attachment")
+  msg["Subject"] = subject
+  msg["From"] = ALERTS_EMAIL
+  msg["To"] = ", ".join([str(recipient) for recipient in ALERTS_RECIPIENTS])
+  context = ssl.create_default_context()
+
+  msg.add_attachment(
+    "\ufeff" + content,  # UTF-8 BOM so Windows apps detect encoding correctly
+    subtype="plain",
+    filename="alert.txt",
+    charset="utf-8",
+  )
+
+  try:
+    with smtplib.SMTP(SETTINGS.alerts_smtp_server, SETTINGS.alerts_smtp_port) as server:
+      server.ehlo()
+      server.starttls(context=context)
+      server.ehlo()
+      server.login(ALERTS_EMAIL, ALERTS_EMAIL_PWD)
+      server.send_message(msg)
+    logger.debug("Alert email sent successfully.")
+  except Exception:
+    logger.error("Failed to send alert email.", exc_info=True)