Skip to content

Commit 7ba9d36

Browse files
committed
CH-123 add parameters to Sentry init
1 parent c958a9e commit 7ba9d36

1 file changed

Lines changed: 18 additions & 10 deletions

File tree

  • libraries/cloudharness-common/cloudharness/sentry

libraries/cloudharness-common/cloudharness/sentry/__init__.py

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import requests
33

44
from cloudharness.utils.env import get_common_service_cluster_address
5+
from cloudharness.applications import get_current_configuration
56

67
sentry_environment = os.environ.get("DOMAIN", "Production")
78

@@ -29,32 +30,39 @@ def get_dsn(appname):
2930
else:
3031
return None
3132

32-
def init(appname, traces_sample_rate=0):
33+
def init(appname=None, traces_sample_rate=0, integrations=None, **kwargs):
3334
"""
3435
Init cloudharness Sentry functionality for the current app
3536
3637
Args:
3738
appname: the slug of the application
38-
traces_sample_rate: performance trace sample rate
39+
others/kwargs: additional parameters for sentry_sdk.init
40+
3941
4042
Usage examples:
4143
import cloudharness.sentry as sentry
4244
sentry.init('notifications')
4345
"""
46+
if appname is None:
47+
appname = get_current_configuration().harness.name
48+
4449
dsn = get_dsn(appname)
50+
4551
if dsn:
4652
import sentry_sdk
47-
try:
48-
from flask import current_app as app
49-
from sentry_sdk.integrations.flask import FlaskIntegration
50-
integrations = [FlaskIntegration()]
51-
except:
52-
integrations = []
53+
if not integrations:
54+
try:
55+
from flask import current_app as app
56+
from sentry_sdk.integrations.flask import FlaskIntegration
57+
integrations = [FlaskIntegration()]
58+
except:
59+
integrations = []
5360
sentry_sdk.init(
5461
dsn=dsn,
55-
traces_sample_rate=traces_sample_rate,
5662
environment=sentry_environment,
57-
integrations=integrations
63+
integrations=integrations,
64+
traces_sample_rate=traces_sample_rate,
65+
**kwargs
5866
)
5967

6068
__all__ = ['get_dsn', 'init']

0 commit comments

Comments
 (0)