fix: set posthoganalytics key in asgi

posthog/asgi.py

@@ -1,4 +1,6 @@
 import os
+
+from django.conf import settings
 from django.core.asgi import get_asgi_application
 from django.http.response import HttpResponse
 
@@ -18,4 +20,22 @@
     return inner
 
 
-application = lifetime_wrapper(get_asgi_application())
+# PostHogConfig.ready() handles setting the global analytics key in WSGI. The same code couldn't run
+# in ASGI because ready() doesn't expose an async interface.
+def wrap_debug_analytics(func):
+    async def inner(scope, receive, send):
+        if not getattr(inner, "debug_analytics_initialized", False):
+            from posthog.utils import aset_debugging_analytics_key
+
+            await aset_debugging_analytics_key()
+            # Set a flag to indicate that the analytics key has been set, so we don't run the code on every request.
+            inner.debug_analytics_initialized = True
+        return await func(scope, receive, send)
+
+    return inner
+
+
+if settings.DEBUG:
+    application = lifetime_wrapper(wrap_debug_analytics(get_asgi_application()))
+else:
+    application = lifetime_wrapper(get_asgi_application())

posthog/utils.py

@@ -40,14 +40,14 @@
 from rest_framework import serializers
 from rest_framework.request import Request
 from sentry_sdk import configure_scope
-from posthog.exceptions_capture import capture_exception
 
 from posthog.cloud_utils import get_cached_instance_license, is_cloud
 from posthog.constants import AvailableFeature
 from posthog.exceptions import (
     RequestParsingError,
     UnspecifiedCompressionFallbackParsingError,
 )
+from posthog.exceptions_capture import capture_exception
 from posthog.git import get_git_branch, get_git_commit_short
 from posthog.metrics import KLUDGES_COUNTER
 from posthog.redis import get_client
@@ -517,6 +517,37 @@ def get_self_capture_api_token(user: Optional[Union["AbstractBaseUser", "Anonymo
     return None
 
 
+async def aset_debugging_analytics_key():
+    """
+    ASGI alternative to get_self_capture_api_token.
+    """
+    from django.apps import apps
+
+    User = apps.get_model("posthog", "User")
+    Team = apps.get_model("posthog", "Team")
+    try:
+        user = (
+            await User.objects.filter(last_login__isnull=False)
+            .order_by("-last_login")
+            .select_related("current_team")
+            .afirst()
+        )
+        # Get the current user's team (or first team in the instance) to set self capture configs
+        team = None
+        if user and getattr(user, "team", None):
+            team = user.current_team
+        else:
+            team = await Team.objects.only("api_token").afirst()
+        local_api_key = team.api_token
+    except:
+        local_api_key = None
+    # apps.ready() is already called, so reset the already set config.
+    if local_api_key is not None:
+        posthoganalytics.api_key = local_api_key
+        posthoganalytics.host = settings.SITE_URL
+        posthoganalytics.disabled = False
+
+
 def get_default_event_name(team: "Team"):
     from posthog.models import EventDefinition