only call init on sentry sdk if enabled

fullerzz · Sep 29, 2024 · 684bf4b · 684bf4b
1 parent 513f601
commit 684bf4b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/src/smolvault/config.py b/src/smolvault/config.py
@@ -13,6 +13,7 @@ class Settings(BaseSettings):
     user_whitelist: str
     users_limit: int
     daily_upload_limit_bytes: int
+    sentry_enabled: bool
     sentry_dsn: str
 
     model_config = SettingsConfigDict(env_file=".env")

diff --git a/src/smolvault/main.py b/src/smolvault/main.py
@@ -33,11 +33,12 @@
 logger = logging.getLogger(__name__)
 
 settings: Settings = get_settings()
-sentry_sdk.init(
-    dsn=settings.sentry_dsn,
-    traces_sample_rate=1.0,  # Set traces_sample_rate to 1.0 to capture 100%
-    profiles_sample_rate=1.0,
-)
+if settings.sentry_enabled:
+    sentry_sdk.init(
+        dsn=settings.sentry_dsn,
+        traces_sample_rate=1.0,  # Set traces_sample_rate to 1.0 to capture 100%
+        profiles_sample_rate=1.0,
+    )
 
 app = FastAPI(title="smolvault", docs_url=None, redoc_url=None)
 

diff --git a/tests/testing.env b/tests/testing.env
@@ -5,3 +5,5 @@ SMOLVAULT_CACHE="./uploads/"
 USER_WHITELIST="1,2"
 USERS_LIMIT="3"
 DAILY_UPLOAD_LIMIT_BYTES="50000"
+SENTRY_ENABLED="false"
+SENTRY_DSN="none"