Tracking user account creation event in posthog (#2237)

* capture user account creation in posthog via apisix middleware

* capture user account creation in posthog via apisix middleware

* add $current_url

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Add PostHogEvents constant

* update PostHogEvents to use Enum

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix flaky test

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* forgot value

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: rachellougee

main/constants.py

@@ -1,5 +1,7 @@
 """main constants"""
 
+from enum import Enum
+
 from rest_framework import status
 
 PERMISSION_DENIED_ERROR_TYPE = "PermissionDenied"
@@ -38,3 +40,11 @@
     "ul",
 }
 ALLOWED_HTML_ATTRIBUTES = {}
+
+
+class PostHogEvents(Enum):
+    """
+    Enum for PostHog events
+    """
+
+    ACCOUNT_CREATED = "user_account_created"

main/middleware/apisix_user.py

@@ -8,8 +8,10 @@
 from django.contrib.auth import get_user_model, login, logout
 from django.contrib.auth.middleware import RemoteUserMiddleware
 from django.db.models import Q
+from posthog import Posthog
 
 from authentication.api import user_created_actions
+from main.constants import PostHogEvents
 
 log = logging.getLogger(__name__)
 
@@ -99,6 +101,22 @@ def get_user_from_apisix_headers(request, decoded_headers, original_header):
             "get_user_from_apisix_headers: User %s not found, created new",
             global_id,
         )
+        # Send user creation event to PostHog
+        posthog = Posthog(
+            settings.POSTHOG_PROJECT_API_KEY, host=settings.POSTHOG_API_HOST
+        )
+        posthog.capture(
+            user.id,
+            event=PostHogEvents.ACCOUNT_CREATED.value,
+            properties={
+                "$current_url": request.build_absolute_uri(),
+                "global_id": global_id,
+                "email": email,
+                "first_name": decoded_headers.get("first_name", ""),
+                "last_name": decoded_headers.get("last_name", ""),
+            },
+        )
+
         user.set_unusable_password()
         user.is_active = True
         user.save()

main/middleware/apisix_user_test.py

@@ -44,6 +44,12 @@ def setup_test_database():
 def test_get_request(mocker, mock_login):
     """Test that a valid request creates a new user."""
     close_old_connections()
+    mocker.patch(
+        "main.middleware.apisix_user.Posthog",
+        mocker.Mock(
+            capture=mocker.ANY,
+        ),
+    )
     mock_request = mocker.Mock(
         META={
             "HTTP_X_USERINFO": b64encode(json.dumps(apisix_user_info).encode()),