Merge #140 - add pyproject.toml and versioning

1 files changed, 35 insertions, 0 deletions

diff --git a/snekbox/utils/logging.py b/snekbox/utils/logging.py
new file mode 100644
index 0000000..c15e3f1
--- /dev/null
+++ b/snekbox/utils/logging.py
@@ -0,0 +1,35 @@
+import logging
+import os
+import sys
+
+__all__ = ("FORMAT", "init_logger", "init_sentry")
+
+FORMAT = "%(asctime)s | %(process)5s | %(name)30s | %(levelname)8s | %(message)s"
+
+
+def init_logger(debug: bool) -> None:
+    """Initialise the root logger with a handler that outputs to stdout."""
+    log = logging.getLogger("snekbox")
+    log.setLevel(logging.DEBUG if debug else logging.INFO)
+    log.propagate = True
+
+    formatter = logging.Formatter(FORMAT)
+    handler = logging.StreamHandler(sys.stdout)
+    handler.setFormatter(formatter)
+    log.addHandler(handler)
+
+
+def init_sentry(version: str) -> None:
+    """Initialise the Sentry SDK if it's installed."""
+    try:
+        import sentry_sdk
+        from sentry_sdk.integrations.falcon import FalconIntegration
+    except ImportError:
+        return
+
+    sentry_sdk.init(
+        dsn=os.environ.get("SNEKBOX_SENTRY_DSN", ""),
+        integrations=[FalconIntegration()],
+        send_default_pii=True,
+        release=f"snekbox@{version}"
+    )