Skip to content

Commit

Permalink
feat: send telemetry heartbeat post migrations are applied (#3351)
Browse files Browse the repository at this point in the history
  • Loading branch information
tushar5526 committed Jan 30, 2024
1 parent c8db43e commit 31af594
Showing 1 changed file with 16 additions and 10 deletions.
26 changes: 16 additions & 10 deletions api/telemetry/apps.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,21 +2,27 @@

from django.apps import AppConfig
from django.conf import settings
from django.db.models.signals import post_migrate

logger = logging.getLogger(__name__)


def send_telemetry_callback(sender, **kwargs):
if settings.ENABLE_TELEMETRY:
try:
from .telemetry import SelfHostedTelemetryWrapper

telemetry = SelfHostedTelemetryWrapper()
telemetry.send_heartbeat()
except Exception as e:
logger.debug(
"Failed to send Telemetry data to Flagsmith. Exception was %s" % e
)
pass


class TelemetryConfig(AppConfig):
name = "telemetry"

def ready(self):
if settings.ENABLE_TELEMETRY:
try:
from .telemetry import SelfHostedTelemetryWrapper

telemetry = SelfHostedTelemetryWrapper()
telemetry.send_heartbeat()
except Exception as e:
logger.debug(
"Failed to send Telemetry data to Flagsmith. Exception was %s" % e
)
post_migrate.connect(send_telemetry_callback, sender=self)

0 comments on commit 31af594

Please sign in to comment.