From 63e6f9eb24e3da63accd3a69e2f825aad932927c Mon Sep 17 00:00:00 2001
From: Ian Woodard <17186604+IanWoodard@users.noreply.github.com>
Date: Fri, 13 Sep 2024 11:05:24 -0700
Subject: [PATCH] fix(cli): Adding atexit handler to avoid sending sentry logs
 to console

---
 src/main.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main.py b/src/main.py
index 5c049194..c3333404 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import argparse
+import atexit
 
 import sentry_sdk
 from commands import list_dependencies
@@ -21,6 +22,11 @@
 )
 
 
+@atexit.register
+def cleanup() -> None:
+    sentry_sdk.flush()
+
+
 def main() -> None:
     parser = argparse.ArgumentParser(
         description="DevServices CLI tool for managing Docker Compose services."