Skip to content

Commit

Permalink
Merge pull request #9 from zimran-tech/feature/additional-configs
Browse files Browse the repository at this point in the history
add default integrations
  • Loading branch information
talgat-abdraimov authored Dec 25, 2023
2 parents 43a3925 + 6ab0588 commit ba60edc
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions zimran/logging/utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,10 @@
import sys
from typing import Any

from loguru import logger
from sentry_sdk import init
from sentry_sdk.integrations.fastapi import FastApiIntegration
from sentry_sdk.integrations.loguru import LoguruIntegration

from zimran.logging.exceptions import InvalidEnvironmentError

Expand All @@ -25,10 +28,12 @@ def setup_logger(debug: bool) -> None:
logger.add(sys.stdout, level='INFO', serialize=True)


def setup_sentry(dsn: str, environment: str, **kwargs) -> None:
def setup_sentry(dsn: str, environment: str, **kwargs: dict[str, Any]) -> None:
try:
sample_rate = _get_sample_rate(environment)
except InvalidEnvironmentError:
return

init(dsn=dsn, environment=environment, sample_rate=sample_rate, **kwargs)
kwargs.setdefault('integrations', [FastApiIntegration(), LoguruIntegration()]) # type: ignore

init(dsn=dsn, environment=environment, sample_rate=sample_rate, **kwargs) # type: ignore

0 comments on commit ba60edc

Please sign in to comment.