changelog

.github/workflows/call-flags-project-board.yml

@@ -1,17 +0,0 @@
-# This workflow is used to call the flags-project-board workflow when a pull request is opened, ready for review, review requested, synchronized, converted to draft, or reopened.
-# It is used to update the feature flags project board with the pull request information.
-
-name: Call Feature Flags Project Workflow
-
-on:
-    pull_request:
-        types: [opened, ready_for_review, review_requested, synchronize, converted_to_draft, reopened]
-
-jobs:
-    call-flags-project:
-        uses: PostHog/.github/.github/workflows/flags-project-board.yml@main
-        with:
-            pr_number: ${{ github.event.pull_request.number }}
-            pr_node_id: ${{ github.event.pull_request.node_id }}
-            is_draft: ${{ github.event.pull_request.draft }}
-        secrets: inherit

.github/workflows/ci.yml

@@ -18,16 +18,17 @@ jobs:
               with:
                   python-version: 3.11.11
 
-            - name: Install uv
-              uses: astral-sh/setup-uv@0c5e2b8115b80b4c7c5ddf6ffdd634974642d182 # v5.4.1
+            - uses: actions/cache@5a3ec84eff668545956fd18022155c47e93e2684
               with:
-                  enable-cache: true
-                  pyproject-file: 'pyproject.toml'
+                  path: ~/.cache/pip
+                  key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}
+                  restore-keys: |
+                      ${{ runner.os }}-pip-
 
             - name: Install dev dependencies
-              shell: bash
               run: |
-                  UV_PROJECT_ENVIRONMENT=$pythonLocation uv sync --extra dev
+                  python -m pip install -e .[dev]
+              if: steps.cache.outputs.cache-hit != 'true'
 
             - name: Check formatting with ruff
               run: |
@@ -54,16 +55,9 @@ jobs:
               with:
                   python-version: ${{ matrix.python-version }}
 
-            - name: Install uv
-              uses: astral-sh/setup-uv@0c5e2b8115b80b4c7c5ddf6ffdd634974642d182 # v5.4.1
-              with:
-                  enable-cache: true
-                  pyproject-file: 'pyproject.toml'
-
-            - name: Install test dependencies
-              shell: bash
+            - name: Install requirements.txt dependencies with pip
               run: |
-                  UV_PROJECT_ENVIRONMENT=$pythonLocation uv sync --extra test
+                  python -m pip install -e .[test]
 
             - name: Run posthog tests
               run: |

.github/workflows/release.yaml

@@ -24,23 +24,15 @@ jobs:
 
       - name: Set up Python
         uses: actions/setup-python@8d9ed9ac5c53483de85588cdf95a591a75ab9f55
-        with:
-          python-version: 3.11.11
 
-      - name: Install uv
-        uses: astral-sh/setup-uv@0c5e2b8115b80b4c7c5ddf6ffdd634974642d182 # v5.4.1
-        with:
-            enable-cache: true
-            pyproject-file: 'pyproject.toml'
-            
       - name: Detect version
         run: echo "REPO_VERSION=$(python3 posthog/version.py)" >> $GITHUB_ENV
 
       - name: Prepare for building release
-        run: uv sync --extra dev
+        run: pip install -U pip setuptools packaging wheel twine
 
-      - name: Push releases to PyPI
-        run: uv run make release && uv run make release_analytics
+      - name: Push release to PyPI
+        run: make release && make release_analytics
 
       - name: Create GitHub release
         uses: actions/create-release@0cb9c9b65d5d1901c1f53e5e66eaf4afd303e70e # v1

.gitignore

@@ -17,4 +17,3 @@ posthog-analytics
 .coverage
 pyrightconfig.json
 .env
-.DS_Store

BEFORE_SEND.md

@@ -1,237 +0,0 @@
-# Before Send Hook
-
-The `before_send` parameter allows you to modify or filter events before they are sent to PostHog. This is useful for:
-
-- **Privacy**: Removing or masking sensitive data (PII)
-- **Filtering**: Dropping unwanted events (test events, internal users, etc.)
-- **Enhancement**: Adding custom properties to all events
-- **Transformation**: Modifying event names or property formats
-
-## Basic Usage
-
-```python
-import posthog
-from typing import Optional, Dict, Any
-
-def my_before_send(event: Dict[str, Any]) -> Optional[Dict[str, Any]]:
-    """
-    Process event before sending to PostHog.
-    
-    Args:
-        event: The event dictionary containing 'event', 'distinct_id', 'properties', etc.
-    
-    Returns:
-        Modified event dictionary to send, or None to drop the event
-    """
-    # Your processing logic here
-    return event
-
-# Initialize client with before_send hook
-client = posthog.Client(
-    api_key="your-project-api-key",
-    before_send=my_before_send
-)
-```
-
-## Common Use Cases
-
-### 1. Filter Out Events
-
-```python
-from typing import Optional, Any
-
-def filter_events_by_property_or_event_name(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Drop events from internal users or test environments."""
-    properties = event.get("properties", {})
-    
-    # Choose some property from your events
-    event_source = properties.get("event_source", "")
-    if event_source.endswith("internal"):
-        return None  # Drop the event
-    
-    # Filter out test events
-    if event.get("event") == "test_event":
-        return None
-    
-    return event
-```
-
-### 2. Remove/Mask PII Data
-
-```python
-from typing import Optional, Any
-
-def scrub_pii(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Remove or mask personally identifiable information."""
-    properties = event.get("properties", {})
-    
-    # Mask email but keep domain for analytics
-    if "email" in properties:
-        email = properties["email"]
-        if "@" in email:
-            domain = email.split("@")[1]
-            properties["email"] = f"***@{domain}"
-        else:
-            properties["email"] = "***"
-    
-    # Remove sensitive fields entirely
-    sensitive_fields = ["my_business_info", "secret_things"]
-    for field in sensitive_fields:
-        properties.pop(field, None)
-    
-    return event
-```
-
-### 3. Add Custom Properties
-
-```python
-from typing import Optional, Any
-
-from datetime import datetime
-from typing import Optional, Any
-
-def add_context(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Add custom properties to all events."""
-    if "properties" not in event:
-        event["properties"] = {}
-    
-    event["properties"].update({
-        "app_version": "2.1.0",
-        "environment": "production", 
-        "processed_at": datetime.now().isoformat()
-    })
-    
-    return event
-```
-
-### 4. Transform Event Names
-
-```python
-from typing import Optional, Any
-
-def normalize_event_names(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Convert event names to a consistent format."""
-    original_event = event.get("event")
-    if original_event:
-        # Convert to snake_case
-        normalized = original_event.lower().replace(" ", "_").replace("-", "_")
-        event["event"] = f"app_{normalized}"
-    
-    return event
-```
-
-### 5. Log and drop in "dev" mode
-
-When running in local dev often, you want to log but drop all events
-
-
-```python
-from typing import Optional, Any
-
-def log_and_drop_all(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Convert event names to a consistent format."""
-    print(event)
-    
-    return None
-```
-
-### 6. Combined Processing
-
-```python
-from typing import Optional, Any
-
-def comprehensive_processor(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    """Apply multiple transformations in sequence."""
-    
-    # Step 1: Filter unwanted events
-    if should_drop_event(event):
-        return None
-    
-    # Step 2: Scrub PII
-    event = scrub_pii(event)
-    
-    # Step 3: Add context
-    event = add_context(event)
-    
-    # Step 4: Normalize names
-    event = normalize_event_names(event)
-    
-    return event
-
-def should_drop_event(event: dict[str, Any]) -> bool:
-    """Determine if event should be dropped."""
-    # Your filtering logic
-    return False
-```
-
-## Error Handling
-
-If your `before_send` function raises an exception, PostHog will:
-
-1. Log the error 
-2. Continue with the original, unmodified event
-3. Not crash your application
-
-```python
-from typing import Optional, Any
-
-def risky_before_send(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    # If this raises an exception, the original event will be sent
-    risky_operation()
-    return event
-```
-
-## Complete Example
-
-```python
-import posthog
-from typing import Optional, Any
-import re
-
-def production_before_send(event: dict[str, Any]) -> Optional[dict[str, Any]]:
-    try:
-        properties = event.get("properties", {})
-        
-        # 1. Filter out bot traffic
-        user_agent = properties.get("$user_agent", "")
-        if re.search(r'bot|crawler|spider', user_agent, re.I):
-            return None
-        
-        # 2. Filter out internal traffic  
-        ip = properties.get("$ip", "")
-        if ip.startswith("192.168.") or ip.startswith("10."):
-            return None
-        
-        # 3. Scrub email PII but keep domain
-        if "email" in properties:
-            email = properties["email"]
-            if "@" in email:
-                domain = email.split("@")[1]
-                properties["email"] = f"***@{domain}"
-        
-        # 4. Add custom context
-        properties.update({
-            "app_version": "1.0.0",
-            "build_number": "123"
-        })
-        
-        # 5. Normalize event name
-        if event.get("event"):
-            event["event"] = event["event"].lower().replace(" ", "_")
-        
-        return event
-        
-    except Exception as e:
-        # Log error but don't crash
-        print(f"Error in before_send: {e}")
-        return event  # Return original event on error
-
-# Usage
-client = posthog.Client(
-    api_key="your-api-key",
-    before_send=production_before_send
-)
-
-# All events will now be processed by your before_send function
-client.capture("user_123", "Page View", {"url": "/home"})
-```

CHANGELOG.md

@@ -1,102 +1,10 @@
-# 6.0.1
+## 4.3.3 - 2025-06-06
 
-- fix: response `$process_person_profile` property when passed to capture
-
-# 6.0.0
-
-This release contains a number of major breaking changes:
-- feat: make distinct_id an optional parameter in posthog.capture and related functions
-- feat: make capture and related functions return `Optional[str]`, which is the UUID of the sent event, if it was sent
-- fix: remove `identify` (prefer `posthog.set()`), and `page` and `screen` (prefer `posthog.capture()`)
-- fix: delete exception-capture specific integrations module. Prefer the general-purpose django middleware as a replacement for the django `Integration`.
-
-To migrate to this version, you'll mostly just need to switch to using named keyword arguments, rather than positional ones. For example:
-```python
-# Old calling convention
-posthog.capture("user123", "button_clicked", {"button_id": "123"})
-# New calling convention
-posthog.capture(distinct_id="user123", event="button_clicked", properties={"button_id": "123"})
-
-# Better pattern
-with posthog.new_context():
-    posthog.identify_context("user123")
-
-    # The event name is the first argument, and can be passed positionally, or as a keyword argument in a later position
-    posthog.capture("button_pressed")
-```
-
-Generally, arguments are now appropriately typed, and docstrings have been updated. If something is unclear, please open an issue, or submit a PR!
-
-# 5.4.0 - 2025-06-20
-
-- feat: add support to session_id context on page method
-
-# 5.3.0 - 2025-06-19
-
-- fix: safely handle exception values
-
-# 5.2.0 - 2025-06-19
-
-- feat: construct artificial stack traces if no traceback is available on a captured exception
-
-## 5.1.0 - 2025-06-18
-
-- feat: session and distinct ID's can now be associated with contexts, and are used as such
-- feat: django http request middleware
-
-## 5.0.0 - 2025-06-16
-
-- fix: removed deprecated sentry integration
-
-## 4.10.0 - 2025-06-13
-
-- fix: no longer fail in autocapture.
-
-## 4.9.0 - 2025-06-13
-
-- feat(ai): track reasoning and cache tokens in the LangChain callback
-
-## 4.8.0 - 2025-06-10
-
-- fix: export scoped, rather than tracked, decorator
-- feat: allow use of contexts without error tracking
-
-## 4.7.0 - 2025-06-10
-
-- feat: add support for parse endpoint in responses API (no longer beta)
-
-## 4.6.2 - 2025-06-09
-
-- fix: replace `import posthog` with direct method imports
-
-## 4.6.1 - 2025-06-09
-
-- fix: replace `import posthog` in `posthoganalytics` package
-
-## 4.6.0 - 2025-06-09
-
-- feat: add additional user and request context to captured exceptions via the Django integration
-- feat: Add `setup()` function to initialise default client
-
-## 4.5.0 - 2025-06-09
-
-- feat: add before_send callback (#249)
-
-## 4.4.2- 2025-06-09
-
-- empty point release to fix release automation
-
-## 4.4.1 2025-06-09
-
-- empty point release to fix release automation
-
-## 4.4.0 - 2025-06-09
-
-- Use the new `/flags` endpoint for all feature flag evaluations (don't fall back to `/decide` at all)
+Add `setup()` function to initialise default client
 
 ## 4.3.2 - 2025-06-06
 
-1. Add context management:
+Add context management:
 
 - New context manager with `posthog.new_context()`
 - Tag functions: `posthog.tag()`, `posthog.get_tags()`, `posthog.clear_tags()`
@@ -104,10 +12,12 @@ Generally, arguments are now appropriately typed, and docstrings have been updat
   - `@posthog.scoped` - Creates context and captures exceptions thrown within the function
 - Automatic deduplication of exceptions to ensure each exception is only captured once
 
-2. fix: feature flag request use geoip_disable (#235)
-3. chore: pin actions versions (#210)
-4. fix: opinionated setup and clean fn fix (#240)
-5. fix: release action failed (#241)
+## 4.2.1 - 2025-6-05
+
+1. fix: feature flag request use geoip_disable (#235)
+2. chore: pin actions versions (#210)
+3. fix: opinionated setup and clean fn fix (#240)
+4. fix: release action failed (#241)
 
 ## 4.2.0 - 2025-05-22
 

Makefile

@@ -16,42 +16,22 @@ release_analytics:
 	rm -rf posthoganalytics
 	mkdir posthoganalytics
 	cp -r posthog/* posthoganalytics/
-	find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthog /from posthoganalytics /g' {} \;
-	find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthog\./from posthoganalytics\./g' {} \;
-	find ./posthoganalytics -name "*.bak" -delete
+	find ./posthoganalytics -type f -exec sed -i '' -e 's/from posthog /from posthoganalytics /g' {} \;
+	find ./posthoganalytics -type f -exec sed -i '' -e 's/from posthog\./from posthoganalytics\./g' {} \;
 	rm -rf posthog
 	python setup_analytics.py sdist bdist_wheel
 	twine upload dist/*
 	mkdir posthog
-	find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthoganalytics /from posthog /g' {} \;
-	find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthoganalytics\./from posthog\./g' {} \;
-	find ./posthoganalytics -name "*.bak" -delete
+	find ./posthoganalytics -type f -exec sed -i '' -e 's/from posthoganalytics /from posthog /g' {} \;
+	find ./posthoganalytics -type f -exec sed -i '' -e 's/from posthoganalytics\./from posthog\./g' {} \;
 	cp -r posthoganalytics/* posthog/
 	rm -rf posthoganalytics
-	rm -f pyproject.toml
-	cp pyproject.toml.backup pyproject.toml
-	rm -f pyproject.toml.backup
 
 e2e_test:
 	.buildscripts/e2e.sh
 
-prep_local:
-	rm -rf ../posthog-python-local
-	mkdir ../posthog-python-local
-	cp -r . ../posthog-python-local/
-	cd ../posthog-python-local && rm -rf dist build posthoganalytics .git
-	cd ../posthog-python-local && mkdir posthoganalytics
-	cd ../posthog-python-local && cp -r posthog/* posthoganalytics/
-	cd ../posthog-python-local && find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthog /from posthoganalytics /g' {} \;
-	cd ../posthog-python-local && find ./posthoganalytics -type f -name "*.py" -exec sed -i.bak -e 's/from posthog\./from posthoganalytics\./g' {} \;
-	cd ../posthog-python-local && find ./posthoganalytics -name "*.bak" -delete
-	cd ../posthog-python-local && rm -rf posthog
-	cd ../posthog-python-local && sed -i.bak 's/from version import VERSION/from posthoganalytics.version import VERSION/' setup_analytics.py
-	cd ../posthog-python-local && rm setup_analytics.py.bak
-	cd ../posthog-python-local && sed -i.bak 's/"posthog"/"posthoganalytics"/' setup.py
-	cd ../posthog-python-local && rm setup.py.bak
-	cd ../posthog-python-local && python -c "import setup_analytics" 2>/dev/null || true
-	@echo "Local copy created at ../posthog-python-local"
-	@echo "Install with: pip install -e ../posthog-python-local"
+django_example:
+	python -m pip install -e ".[sentry]"
+	cd sentry_django_example && python manage.py runserver 8080
 
-.PHONY: test lint release e2e_test prep_local
+.PHONY: test lint release e2e_test

README.md

@@ -16,13 +16,11 @@ Please see the [Python integration docs](https://posthog.com/docs/integrations/p
 
 ### Testing Locally
 
-We recommend using [uv](https://docs.astral.sh/uv/). It's super fast.
-
-1. Run `uv venv env` (creates virtual environment called "env")
-    * or `python3 -m venv env`
+1. Run `python3 -m venv env` (creates virtual environment called "env")
+    * or `uv venv env`
 2. Run `source env/bin/activate` (activates the virtual environment)
-3. Run `uv sync --extra dev --extra test` (installs the package in develop mode, along with test dependencies)
-    * or `pip install -e ".[dev,test]"`
+3. Run `python3 -m pip install -e ".[test]"` (installs the package in develop mode, along with test dependencies)
+    * or `uv pip install -e ".[test]"`
 4. you have to run `pre-commit install` to have auto linting pre commit
 5. Run `make test`
   1. To run a specific test do `pytest -k test_no_api_key`
@@ -34,7 +32,7 @@ uv python install 3.9.19
 uv python pin 3.9.19
 uv venv env
 source env/bin/activate
-uv sync --extra dev --extra test
+uv pip install --editable ".[dev,test]"
 pre-commit install
 make test
 ```
@@ -43,24 +41,24 @@ make test
 
 Assuming you have a [local version of PostHog](https://posthog.com/docs/developing-locally) running, you can run `python3 example.py` to see the library in action.
 
+### Running the Django Sentry Integration Locally
+
+There's a sample Django project included, called `sentry_django_example`, which explains how to use PostHog with Sentry.
+
+There's 2 places of importance (Changes required are all marked with TODO in the sample project directory)
+
+1. Settings.py
+    1. Input your Sentry DSN
+    2. Input your Sentry Org and ProjectID details into `PosthogIntegration()`
+    3. Add `POSTHOG_DJANGO` to settings.py. This allows the `PosthogDistinctIdMiddleware` to get the distinct_ids
+
+2. urls.py
+    1. This includes the `sentry-debug/` endpoint, which generates an exception
+
+To run things: `make django_example`. This installs the posthog-python library with the sentry-sdk add-on, and then runs the django app.
+Also start the PostHog app locally.
+Then navigate to `http://127.0.0.1:8080/sentry-debug/` and you should get an event in both Sentry and PostHog, with links to each other.
+
 ### Releasing Versions
 
-Updates are released automatically using GitHub Actions when `version.py` is updated on `master`. After bumping `version.py` in `master` and adding to `CHANGELOG.md`, the [release workflow](https://github.com/PostHog/posthog-python/blob/master/.github/workflows/release.yaml) will automatically trigger and deploy the new version.
-
-If you need to check the latest runs or manually trigger a release, you can go to [our release workflow's page](https://github.com/PostHog/posthog-python/actions/workflows/release.yaml) and dispatch it manually, using workflow from `master`.
-
-
-### Testing changes locally with the PostHog app
-
-You can run `make prep_local`, and it'll create a new folder alongside the SDK repo one called `posthog-python-local`, which you can then import into the posthog project by changing pyproject.toml to look like this:
-```toml
-dependencies = [
-    ...
-    "posthoganalytics" #NOTE: no version number
-    ...
-]
-...
-[tools.uv.sources]
-posthoganalytics = { path = "../posthog-python-local" }
-```
-This'll let you build and test SDK changes fully locally, incorporating them into your local posthog app stack. It mainly takes care of the `posthog -> posthoganalytics` module renaming. You'll need to re-run `make prep_local` each time you make a change, and re-run `uv sync --active` in the posthog app project.
+Updated are released using GitHub Actions: after bumping `version.py` in `master` and adding to `CHANGELOG.md`, go to [our release workflow's page](https://github.com/PostHog/posthog-python/actions/workflows/release.yaml) and dispatch it manually, using workflow from `master`.

example.py

@@ -41,9 +41,9 @@ print(
 
 # Capture an event
 posthog.capture(
+    "distinct_id",
     "event",
-    distinct_id="distinct_id",
-    properties={"property1": "value", "property2": "value"},
+    {"property1": "value", "property2": "value"},
     send_feature_flags=True,
 )
 
@@ -65,35 +65,31 @@ exit()
 posthog.alias("distinct_id", "new_distinct_id")
 
 posthog.capture(
-    "event2",
-    distinct_id="new_distinct_id",
-    properties={"property1": "value", "property2": "value"},
+    "new_distinct_id", "event2", {"property1": "value", "property2": "value"}
 )
 posthog.capture(
+    "new_distinct_id",
     "event-with-groups",
-    distinct_id="new_distinct_id",
-    properties={"property1": "value", "property2": "value"},
+    {"property1": "value", "property2": "value"},
     groups={"company": "id:5"},
 )
 
 # # Add properties to the person
-posthog.set(
-    distinct_id="new_distinct_id", properties={"email": "something@something.com"}
-)
+posthog.identify("new_distinct_id", {"email": "something@something.com"})
 
 # Add properties to a group
 posthog.group_identify("company", "id:5", {"employees": 11})
 
 # properties set only once to the person
-posthog.set_once(distinct_id="new_distinct_id", properties={"self_serve_signup": True})
+posthog.set_once("new_distinct_id", {"self_serve_signup": True})
 
 
 posthog.set_once(
-    distinct_id="new_distinct_id", properties={"self_serve_signup": False}
+    "new_distinct_id", {"self_serve_signup": False}
 )  # this will not change the property (because it was already set)
 
-posthog.set(distinct_id="new_distinct_id", properties={"current_browser": "Chrome"})
-posthog.set(distinct_id="new_distinct_id", properties={"current_browser": "Firefox"})
+posthog.set("new_distinct_id", {"current_browser": "Chrome"})
+posthog.set("new_distinct_id", {"current_browser": "Firefox"})
 
 
 # #############################################################################

mypy-baseline.txt

@@ -22,23 +22,19 @@ posthog/client.py:0: error: Library stubs not installed for "six"  [import-untyp
 posthog/client.py:0: note: Hint: "python3 -m pip install types-six"
 posthog/client.py:0: error: Name "queue" already defined (by an import)  [no-redef]
 posthog/client.py:0: error: Need type annotation for "queue"  [var-annotated]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "Any | list[Any]", variable has type "None")  [assignment]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "dict[Any, Any]", variable has type "None")  [assignment]
-posthog/client.py:0: error: "None" has no attribute "__iter__" (not iterable)  [attr-defined]
-posthog/client.py:0: error: Statement is unreachable  [unreachable]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "Any | dict[Any, Any]", variable has type "None")  [assignment]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "Any | dict[Any, Any]", variable has type "None")  [assignment]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "dict[Never, Never]", variable has type "None")  [assignment]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "dict[Never, Never]", variable has type "None")  [assignment]
-posthog/client.py:0: error: Right operand of "and" is never evaluated  [unreachable]
-posthog/client.py:0: error: Incompatible types in assignment (expression has type "Poller", variable has type "None")  [assignment]
-posthog/client.py:0: error: "None" has no attribute "start"  [attr-defined]
-posthog/client.py:0: error: "None" has no attribute "get"  [attr-defined]
-posthog/client.py:0: error: Statement is unreachable  [unreachable]
-posthog/client.py:0: error: Statement is unreachable  [unreachable]
+posthog/client.py:0: error: Item "None" of "Any | None" has no attribute "get"  [union-attr]
+simulator.py:0: error: Unexpected keyword argument "anonymous_id" for "capture"  [call-arg]
+posthog/__init__.py:0: note: "capture" defined here
+simulator.py:0: error: Unexpected keyword argument "anonymous_id" for "identify"  [call-arg]
+posthog/__init__.py:0: note: "identify" defined here
+simulator.py:0: error: Unexpected keyword argument "traits" for "identify"  [call-arg]
+posthog/__init__.py:0: note: "identify" defined here
 example.py:0: error: Statement is unreachable  [unreachable]
+posthog/sentry/posthog_integration.py:0: error: Statement is unreachable  [unreachable]
 posthog/ai/utils.py:0: error: Need type annotation for "output" (hint: "output: list[<type>] = ...")  [var-annotated]
 posthog/ai/utils.py:0: error: Function "builtins.any" is not valid as a type  [valid-type]
 posthog/ai/utils.py:0: note: Perhaps you meant "typing.Any" instead of "any"?
 posthog/ai/utils.py:0: error: Function "builtins.any" is not valid as a type  [valid-type]
 posthog/ai/utils.py:0: note: Perhaps you meant "typing.Any" instead of "any"?
+sentry_django_example/sentry_django_example/settings.py:0: error: Need type annotation for "ALLOWED_HOSTS" (hint: "ALLOWED_HOSTS: list[<type>] = ...")  [var-annotated]
+sentry_django_example/sentry_django_example/settings.py:0: error: Incompatible types in assignment (expression has type "str", variable has type "None")  [assignment]

posthog/__init__.py

@@ -1,43 +1,21 @@
 import datetime  # noqa: F401
-from typing import Callable, Dict, Optional, Any  # noqa: F401
-from typing_extensions import Unpack
+import warnings
+from typing import Callable, Dict, List, Optional, Tuple  # noqa: F401
 
-from posthog.args import OptionalCaptureArgs, OptionalSetArgs, ExceptionArg
 from posthog.client import Client
-from posthog.contexts import (
-    new_context as inner_new_context,
-    scoped as inner_scoped,
-    tag as inner_tag,
-    set_context_session as inner_set_context_session,
-    identify_context as inner_identify_context,
-)
+from posthog.exception_capture import Integrations  # noqa: F401
+from posthog.scopes import clear_tags, get_tags, new_context, scoped, tag
 from posthog.types import FeatureFlag, FlagsAndPayloads
 from posthog.version import VERSION
 
 __version__ = VERSION
 
 """Context management."""
-
-
-def new_context(fresh=False, capture_exceptions=True):
-    return inner_new_context(fresh=fresh, capture_exceptions=capture_exceptions)
-
-
-def scoped(fresh=False, capture_exceptions=True):
-    return inner_scoped(fresh=fresh, capture_exceptions=capture_exceptions)
-
-
-def set_context_session(session_id: str):
-    return inner_set_context_session(session_id)
-
-
-def identify_context(distinct_id: str):
-    return inner_identify_context(distinct_id)
-
-
-def tag(name: str, value: Any):
-    return inner_tag(name, value)
-
+new_context = new_context
+tag = tag
+get_tags = get_tags
+clear_tags = clear_tags
+tracked = scoped
 
 """Settings."""
 api_key = None  # type: Optional[str]
@@ -55,6 +33,7 @@ feature_flags_request_timeout_seconds = 3  # type: int
 super_properties = None  # type: Optional[Dict]
 # Currently alpha, use at your own risk
 enable_exception_autocapture = False  # type: bool
+exception_autocapture_integrations = []  # type: List[Integrations]
 log_captured_exceptions = False  # type: bool
 # Used to determine in app paths for exception autocapture. Defaults to the current working directory
 project_root = None  # type: Optional[str]
@@ -64,100 +43,206 @@ privacy_mode = False  # type: bool
 default_client = None  # type: Optional[Client]
 
 
-# NOTE - this and following functions take unpacked kwargs because we needed to make
-# it impossible to write `posthog.capture(distinct-id, event-name)` - basically, to enforce
-# the breaking change made between 5.3.0 and 6.0.0. This decision can be unrolled in later
-# versions, without a breaking change, to get back the type information in function signatures
-def capture(event: str, **kwargs: Unpack[OptionalCaptureArgs]) -> Optional[str]:
+def capture(
+    distinct_id,  # type: str
+    event,  # type: str
+    properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
+    timestamp=None,  # type: Optional[datetime.datetime]
+    uuid=None,  # type: Optional[str]
+    groups=None,  # type: Optional[Dict]
+    send_feature_flags=False,
+    disable_geoip=None,  # type: Optional[bool]
+):
+    # type: (...) -> Tuple[bool, dict]
     """
     Capture allows you to capture anything a user does within your system, which you can later use in PostHog to find patterns in usage, work out which features to improve or where people are giving up.
 
     A `capture` call requires
+    - `distinct id` which uniquely identifies your user
     - `event name` to specify the event
     - We recommend using [verb] [noun], like `movie played` or `movie updated` to easily identify what your events mean later on.
 
-    Capture takes a number of optional arguments, which are defined by the `OptionalCaptureArgs` type.
+    Optionally you can submit
+    - `properties`, which can be a dict with any information you'd like to add
+    - `groups`, which is a dict of group type -> group key mappings
 
     For example:
     ```python
-    # Enter a new context (e.g. a request/response cycle, an instance of a background job, etc)
-    with posthog.new_context():
-        # Associate this context with some user, by distinct_id
-        posthog.identify_context('some user')
+    posthog.capture('distinct id', 'opened app')
+    posthog.capture('distinct id', 'movie played', {'movie_id': '123', 'category': 'romcom'})
 
-        # Capture an event, associated with the context-level distinct ID ('some user')
-        posthog.capture('movie started')
-
-        # Capture an event associated with some other user (overriding the context-level distinct ID)
-        posthog.capture('movie joined', distinct_id='some-other-user')
-
-        # Capture an event with some properties
-        posthog.capture('movie played', properties={'movie_id': '123', 'category': 'romcom'})
-
-        # Capture an event with some properties
-        posthog.capture('purchase', properties={'product_id': '123', 'category': 'romcom'})
-        # Capture an event with some associated group
-        posthog.capture('purchase', groups={'company': 'id:5'})
-
-        # Adding a tag to the current context will cause it to appear on all subsequent events
-        posthog.tag_context('some-tag', 'some-value')
-
-        posthog.capture('another-event') # Will be captured with `'some-tag': 'some-value'` in the properties dict
+    posthog.capture('distinct id', 'purchase', groups={'company': 'id:5'})
     ```
     """
 
-    return _proxy("capture", event, **kwargs)
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
+    return _proxy(
+        "capture",
+        distinct_id=distinct_id,
+        event=event,
+        properties=properties,
+        context=context,
+        timestamp=timestamp,
+        uuid=uuid,
+        groups=groups,
+        send_feature_flags=send_feature_flags,
+        disable_geoip=disable_geoip,
+    )
 
 
-def set(**kwargs: Unpack[OptionalSetArgs]) -> Optional[str]:
+def identify(
+    distinct_id,  # type: str
+    properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
+    timestamp=None,  # type: Optional[datetime.datetime]
+    uuid=None,  # type: Optional[str]
+    disable_geoip=None,  # type: Optional[bool]
+):
+    # type: (...) -> Tuple[bool, dict]
+    """
+    Identify lets you add metadata on your users so you can more easily identify who they are in PostHog, and even do things like segment users by these properties.
+
+    An `identify` call requires
+    - `distinct id` which uniquely identifies your user
+    - `properties` with a dict with any key: value pairs
+
+    For example:
+    ```python
+    posthog.identify('distinct id', {
+        'email': 'dwayne@gmail.com',
+        'name': 'Dwayne Johnson'
+    })
+    ```
+    """
+
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
+    return _proxy(
+        "identify",
+        distinct_id=distinct_id,
+        properties=properties,
+        context=context,
+        timestamp=timestamp,
+        uuid=uuid,
+        disable_geoip=disable_geoip,
+    )
+
+
+def set(
+    distinct_id,  # type: str
+    properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
+    timestamp=None,  # type: Optional[datetime.datetime]
+    uuid=None,  # type: Optional[str]
+    disable_geoip=None,  # type: Optional[bool]
+):
+    # type: (...) -> Tuple[bool, dict]
     """
     Set properties on a user record.
-    This will overwrite previous people property values. Generally operates similar to `capture`, with
-    distinct_id being an optional argument, defaulting to the current context's distinct ID.
+    This will overwrite previous people property values, just like `identify`.
 
-    If there is no context-level distinct ID, and no override distinct_id is passed, this function
-    will do nothing.
-
-    Context tags are folded into $set properties, so tagging the current context and then calling `set` will
-    cause those tags to be set on the user (unlike capture, which causes them to just be set on the event).
+     A `set` call requires
+     - `distinct id` which uniquely identifies your user
+     - `properties` with a dict with any key: value pairs
 
      For example:
      ```python
-     posthog.set(distinct_id='distinct id', properties={
+     posthog.set('distinct id', {
          'current_browser': 'Chrome',
      })
      ```
     """
 
-    return _proxy("set", **kwargs)
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
+    return _proxy(
+        "set",
+        distinct_id=distinct_id,
+        properties=properties,
+        context=context,
+        timestamp=timestamp,
+        uuid=uuid,
+        disable_geoip=disable_geoip,
+    )
 
 
-def set_once(**kwargs: Unpack[OptionalSetArgs]) -> Optional[str]:
+def set_once(
+    distinct_id,  # type: str
+    properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
+    timestamp=None,  # type: Optional[datetime.datetime]
+    uuid=None,  # type: Optional[str]
+    disable_geoip=None,  # type: Optional[bool]
+):
+    # type: (...) -> Tuple[bool, dict]
     """
     Set properties on a user record, only if they do not yet exist.
-    This will not overwrite previous people property values, unlike `set`.
+    This will not overwrite previous people property values, unlike `identify`.
 
-    Otherwise, operates in an identical manner to `set`.
+     A `set_once` call requires
+     - `distinct id` which uniquely identifies your user
+     - `properties` with a dict with any key: value pairs
+
+     For example:
+     ```python
+     posthog.set_once('distinct id', {
+         'referred_by': 'friend',
+     })
      ```
     """
-    return _proxy("set_once", **kwargs)
+
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
+    return _proxy(
+        "set_once",
+        distinct_id=distinct_id,
+        properties=properties,
+        context=context,
+        timestamp=timestamp,
+        uuid=uuid,
+        disable_geoip=disable_geoip,
+    )
 
 
 def group_identify(
     group_type,  # type: str
     group_key,  # type: str
     properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
     timestamp=None,  # type: Optional[datetime.datetime]
     uuid=None,  # type: Optional[str]
     disable_geoip=None,  # type: Optional[bool]
 ):
-    # type: (...) -> Optional[str]
+    # type: (...) -> Tuple[bool, dict]
     """
     Set properties on a group
 
      A `group_identify` call requires
      - `group_type` type of your group
      - `group_key` unique identifier of the group
+     - `properties` with a dict with any key: value pairs
 
      For example:
      ```python
@@ -167,11 +252,19 @@ def group_identify(
      ```
     """
 
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
     return _proxy(
         "group_identify",
         group_type=group_type,
         group_key=group_key,
         properties=properties,
+        context=context,
         timestamp=timestamp,
         uuid=uuid,
         disable_geoip=disable_geoip,
@@ -181,16 +274,18 @@ def group_identify(
 def alias(
     previous_id,  # type: str
     distinct_id,  # type: str
+    context=None,  # type: Optional[Dict]
     timestamp=None,  # type: Optional[datetime.datetime]
     uuid=None,  # type: Optional[str]
     disable_geoip=None,  # type: Optional[bool]
 ):
-    # type: (...) -> Optional[str]
+    # type: (...) -> Tuple[bool, dict]
     """
-    To marry up whatever a user does before they sign up or log in with what they do after you need to make an alias call.
-    This will allow you to answer questions like "Which marketing channels leads to users churning after a month?" or
-    "What do users do on our website before signing up?". Particularly useful for associating user behaviour before and after
-    they e.g. register, login, or perform some other identifying action.
+    To marry up whatever a user does before they sign up or log in with what they do after you need to make an alias call. This will allow you to answer questions like "Which marketing channels leads to users churning after a month?" or "What do users do on our website before signing up?"
+
+    In a purely back-end implementation, this means whenever an anonymous user does something, you'll want to send a session ID ([Django](https://stackoverflow.com/questions/526179/in-django-how-can-i-find-out-the-request-session-sessionid-and-use-it-as-a-vari), [Flask](https://stackoverflow.com/questions/15156132/flask-login-how-to-get-session-id)) with the capture call. Then, when that users signs up, you want to do an alias call with the session ID and the newly created user ID.
+
+    The same concept applies for when a user logs in.
 
     An `alias` call requires
     - `previous distinct id` the unique ID of the user before
@@ -202,10 +297,18 @@ def alias(
     ```
     """
 
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
     return _proxy(
         "alias",
         previous_id=previous_id,
         distinct_id=distinct_id,
+        context=context,
         timestamp=timestamp,
         uuid=uuid,
         disable_geoip=disable_geoip,
@@ -213,33 +316,58 @@ def alias(
 
 
 def capture_exception(
-    exception: Optional[ExceptionArg] = None,
-    **kwargs: Unpack[OptionalCaptureArgs],
+    exception=None,  # type: Optional[BaseException]
+    distinct_id=None,  # type: Optional[str]
+    properties=None,  # type: Optional[Dict]
+    context=None,  # type: Optional[Dict]
+    timestamp=None,  # type: Optional[datetime.datetime]
+    uuid=None,  # type: Optional[str]
+    groups=None,  # type: Optional[Dict]
+    **kwargs,
 ):
+    # type: (...) -> Tuple[bool, dict]
     """
-    capture_exception allows you to capture exceptions that happen in your code.
+    capture_exception allows you to capture exceptions that happen in your code. This is useful for debugging and understanding what errors your users are encountering.
+    This function never raises an exception, even if it fails to send the event.
 
-    Capture exception is idempotent - if it is called twice with the same exception instance, only a occurrence will be tracked in posthog.
-    This is because, generally, contexts will cause exceptions to be captured automatically. However, to ensure you track an exception,
-    if you catch and do not re-raise it, capturing it manually is recommended, unless you are certain it will have crossed a context
-    boundary (e.g. by existing a `with posthog.new_context():` block already)
-
-    A `capture_exception` call does not require any fields, but we recommend passing an exception of some kind:
+    A `capture_exception` call does not require any fields, but we recommend sending:
+    - `distinct id` which uniquely identifies your user for which this exception happens
     - `exception` to specify the exception to capture. If not provided, the current exception is captured via `sys.exc_info()`
 
-    If the passed exception was raised and caught, the captured stack trace will consist of every frame between where the exception was raised
-    and the point at which it is captured (the "traceback").
+    Optionally you can submit
+    - `properties`, which can be a dict with any information you'd like to add
+    - `groups`, which is a dict of group type -> group key mappings
+    - remaining `kwargs` will be logged if `log_captured_exceptions` is enabled
 
-    If the passed exception was never raised, e.g. if you call `posthog.capture_exception(ValueError("Some Error"))`, the stack trace
-    captured will be the full stack trace at the moment the exception was captured.
+    For example:
+    ```python
+    try:
+        1 / 0
+    except Exception as e:
+        posthog.capture_exception(e, 'my specific distinct id')
+        posthog.capture_exception(distinct_id='my specific distinct id')
 
-    Note that heavy use of contexts will lead to truncated stack traces, as the exception will be captured by the context entered most recently,
-    which may not be the point you catch the exception for the final time in your code. It's recommended to use contexts sparingly, for this reason.
-
-    `capture_exception` takes the same set of optional arguments as `capture`.
+    ```
     """
 
-    return _proxy("capture_exception", exception=exception, **kwargs)
+    if context is not None:
+        warnings.warn(
+            "The 'context' parameter is deprecated and will be removed in a future version.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+
+    return _proxy(
+        "capture_exception",
+        exception=exception,
+        distinct_id=distinct_id,
+        properties=properties,
+        context=context,
+        timestamp=timestamp,
+        uuid=uuid,
+        groups=groups,
+        **kwargs,
+    )
 
 
 def feature_enabled(
@@ -426,6 +554,16 @@ def load_feature_flags():
     return _proxy("load_feature_flags")
 
 
+def page(*args, **kwargs):
+    """Send a page call."""
+    _proxy("page", *args, **kwargs)
+
+
+def screen(*args, **kwargs):
+    """Send a screen call."""
+    _proxy("screen", *args, **kwargs)
+
+
 def flush():
     """Tell the client to flush."""
     _proxy("flush")
@@ -445,8 +583,6 @@ def shutdown():
 def setup():
     global default_client
     if not default_client:
-        if not api_key:
-            raise ValueError("API key is required")
         default_client = Client(
             api_key,
             host=host,
@@ -455,6 +591,7 @@ def setup():
             send=send,
             sync_mode=sync_mode,
             personal_api_key=personal_api_key,
+            project_api_key=project_api_key,
             poll_interval=poll_interval,
             disabled=disabled,
             disable_geoip=disable_geoip,
@@ -465,6 +602,7 @@ def setup():
             # or deprecate this proxy option fully (it's already in the process of deprecation, no new clients should be using this method since like 5-6 months)
             enable_exception_autocapture=enable_exception_autocapture,
             log_captured_exceptions=log_captured_exceptions,
+            exception_autocapture_integrations=exception_autocapture_integrations,
         )
 
     # always set incase user changes it

posthog/ai/langchain/callbacks.py

@@ -14,6 +14,7 @@ from typing import (
     List,
     Optional,
     Sequence,
+    Tuple,
     Union,
     cast,
 )
@@ -84,7 +85,7 @@ class CallbackHandler(BaseCallbackHandler):
     _client: Client
     """PostHog client instance."""
 
-    _distinct_id: Optional[Union[str, int, UUID]]
+    _distinct_id: Optional[Union[str, int, float, UUID]]
     """Distinct ID of the user to associate the trace with."""
 
     _trace_id: Optional[Union[str, int, float, UUID]]
@@ -112,7 +113,7 @@ class CallbackHandler(BaseCallbackHandler):
         self,
         client: Optional[Client] = None,
         *,
-        distinct_id: Optional[Union[str, int, UUID]] = None,
+        distinct_id: Optional[Union[str, int, float, UUID]] = None,
         trace_id: Optional[Union[str, int, float, UUID]] = None,
         properties: Optional[Dict[str, Any]] = None,
         privacy_mode: bool = False,
@@ -568,14 +569,9 @@ class CallbackHandler(BaseCallbackHandler):
             event_properties["$ai_is_error"] = True
         else:
             # Add usage
-            usage = _parse_usage(output)
-            event_properties["$ai_input_tokens"] = usage.input_tokens
-            event_properties["$ai_output_tokens"] = usage.output_tokens
-            event_properties["$ai_cache_creation_input_tokens"] = (
-                usage.cache_write_tokens
-            )
-            event_properties["$ai_cache_read_input_tokens"] = usage.cache_read_tokens
-            event_properties["$ai_reasoning_tokens"] = usage.reasoning_tokens
+            input_tokens, output_tokens = _parse_usage(output)
+            event_properties["$ai_input_tokens"] = input_tokens
+            event_properties["$ai_output_tokens"] = output_tokens
 
             # Generation results
             generation_result = output.generations[-1]
@@ -651,18 +647,9 @@ def _convert_message_to_dict(message: BaseMessage) -> Dict[str, Any]:
     return message_dict
 
 
-@dataclass
-class ModelUsage:
-    input_tokens: Optional[int]
-    output_tokens: Optional[int]
-    cache_write_tokens: Optional[int]
-    cache_read_tokens: Optional[int]
-    reasoning_tokens: Optional[int]
-
-
 def _parse_usage_model(
-    usage: Union[BaseModel, dict],
-) -> ModelUsage:
+    usage: Union[BaseModel, Dict],
+) -> Tuple[Union[int, None], Union[int, None]]:
     if isinstance(usage, BaseModel):
         usage = usage.__dict__
 
@@ -670,23 +657,15 @@ def _parse_usage_model(
         # https://pypi.org/project/langchain-anthropic/ (works also for Bedrock-Anthropic)
         ("input_tokens", "input"),
         ("output_tokens", "output"),
-        ("cache_creation_input_tokens", "cache_write"),
-        ("cache_read_input_tokens", "cache_read"),
         # https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/get-token-count
         ("prompt_token_count", "input"),
         ("candidates_token_count", "output"),
-        ("cached_content_token_count", "cache_read"),
-        ("thoughts_token_count", "reasoning"),
         # Bedrock: https://docs.aws.amazon.com/bedrock/latest/userguide/monitoring-cw.html#runtime-cloudwatch-metrics
         ("inputTokenCount", "input"),
         ("outputTokenCount", "output"),
-        ("cacheCreationInputTokenCount", "cache_write"),
-        ("cacheReadInputTokenCount", "cache_read"),
         # Bedrock Anthropic
         ("prompt_tokens", "input"),
         ("completion_tokens", "output"),
-        ("cache_creation_input_tokens", "cache_write"),
-        ("cache_read_input_tokens", "cache_read"),
         # langchain-ibm https://pypi.org/project/langchain-ibm/
         ("input_token_count", "input"),
         ("generated_token_count", "output"),
@@ -704,45 +683,13 @@ def _parse_usage_model(
 
             parsed_usage[type_key] = final_count
 
-    # Caching (OpenAI & langchain 0.3.9+)
-    if "input_token_details" in usage and isinstance(
-        usage["input_token_details"], dict
-    ):
-        parsed_usage["cache_write"] = usage["input_token_details"].get("cache_creation")
-        parsed_usage["cache_read"] = usage["input_token_details"].get("cache_read")
-
-    # Reasoning (OpenAI & langchain 0.3.9+)
-    if "output_token_details" in usage and isinstance(
-        usage["output_token_details"], dict
-    ):
-        parsed_usage["reasoning"] = usage["output_token_details"].get("reasoning")
-
-    field_mapping = {
-        "input": "input_tokens",
-        "output": "output_tokens",
-        "cache_write": "cache_write_tokens",
-        "cache_read": "cache_read_tokens",
-        "reasoning": "reasoning_tokens",
-    }
-    return ModelUsage(
-        **{
-            dataclass_key: parsed_usage.get(mapped_key) or 0
-            for mapped_key, dataclass_key in field_mapping.items()
-        },
-    )
+    return parsed_usage.get("input"), parsed_usage.get("output")
 
 
-def _parse_usage(response: LLMResult) -> ModelUsage:
+def _parse_usage(response: LLMResult):
     # langchain-anthropic uses the usage field
     llm_usage_keys = ["token_usage", "usage"]
-    llm_usage: ModelUsage = ModelUsage(
-        input_tokens=None,
-        output_tokens=None,
-        cache_write_tokens=None,
-        cache_read_tokens=None,
-        reasoning_tokens=None,
-    )
-
+    llm_usage: Tuple[Union[int, None], Union[int, None]] = (None, None)
     if response.llm_output is not None:
         for key in llm_usage_keys:
             if response.llm_output.get(key):

posthog/ai/openai/openai.py

@@ -230,42 +230,6 @@ class WrappedResponses:
                 groups=posthog_groups,
             )
 
-    def parse(
-        self,
-        posthog_distinct_id: Optional[str] = None,
-        posthog_trace_id: Optional[str] = None,
-        posthog_properties: Optional[Dict[str, Any]] = None,
-        posthog_privacy_mode: bool = False,
-        posthog_groups: Optional[Dict[str, Any]] = None,
-        **kwargs: Any,
-    ):
-        """
-        Parse structured output using OpenAI's 'responses.parse' method, but also track usage in PostHog.
-
-        Args:
-            posthog_distinct_id: Optional ID to associate with the usage event.
-            posthog_trace_id: Optional trace UUID for linking events.
-            posthog_properties: Optional dictionary of extra properties to include in the event.
-            posthog_privacy_mode: Whether to anonymize the input and output.
-            posthog_groups: Optional dictionary of groups to associate with the event.
-            **kwargs: Any additional parameters for the OpenAI Responses Parse API.
-
-        Returns:
-            The response from OpenAI's responses.parse call.
-        """
-        return call_llm_and_track_usage(
-            posthog_distinct_id,
-            self._client._ph_client,
-            "openai",
-            posthog_trace_id,
-            posthog_properties,
-            posthog_privacy_mode,
-            posthog_groups,
-            self._client.base_url,
-            self._original.parse,
-            **kwargs,
-        )
-
 
 class WrappedChat:
     """Wrapper for OpenAI chat that tracks usage in PostHog."""

posthog/ai/openai/openai_async.py

@@ -230,42 +230,6 @@ class WrappedResponses:
                 groups=posthog_groups,
             )
 
-    async def parse(
-        self,
-        posthog_distinct_id: Optional[str] = None,
-        posthog_trace_id: Optional[str] = None,
-        posthog_properties: Optional[Dict[str, Any]] = None,
-        posthog_privacy_mode: bool = False,
-        posthog_groups: Optional[Dict[str, Any]] = None,
-        **kwargs: Any,
-    ):
-        """
-        Parse structured output using OpenAI's 'responses.parse' method, but also track usage in PostHog.
-
-        Args:
-            posthog_distinct_id: Optional ID to associate with the usage event.
-            posthog_trace_id: Optional trace UUID for linking events.
-            posthog_properties: Optional dictionary of extra properties to include in the event.
-            posthog_privacy_mode: Whether to anonymize the input and output.
-            posthog_groups: Optional dictionary of groups to associate with the event.
-            **kwargs: Any additional parameters for the OpenAI Responses Parse API.
-
-        Returns:
-            The response from OpenAI's responses.parse call.
-        """
-        return await call_llm_and_track_usage_async(
-            posthog_distinct_id,
-            self._client._ph_client,
-            "openai",
-            posthog_trace_id,
-            posthog_properties,
-            posthog_privacy_mode,
-            posthog_groups,
-            self._client.base_url,
-            self._original.parse,
-            **kwargs,
-        )
-
 
 class WrappedChat:
     """Async wrapper for OpenAI chat that tracks usage in PostHog."""

posthog/args.py

@@ -1,68 +0,0 @@
-from typing import TypedDict, Optional, Any, Dict, Union, Tuple, Type
-from types import TracebackType
-from typing_extensions import NotRequired  # For Python < 3.11 compatibility
-from datetime import datetime
-import numbers
-from uuid import UUID
-
-ID_TYPES = Union[numbers.Number, str, UUID, int]
-
-
-class OptionalCaptureArgs(TypedDict):
-    """Optional arguments for the capture method.
-
-    Args:
-        distinct_id: Unique identifier for the person associated with this event. If not set, the context
-            distinct_id is used, if available, otherwise a UUID is generated, and the event is marked
-            as personless. Setting context-level distinct_id's is recommended.
-        properties: Dictionary of properties to track with the event
-        timestamp: When the event occurred (defaults to current time)
-        uuid: Unique identifier for this specific event. If not provided, one is generated. The event
-            UUID is returned, so you can correlate it with actions in your app (like showing users an
-            error ID if you capture an exception).
-        groups: Group identifiers to associate with this event (format: {group_type: group_key})
-        send_feature_flags: Whether to include currently active feature flags in the event properties.
-            Defaults to True
-        disable_geoip: Whether to disable GeoIP lookup for this event. Defaults to False.
-    """
-
-    distinct_id: NotRequired[Optional[ID_TYPES]]
-    properties: NotRequired[Optional[Dict[str, Any]]]
-    timestamp: NotRequired[Optional[Union[datetime, str]]]
-    uuid: NotRequired[Optional[str]]
-    groups: NotRequired[Optional[Dict[str, str]]]
-    send_feature_flags: NotRequired[
-        Optional[bool]
-    ]  # Optional so we can tell if the user is intentionally overriding a client setting or not
-    disable_geoip: NotRequired[
-        Optional[bool]
-    ]  # As above, optional so we can tell if the user is intentionally overriding a client setting or not
-
-
-class OptionalSetArgs(TypedDict):
-    """Optional arguments for the set method.
-
-    Args:
-        distinct_id: Unique identifier for the user to set properties on. If not set, the context
-            distinct_id is used, if available, otherwise this function does nothing. Setting
-            context-level distinct_id's is recommended.
-        properties: Dictionary of properties to set on the person
-        timestamp: When the properties were set (defaults to current time)
-        uuid: Unique identifier for this operation. If not provided, one is generated. This
-            UUID is returned, so you can correlate it with actions in your app.
-        disable_geoip: Whether to disable GeoIP lookup for this operation. Defaults to False.
-    """
-
-    distinct_id: NotRequired[Optional[ID_TYPES]]
-    properties: NotRequired[Optional[Dict[str, Any]]]
-    timestamp: NotRequired[Optional[Union[datetime, str]]]
-    uuid: NotRequired[Optional[str]]
-    disable_geoip: NotRequired[Optional[bool]]
-
-
-ExcInfo = Union[
-    Tuple[Type[BaseException], BaseException, Optional[TracebackType]],
-    Tuple[None, None, None],
-]
-
-ExceptionArg = Union[BaseException, ExcInfo]

posthog/client.py

@@ -1,24 +1,25 @@
 import atexit
+import hashlib
 import logging
+import numbers
 import os
+import platform
 import sys
+import warnings
 from datetime import datetime, timedelta
-from typing import Any, Dict, Optional, Union
-from typing_extensions import Unpack
-from uuid import uuid4
+from typing import Any, Optional, Union
+from uuid import UUID, uuid4
 
+import distro  # For Linux OS detection
 from dateutil.tz import tzutc
 from six import string_types
 
-from posthog.args import OptionalCaptureArgs, OptionalSetArgs, ID_TYPES, ExceptionArg
 from posthog.consumer import Consumer
 from posthog.exception_capture import ExceptionCapture
 from posthog.exception_utils import (
     exc_info_from_error,
     exceptions_from_error_tuple,
     handle_in_app,
-    exception_is_already_captured,
-    mark_exception_as_captured,
 )
 from posthog.feature_flags import InconclusiveMatchError, match_feature_flag_properties
 from posthog.poller import Poller
@@ -26,17 +27,13 @@ from posthog.request import (
     DEFAULT_HOST,
     APIError,
     batch_post,
+    decide,
     determine_server_host,
     flags,
     get,
     remote_config,
 )
-from posthog.contexts import (
-    _get_current_context,
-    get_context_distinct_id,
-    get_context_session_id,
-    new_context,
-)
+from posthog.scopes import get_tags
 from posthog.types import (
     FeatureFlag,
     FeatureFlagResult,
@@ -49,13 +46,7 @@ from posthog.types import (
     to_payloads,
     to_values,
 )
-from posthog.utils import (
-    SizeLimitedDict,
-    clean,
-    guess_timezone,
-    remove_trailing_slash,
-    system_context,
-)
+from posthog.utils import SizeLimitedDict, clean, guess_timezone, remove_trailing_slash
 from posthog.version import VERSION
 
 try:
@@ -64,34 +55,171 @@ except ImportError:
     import Queue as queue
 
 
+ID_TYPES = (numbers.Number, string_types, UUID)
 MAX_DICT_SIZE = 50_000
 
-
-def get_identity_state(passed) -> tuple[str, bool]:
-    """Returns the distinct id to use, and whether this is a personless event or not"""
-    stringified = stringify_id(passed)
-    if stringified and len(stringified):
-        return (stringified, False)
-
-    context_id = get_context_distinct_id()
-    if context_id:
-        return (context_id, False)
-
-    return (str(uuid4()), True)
+# TODO: Get rid of these when you're done rolling out `/flags` to all customers
+ROLLOUT_PERCENTAGE = 1
+INCLUDED_HASHES = set(
+    {"bc94e67150c97dbcbf52549d50a7b80814841dbf"}
+)  # this is PostHog's API key
+# Explicitly excluding all the API tokens associated with the top 10 customers; we'll get to them soon, but don't want to rollout to them just yet
+EXCLUDED_HASHES = set(
+    {
+        "03005596796f9ee626e9596b8062972cb6a556a0",
+        "05620a20b287e0d5cb1d4a0dd492797f36b952c5",
+        "0f95b5ca12878693c01c6420e727904f1737caa7",
+        "1212b6287a6e7e5ff6be5cb30ec563f35c2139d6",
+        "171ec1bb2caf762e06b1fde2e36a38c4638691a8",
+        "171faa9fc754b1aa42252a4eedb948b7c805d5cb",
+        "178ddde3f628fb0030321387acf939e4e6946d35",
+        "1790085d7e9aa136e8b73c180dd6a6060e2ef949",
+        "1895a3349c2371559c886f19ef1bf60617a934e0",
+        "1f01267d4f0295f88e8943bc963d816ee4abc84b",
+        "213df54990a34e62e3570b430f7ee36ec0928743",
+        "23d235537d988ab98ad259853eab02b07d828c2b",
+        "27135f7ae8f936222a5fcfcdc75c139b27dd3254",
+        "2817396d80fafc86c0816af8e73880f8b3e54320",
+        "29d3235e63db42056858ef04c6a5488c2a459eaa",
+        "2a76d9b5eb9307e540de9d516aa80f6cb5a0292f",
+        "2a92965a1344ab8a1f7dac2507e858f579a88ac2",
+        "2d5823818261512d616161de2bb8a161d48f1e35",
+        "32942f6a879dbfa8011cc68288c098e4a76e6cc0",
+        "3db6c17ab65827ceadf77d9a8462fabd94170ca6",
+        "4975b24f9ced9b2c06b604ddc9612f663f9452d5",
+        "497c7b017b13cd6cdbfe641c71f0dfb660a4c518",
+        "49c79e1dbce4a7b9394d6c14bf0421e04cecb445",
+        "4d63e1c5cd3a80972eac4e7526f03357ac538043",
+        "4da0f42a6f8f116822411152e5cda3c65ed2561f",
+        "4e494675ecd2b841784d6f29b658b38a0877a62e",
+        "4e852d8422130cec991eca2d6416dbe321d0a689",
+        "5120bfd92c9c6731074a89e4a82f49e947d34369",
+        "512cd72f9aa7ab11dfd012cc2e19394a020bd9a8",
+        "5b175d4064cc62f01118a2c6818c2c02fc8f27e1",
+        "5ba4bba3979e97d2c84df2aba394ca29c6c43187",
+        "639014946463614353ca640b268dc6592f62b652",
+        "643b9be9d50104e2b4ba94bc56688adba69c80fe",
+        "658f92992af9fc6a360143d72d93a36f63bbccb0",
+        "673a59c99739dfcee35202e428dd020b94866d52",
+        "67a9829b4997f5c6f3ab8173ad299f634adcfa53",
+        "6d686043e914ae8275df65e1ad890bd32a3b6fdd",
+        "6e4b5e1d649ad006d78f1f1617a9a0f35fc73078",
+        "6f1fc3a8fa9df54d00cbc1ef9ad5f24640589fd0",
+        "764e5fec2c7899cfee620fae8450fcc62cd72bf0",
+        "80ea6d6ed9a5895633c7bee7aba4323eeacdc90e",
+        "872e420156f583bc97351f3d83c02dae734a85df",
+        "8a24844cbeae31e74b4372964cdea74e99d9c0e2",
+        "975ae7330506d4583b000f96ad87abb41a0141ce",
+        "9e3d71378b340def3080e0a3a785a1b964cf43ef",
+        "9ede7b21365661331d024d92915de6e69749892b",
+        "a1ed1b4216ef4cec542c6b3b676507770be24ddc",
+        "a4f66a70a9647b3b89fc59f7642af8ffab073ba1",
+        "a7adb80be9e90948ab6bb726cc6e8e52694aec74",
+        "bca4b14ac8de49cccc02306c7bb6e5ae2acc0f72",
+        "bde5fe49f61e13629c5498d7428a7f6215e482a6",
+        "c54a7074c323aa7c5cb7b24bf826751b2a58f5d8",
+        "c552d20da0c87fb4ebe2da97c7f95c05eef2bca1",
+        "d7682f2d268f3064d433309af34f2935810989d2",
+        "d794ac43d8be26bf99f369ea79501eb774fe1b16",
+        "e0963e2552af77d46bb24d5b5806b5b456c64c5f",
+        "e6f14b2100cb0598925958b097ace82486037a25",
+        "e79ec399ad45f44a4295a5bb1322e2f14600ae39",
+        "eecf29f73f9c31009e5737a6c5ec3f87ec5b8ea6",
+        "f2c01f3cc770c7788257ee60910e2530f92eefc3",
+        "f7bbc58f4122b1e2812c0f1962c584cb404a1ac3",
+    }
+)
 
 
-def add_context_tags(properties):
-    current_context = _get_current_context()
-    if current_context:
-        context_tags = current_context.collect_tags()
-        # We want explicitly passed properties to override context tags
-        context_tags.update(properties)
-        properties = context_tags
+def get_os_info():
+    """
+    Returns standardized OS name and version information.
+    Similar to how user agent parsing works in JS.
+    """
+    os_name = ""
+    os_version = ""
 
-    if "$session_id" not in properties and get_context_session_id():
-        properties["$session_id"] = get_context_session_id()
+    platform_name = sys.platform
 
-    return properties
+    if platform_name.startswith("win"):
+        os_name = "Windows"
+        if hasattr(platform, "win32_ver"):
+            win_version = platform.win32_ver()[0]
+            if win_version:
+                os_version = win_version
+
+    elif platform_name == "darwin":
+        os_name = "Mac OS X"
+        if hasattr(platform, "mac_ver"):
+            mac_version = platform.mac_ver()[0]
+            if mac_version:
+                os_version = mac_version
+
+    elif platform_name.startswith("linux"):
+        os_name = "Linux"
+        linux_info = distro.info()
+        if linux_info["version"]:
+            os_version = linux_info["version"]
+
+    elif platform_name.startswith("freebsd"):
+        os_name = "FreeBSD"
+        if hasattr(platform, "release"):
+            os_version = platform.release()
+
+    else:
+        os_name = platform_name
+        if hasattr(platform, "release"):
+            os_version = platform.release()
+
+    return os_name, os_version
+
+
+def system_context() -> dict[str, Any]:
+    os_name, os_version = get_os_info()
+
+    return {
+        "$python_runtime": platform.python_implementation(),
+        "$python_version": "%s.%s.%s" % (sys.version_info[:3]),
+        "$os": os_name,
+        "$os_version": os_version,
+    }
+
+
+def is_token_in_rollout(
+    token: str,
+    percentage: float = 0,
+    included_hashes: Optional[set[str]] = None,
+    excluded_hashes: Optional[set[str]] = None,
+) -> bool:
+    """
+    Determines if a token should be included in a rollout based on:
+    1. If its hash matches any included_hashes provided
+    2. If its hash falls within the percentage rollout
+
+    Args:
+        token: String to hash (usually API key)
+        percentage: Float between 0 and 1 representing rollout percentage
+        included_hashes: Optional set of specific SHA1 hashes to match against
+        excluded_hashes: Optional set of specific SHA1 hashes to exclude from rollout
+    Returns:
+        bool: True if token should be included in rollout
+    """
+    # First generate SHA1 hash of token
+    token_hash = hashlib.sha1(token.encode("utf-8")).hexdigest()
+
+    # Check if hash matches any included hashes
+    if included_hashes and token_hash in included_hashes:
+        return True
+
+    # Check if hash matches any excluded hashes
+    if excluded_hashes and token_hash in excluded_hashes:
+        return False
+
+    # Convert first 8 chars of hash to int and divide by max value to get number between 0-1
+    hash_int = int(token_hash[:8], 16)
+    hash_float = hash_int / 0xFFFFFFFF
+
+    return hash_float < percentage
 
 
 class Client(object):
@@ -101,7 +229,7 @@ class Client(object):
 
     def __init__(
         self,
-        project_api_key: str,
+        api_key=None,
         host=None,
         debug=False,
         max_queue_size=10000,
@@ -116,6 +244,7 @@ class Client(object):
         thread=1,
         poll_interval=30,
         personal_api_key=None,
+        project_api_key=None,
         disabled=False,
         disable_geoip=True,
         historical_migration=False,
@@ -123,14 +252,16 @@ class Client(object):
         super_properties=None,
         enable_exception_autocapture=False,
         log_captured_exceptions=False,
+        exception_autocapture_integrations=None,
         project_root=None,
         privacy_mode=False,
-        before_send=None,
     ):
         self.queue = queue.Queue(max_queue_size)
 
         # api_key: This should be the Team API Key (token), public
-        self.api_key = project_api_key
+        self.api_key = project_api_key or api_key
+
+        require("api_key", self.api_key, string_types)
 
         self.on_error = on_error
         self.debug = debug
@@ -157,6 +288,7 @@ class Client(object):
         self.super_properties = super_properties
         self.enable_exception_autocapture = enable_exception_autocapture
         self.log_captured_exceptions = log_captured_exceptions
+        self.exception_autocapture_integrations = exception_autocapture_integrations
         self.exception_capture = None
         self.privacy_mode = privacy_mode
 
@@ -178,17 +310,10 @@ class Client(object):
         else:
             self.log.setLevel(logging.WARNING)
 
-        if before_send is not None:
-            if callable(before_send):
-                self.before_send = before_send
-            else:
-                self.log.warning("before_send is not callable, it will be ignored")
-                self.before_send = None
-        else:
-            self.before_send = None
-
         if self.enable_exception_autocapture:
-            self.exception_capture = ExceptionCapture(self)
+            self.exception_capture = ExceptionCapture(
+                self, integrations=self.exception_autocapture_integrations
+            )
 
         if sync_mode:
             self.consumers = None
@@ -221,11 +346,6 @@ class Client(object):
                 if send:
                     consumer.start()
 
-    def new_context(self, fresh=False, capture_exceptions=True):
-        return new_context(
-            fresh=fresh, capture_exceptions=capture_exceptions, client=self
-        )
-
     @property
     def feature_flags(self):
         """
@@ -248,6 +368,36 @@ class Client(object):
             "feature_flags_by_key should be initialized when feature_flags is set"
         )
 
+    def identify(
+        self,
+        distinct_id=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        disable_geoip=None,
+    ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+
+        properties = properties or {}
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("properties", properties, dict)
+
+        msg = {
+            "timestamp": timestamp,
+            "distinct_id": distinct_id,
+            "$set": properties,
+            "event": "$identify",
+            "uuid": uuid,
+        }
+
+        return self._enqueue(msg, disable_geoip)
+
     def get_feature_variants(
         self,
         distinct_id,
@@ -298,8 +448,8 @@ class Client(object):
 
     def get_flags_decision(
         self,
-        distinct_id: Optional[ID_TYPES] = None,
-        groups: Optional[dict] = {},
+        distinct_id,
+        groups=None,
         person_properties=None,
         group_properties=None,
         disable_geoip=None,
@@ -307,14 +457,14 @@ class Client(object):
         """
         Get feature flags decision, using either flags() or decide() API based on rollout.
         """
-
-        if distinct_id is None:
-            distinct_id = get_context_distinct_id()
+        require("distinct_id", distinct_id, ID_TYPES)
 
         if disable_geoip is None:
             disable_geoip = self.disable_geoip
 
-        if not groups:
+        if groups:
+            require("groups", groups, dict)
+        else:
             groups = {}
 
         request_data = {
@@ -325,34 +475,59 @@ class Client(object):
             "geoip_disable": disable_geoip,
         }
 
-        resp_data = flags(
+        use_flags = is_token_in_rollout(
             self.api_key,
-            self.host,
-            timeout=self.feature_flags_request_timeout_seconds,
-            **request_data,
+            ROLLOUT_PERCENTAGE,
+            included_hashes=INCLUDED_HASHES,
+            excluded_hashes=EXCLUDED_HASHES,
         )
 
+        if use_flags:
+            resp_data = flags(
+                self.api_key,
+                self.host,
+                timeout=self.feature_flags_request_timeout_seconds,
+                **request_data,
+            )
+        else:
+            resp_data = decide(
+                self.api_key,
+                self.host,
+                timeout=self.feature_flags_request_timeout_seconds,
+                **request_data,
+            )
+
         return normalize_flags_response(resp_data)
 
     def capture(
-        self, event: str, **kwargs: Unpack[OptionalCaptureArgs]
-    ) -> Optional[str]:
-        distinct_id = kwargs.get("distinct_id", None)
-        properties = kwargs.get("properties", None)
-        timestamp = kwargs.get("timestamp", None)
-        uuid = kwargs.get("uuid", None)
-        groups = kwargs.get("groups", None)
-        send_feature_flags = kwargs.get("send_feature_flags", False)
-        disable_geoip = kwargs.get("disable_geoip", None)
+        self,
+        distinct_id=None,
+        event=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        groups=None,
+        send_feature_flags=False,
+        disable_geoip=None,
+    ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
 
         properties = {**(properties or {}), **system_context()}
 
-        properties = add_context_tags(properties)
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("properties", properties, dict)
+        require("event", event, string_types)
 
-        (distinct_id, personless) = get_identity_state(distinct_id)
-
-        if personless and "$process_person_profile" not in properties:
-            properties["$process_person_profile"] = False
+        # Grab current context tags, if any exist
+        context_tags = get_tags()
+        if context_tags:
+            properties.update(context_tags)
 
         msg = {
             "properties": properties,
@@ -363,6 +538,7 @@ class Client(object):
         }
 
         if groups:
+            require("groups", groups, dict)
             msg["properties"]["$groups"] = groups
 
         extra_properties: dict[str, Any] = {}
@@ -402,21 +578,25 @@ class Client(object):
 
         return self._enqueue(msg, disable_geoip)
 
-    def set(self, **kwargs: Unpack[OptionalSetArgs]) -> Optional[str]:
-        distinct_id = kwargs.get("distinct_id", None)
-        properties = kwargs.get("properties", None)
-        timestamp = kwargs.get("timestamp", None)
-        uuid = kwargs.get("uuid", None)
-        disable_geoip = kwargs.get("disable_geoip", None)
+    def set(
+        self,
+        distinct_id=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        disable_geoip=None,
+    ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
 
         properties = properties or {}
-
-        properties = add_context_tags(properties)
-
-        (distinct_id, personless) = get_identity_state(distinct_id)
-
-        if personless or not properties:
-            return None  # Personless set() does nothing
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("properties", properties, dict)
 
         msg = {
             "timestamp": timestamp,
@@ -428,20 +608,25 @@ class Client(object):
 
         return self._enqueue(msg, disable_geoip)
 
-    def set_once(self, **kwargs: Unpack[OptionalSetArgs]) -> Optional[str]:
-        distinct_id = kwargs.get("distinct_id", None)
-        properties = kwargs.get("properties", None)
-        timestamp = kwargs.get("timestamp", None)
-        uuid = kwargs.get("uuid", None)
-        disable_geoip = kwargs.get("disable_geoip", None)
+    def set_once(
+        self,
+        distinct_id=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        disable_geoip=None,
+    ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+
         properties = properties or {}
-
-        properties = add_context_tags(properties)
-
-        (distinct_id, personless) = get_identity_state(distinct_id)
-
-        if personless or not properties:
-            return None  # Personless set_once() does nothing
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("properties", properties, dict)
 
         msg = {
             "timestamp": timestamp,
@@ -455,18 +640,30 @@ class Client(object):
 
     def group_identify(
         self,
-        group_type: str,
-        group_key: str,
+        group_type=None,
+        group_key=None,
         properties=None,
+        context=None,
         timestamp=None,
         uuid=None,
         disable_geoip=None,
         distinct_id=None,
     ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
         properties = properties or {}
+        require("group_type", group_type, ID_TYPES)
+        require("group_key", group_key, ID_TYPES)
+        require("properties", properties, dict)
 
-        # group_identify is purposefully always personful
-        distinct_id = get_identity_state(distinct_id)[0]
+        if distinct_id:
+            require("distinct_id", distinct_id, ID_TYPES)
+        else:
+            distinct_id = "${}_{}".format(group_type, group_key)
 
         msg = {
             "event": "$groupidentify",
@@ -480,24 +677,26 @@ class Client(object):
             "uuid": uuid,
         }
 
-        # NOTE - group_identify doesn't generally use context properties - should it?
-        if get_context_session_id():
-            msg["properties"]["$session_id"] = get_context_session_id()
-
         return self._enqueue(msg, disable_geoip)
 
     def alias(
         self,
-        previous_id: str,
-        distinct_id: Optional[str],
+        previous_id=None,
+        distinct_id=None,
+        context=None,
         timestamp=None,
         uuid=None,
         disable_geoip=None,
     ):
-        (distinct_id, personless) = get_identity_state(distinct_id)
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
 
-        if personless:
-            return None  # Personless alias() does nothing - should this throw?
+        require("previous_id", previous_id, ID_TYPES)
+        require("distinct_id", distinct_id, ID_TYPES)
 
         msg = {
             "properties": {
@@ -507,32 +706,82 @@ class Client(object):
             "timestamp": timestamp,
             "event": "$create_alias",
             "distinct_id": previous_id,
-            "uuid": uuid,
         }
 
-        if get_context_session_id():
-            msg["properties"]["$session_id"] = get_context_session_id()
+        return self._enqueue(msg, disable_geoip)
+
+    def page(
+        self,
+        distinct_id=None,
+        url=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        disable_geoip=None,
+    ):
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+
+        properties = properties or {}
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("properties", properties, dict)
+
+        require("url", url, string_types)
+        properties["$current_url"] = url
+
+        msg = {
+            "event": "$pageview",
+            "properties": properties,
+            "timestamp": timestamp,
+            "distinct_id": distinct_id,
+            "uuid": uuid,
+        }
 
         return self._enqueue(msg, disable_geoip)
 
     def capture_exception(
         self,
-        exception: Optional[ExceptionArg],
-        **kwargs: Unpack[OptionalCaptureArgs],
+        exception=None,
+        distinct_id=None,
+        properties=None,
+        context=None,
+        timestamp=None,
+        uuid=None,
+        groups=None,
+        **kwargs,
     ):
-        distinct_id = kwargs.get("distinct_id", None)
-        properties = kwargs.get("properties", None)
-        send_feature_flags = kwargs.get("send_feature_flags", True)
-        disable_geoip = kwargs.get("disable_geoip", None)
+        if context is not None:
+            warnings.warn(
+                "The 'context' parameter is deprecated and will be removed in a future version.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+
         # this function shouldn't ever throw an error, so it logs exceptions instead of raising them.
         # this is important to ensure we don't unexpectedly re-raise exceptions in the user's code.
         try:
             properties = properties or {}
 
             # Check if this exception has already been captured
-            if exception is not None and exception_is_already_captured(exception):
+            if exception is not None and hasattr(
+                exception, "__posthog_exception_captured"
+            ):
                 self.log.debug("Exception already captured, skipping")
-                return None
+                return
+
+            # if there's no distinct_id, we'll generate one and set personless mode
+            # via $process_person_profile = false
+            if distinct_id is None:
+                properties["$process_person_profile"] = False
+                distinct_id = uuid4()
+
+            require("distinct_id", distinct_id, ID_TYPES)
+            require("properties", properties, dict)
 
             if exception is not None:
                 exc_info = exc_info_from_error(exception)
@@ -541,7 +790,7 @@ class Client(object):
 
             if exc_info is None or exc_info == (None, None, None):
                 self.log.warning("No exception information available")
-                return None
+                return
 
             # Format stack trace for cymbal
             all_exceptions_with_trace = exceptions_from_error_tuple(exc_info)
@@ -570,54 +819,40 @@ class Client(object):
             if self.log_captured_exceptions:
                 self.log.exception(exception, extra=kwargs)
 
-            timestamp = kwargs.get("timestamp", None)
-            uuid = kwargs.get("uuid", None)
-            groups = kwargs.get("groups", None)
             res = self.capture(
-                "$exception",
-                distinct_id=distinct_id,
-                properties=properties,
-                timestamp=timestamp,
-                uuid=uuid,
-                groups=groups,
-                send_feature_flags=send_feature_flags,
-                disable_geoip=disable_geoip,
+                distinct_id, "$exception", properties, context, timestamp, uuid, groups
             )
 
             # Mark the exception as captured to prevent duplicate captures
-            if exception is not None and res is not None:
-                mark_exception_as_captured(exception, res)
+            if exception is not None:
+                setattr(exception, "__posthog_exception_captured", True)
 
             return res
         except Exception as e:
             self.log.exception(f"Failed to capture exception: {e}")
 
     def _enqueue(self, msg, disable_geoip):
-        # type: (...) -> Optional[str]
         """Push a new `msg` onto the queue, return `(success, msg)`"""
 
         if self.disabled:
-            return None
+            return False, "disabled"
 
         timestamp = msg["timestamp"]
         if timestamp is None:
             timestamp = datetime.now(tz=tzutc())
 
+        require("timestamp", timestamp, datetime)
+
         # add common
         timestamp = guess_timezone(timestamp)
         msg["timestamp"] = timestamp.isoformat()
 
+        # only send if "uuid" is truthy
         if "uuid" in msg:
             uuid = msg.pop("uuid")
             if uuid:
                 msg["uuid"] = stringify_id(uuid)
 
-        if "uuid" not in msg:
-            # Always send a uuid, so we can always return one
-            msg["uuid"] = stringify_id(uuid4())
-
-        sent_uuid = msg["uuid"]
-
         if not msg.get("properties"):
             msg["properties"] = {}
         msg["properties"]["$lib"] = "posthog-python"
@@ -635,23 +870,11 @@ class Client(object):
         msg["distinct_id"] = stringify_id(msg.get("distinct_id", None))
 
         msg = clean(msg)
-
-        if self.before_send:
-            try:
-                modified_msg = self.before_send(msg)
-                if modified_msg is None:
-                    self.log.debug("Event dropped by before_send callback")
-                    return None
-                msg = modified_msg
-            except Exception as e:
-                self.log.exception(f"Error in before_send callback: {e}")
-                # Continue with the original message if callback fails
-
         self.log.debug("queueing: %s", msg)
 
         # if send is False, return msg as if it was successfully queued
         if not self.send:
-            return sent_uuid
+            return True, msg
 
         if self.sync_mode:
             self.log.debug("enqueued with blocking %s.", msg["event"])
@@ -664,15 +887,15 @@ class Client(object):
                 historical_migration=self.historical_migration,
             )
 
-            return sent_uuid
+            return True, msg
 
         try:
             self.queue.put(msg, block=False)
             self.log.debug("enqueued %s.", msg["event"])
-            return sent_uuid
+            return True, msg
         except queue.Full:
             self.log.warning("analytics-python queue is full")
-            return None
+            return False, msg
 
     def flush(self):
         """Forces a flush from the internal queue to the server"""
@@ -850,17 +1073,21 @@ class Client(object):
 
     def _get_feature_flag_result(
         self,
-        key: str,
-        distinct_id: ID_TYPES,
+        key,
+        distinct_id,
         *,
         override_match_value: Optional[FlagValue] = None,
-        groups: Dict[str, str] = {},
+        groups={},
         person_properties={},
         group_properties={},
         only_evaluate_locally=False,
         send_feature_flag_events=True,
         disable_geoip=None,
     ) -> Optional[FeatureFlagResult]:
+        require("key", key, string_types)
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("groups", groups, dict)
+
         if self.disabled:
             return None
 
@@ -986,7 +1213,7 @@ class Client(object):
     def _locally_evaluate_flag(
         self,
         key: str,
-        distinct_id: ID_TYPES,
+        distinct_id: str,
         groups: dict[str, str],
         person_properties: dict[str, str],
         group_properties: dict[str, str],
@@ -1050,7 +1277,7 @@ class Client(object):
     def _get_feature_flag_details_from_decide(
         self,
         key: str,
-        distinct_id: ID_TYPES,
+        distinct_id: str,
         groups: dict[str, str],
         person_properties: dict[str, str],
         group_properties: dict[str, str],
@@ -1069,12 +1296,12 @@ class Client(object):
 
     def _capture_feature_flag_called(
         self,
-        distinct_id: ID_TYPES,
+        distinct_id: str,
         key: str,
         response: Optional[FlagValue],
         payload: Optional[str],
         flag_was_locally_evaluated: bool,
-        groups: Dict[str, str],
+        groups: dict[str, str],
         disable_geoip: Optional[bool],
         request_id: Optional[str],
         flag_details: Optional[FeatureFlag],
@@ -1112,9 +1339,9 @@ class Client(object):
                         properties["$feature_flag_id"] = flag_details.metadata.id
 
             self.capture(
+                distinct_id,
                 "$feature_flag_called",
-                distinct_id=distinct_id,
-                properties=properties,
+                properties,
                 groups=groups,
                 disable_geoip=disable_geoip,
             )
@@ -1232,13 +1459,16 @@ class Client(object):
 
     def _get_all_flags_and_payloads_locally(
         self,
-        distinct_id: ID_TYPES,
+        distinct_id,
         *,
-        groups: Dict[str, Union[str, int]],
+        groups={},
         person_properties={},
         group_properties={},
         warn_on_unknown_groups=False,
     ) -> tuple[FlagsAndPayloads, bool]:
+        require("distinct_id", distinct_id, ID_TYPES)
+        require("groups", groups, dict)
+
         if self.feature_flags is None and self.personal_api_key:
             self.load_feature_flags()
 
@@ -1300,6 +1530,13 @@ class Client(object):
         return all_person_properties, all_group_properties
 
 
+def require(name, field, data_type):
+    """Require that the named `field` has the right `data_type`"""
+    if not isinstance(field, data_type):
+        msg = "{0} must have {1}, got: {2}".format(name, data_type, field)
+        raise AssertionError(msg)
+
+
 def stringify_id(val):
     if val is None:
         return None

posthog/contexts.py

@@ -1,254 +0,0 @@
-import contextvars
-from contextlib import contextmanager
-from typing import Optional, Any, Callable, Dict, TypeVar, cast, TYPE_CHECKING
-
-if TYPE_CHECKING:
-    # To avoid circular imports
-    from posthog.client import Client
-
-
-class ContextScope:
-    def __init__(
-        self,
-        parent=None,
-        fresh: bool = False,
-        capture_exceptions: bool = True,
-        client: Optional["Client"] = None,
-    ):
-        self.client: Optional[Client] = client
-        self.parent = parent
-        self.fresh = fresh
-        self.capture_exceptions = capture_exceptions
-        self.session_id: Optional[str] = None
-        self.distinct_id: Optional[str] = None
-        self.tags: Dict[str, Any] = {}
-
-    def set_session_id(self, session_id: str):
-        self.session_id = session_id
-
-    def set_distinct_id(self, distinct_id: str):
-        self.distinct_id = distinct_id
-
-    def add_tag(self, key: str, value: Any):
-        self.tags[key] = value
-
-    def get_parent(self):
-        return self.parent
-
-    def get_session_id(self) -> Optional[str]:
-        if self.session_id is not None:
-            return self.session_id
-        if self.parent is not None and not self.fresh:
-            return self.parent.get_session_id()
-        return None
-
-    def get_distinct_id(self) -> Optional[str]:
-        if self.distinct_id is not None:
-            return self.distinct_id
-        if self.parent is not None and not self.fresh:
-            return self.parent.get_distinct_id()
-        return None
-
-    def collect_tags(self) -> Dict[str, Any]:
-        tags = self.tags.copy()
-        if self.parent and not self.fresh:
-            # We want child tags to take precedence over parent tags,
-            # so we can't use a simple update here, instead collecting
-            # the parent tags and then updating with the child tags.
-            new_tags = self.parent.collect_tags()
-            tags.update(new_tags)
-        return tags
-
-
-_context_stack: contextvars.ContextVar[Optional[ContextScope]] = contextvars.ContextVar(
-    "posthog_context_stack", default=None
-)
-
-
-def _get_current_context() -> Optional[ContextScope]:
-    return _context_stack.get()
-
-
-@contextmanager
-def new_context(
-    fresh=False, capture_exceptions=True, client: Optional["Client"] = None
-):
-    """
-    Create a new context scope that will be active for the duration of the with block.
-    Any tags set within this scope will be isolated to this context. Any exceptions raised
-    or events captured within the context will be tagged with the context tags.
-
-    Args:
-        fresh: Whether to start with a fresh context (default: False).
-               If False, inherits tags, identity and session id's from parent context.
-               If True, starts with no state
-        capture_exceptions: Whether to capture exceptions raised within the context (default: True).
-               If True, captures exceptions and tags them with the context tags before propagating them.
-               If False, exceptions will propagate without being tagged or captured.
-        client: Optional client instance to use for capturing exceptions (default: None).
-                If provided, the client will be used to capture exceptions within the context.
-                If not provided, the default (global) client will be used. Note that the passed
-                client is only used to capture exceptions within the context - other events captured
-                within the context via `Client.capture` or `posthog.capture` will still carry the context
-                state (tags, identity, session id), but will be captured by the client directly used (or
-                the global one, in the case of `posthog.capture`)
-
-    Examples:
-        # Inherit parent context tags
-        with posthog.new_context():
-            posthog.tag("request_id", "123")
-            # Both this event and the exception will be tagged with the context tags
-            posthog.capture("event_name", {"property": "value"})
-            raise ValueError("Something went wrong")
-
-        # Start with fresh context (no inherited tags)
-        with posthog.new_context(fresh=True):
-            posthog.tag("request_id", "123")
-            # Both this event and the exception will be tagged with the context tags
-            posthog.capture("event_name", {"property": "value"})
-            raise ValueError("Something went wrong")
-
-    """
-    from posthog import capture_exception
-
-    current_context = _get_current_context()
-    new_context = ContextScope(current_context, fresh, capture_exceptions, client)
-    _context_stack.set(new_context)
-
-    try:
-        yield
-    except Exception as e:
-        if new_context.capture_exceptions:
-            if new_context.client:
-                new_context.client.capture_exception(e)
-            else:
-                capture_exception(e)
-        raise
-    finally:
-        _context_stack.set(new_context.get_parent())
-
-
-def tag(key: str, value: Any) -> None:
-    """
-    Add a tag to the current context. All tags are added as properties to any event, including exceptions, captured
-    within the context.
-
-    Args:
-        key: The tag key
-        value: The tag value
-
-    Example:
-        posthog.tag("user_id", "123")
-    """
-    current_context = _get_current_context()
-    if current_context:
-        current_context.add_tag(key, value)
-
-
-def get_tags() -> Dict[str, Any]:
-    """
-    Get all tags from the current context. Note, modifying
-    the returned dictionary will not affect the current context.
-
-    Returns:
-        Dict of all tags in the current context
-    """
-    current_context = _get_current_context()
-    if current_context:
-        return current_context.collect_tags()
-    return {}
-
-
-def identify_context(distinct_id: str) -> None:
-    """
-    Identify the current context with a distinct ID, associating all events captured in this or
-    child contexts with the given distinct ID (unless identify_context is called again). This is overridden by
-    distinct id's passed directly to posthog.capture and related methods (identify, set etc). Entering a
-    fresh context will clear the context-level distinct ID. The distinct-id passed should be uniquely associated
-    with one of your users. Events captured outside of a context, or in a context with no associated distinct
-    ID, will be assigned a random UUID, and captured as "personless".
-
-    Args:
-        distinct_id: The distinct ID to associate with the current context and its children.
-    """
-    current_context = _get_current_context()
-    if current_context:
-        current_context.set_distinct_id(distinct_id)
-
-
-def set_context_session(session_id: str) -> None:
-    """
-    Set the session ID for the current context, associating all events captured in this or
-    child contexts with the given session ID (unless set_context_session is called again).
-    Entering a fresh context will clear the context-level session ID.
-
-    Args:
-        session_id: The session ID to associate with the current context and its children. See https://posthog.com/docs/data/sessions
-    """
-    current_context = _get_current_context()
-    if current_context:
-        current_context.set_session_id(session_id)
-
-
-def get_context_session_id() -> Optional[str]:
-    """
-    Get the session ID for the current context.
-
-    Returns:
-        The session ID if set, None otherwise
-    """
-    current_context = _get_current_context()
-    if current_context:
-        return current_context.get_session_id()
-    return None
-
-
-def get_context_distinct_id() -> Optional[str]:
-    """
-    Get the distinct ID for the current context.
-
-    Returns:
-        The distinct ID if set, None otherwise
-    """
-    current_context = _get_current_context()
-    if current_context:
-        return current_context.get_distinct_id()
-    return None
-
-
-F = TypeVar("F", bound=Callable[..., Any])
-
-
-def scoped(fresh=False, capture_exceptions=True):
-    """
-    Decorator that creates a new context for the function. Simply wraps
-    the function in a with posthog.new_context(): block.
-
-    Args:
-        fresh: Whether to start with a fresh context (default: False)
-        capture_exceptions: Whether to capture and track exceptions with posthog error tracking (default: True)
-
-    Example:
-        @posthog.scoped()
-        def process_payment(payment_id):
-            posthog.tag("payment_id", payment_id)
-            posthog.tag("payment_method", "credit_card")
-
-            # This event will be captured with tags
-            posthog.capture("payment_started")
-            # If this raises an exception, it will be captured with tags
-            # and then re-raised
-            some_risky_function()
-    """
-
-    def decorator(func: F) -> F:
-        from functools import wraps
-
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            with new_context(fresh=fresh, capture_exceptions=capture_exceptions):
-                return func(*args, **kwargs)
-
-        return cast(F, wrapper)
-
-    return decorator

posthog/exception_capture.py

@@ -6,25 +6,47 @@
 import logging
 import sys
 import threading
-from typing import TYPE_CHECKING
+from enum import Enum
+from typing import TYPE_CHECKING, List, Optional
 
 if TYPE_CHECKING:
     from posthog.client import Client
 
 
+class Integrations(str, Enum):
+    Django = "django"
+
+
 class ExceptionCapture:
     # TODO: Add client side rate limiting to prevent spamming the server with exceptions
 
     log = logging.getLogger("posthog")
 
-    def __init__(self, client: "Client"):
+    def __init__(
+        self, client: "Client", integrations: Optional[List[Integrations]] = None
+    ):
         self.client = client
         self.original_excepthook = sys.excepthook
         sys.excepthook = self.exception_handler
         threading.excepthook = self.thread_exception_handler
+        self.enabled_integrations = []
+
+        for integration in integrations or []:
+            # TODO: Maybe find a better way of enabling integrations
+            # This is very annoying currently if we had to add any configuration per integration
+            if integration == Integrations.Django:
+                try:
+                    from posthog.exception_integrations.django import DjangoIntegration
+
+                    enabled_integration = DjangoIntegration(self.exception_receiver)
+                    self.enabled_integrations.append(enabled_integration)
+                except Exception as e:
+                    self.log.exception(f"Failed to enable Django integration: {e}")
 
     def close(self):
         sys.excepthook = self.original_excepthook
+        for integration in self.enabled_integrations:
+            integration.uninstall()
 
     def exception_handler(self, exc_type, exc_value, exc_traceback):
         # don't affect default behaviour.
@@ -44,6 +66,6 @@ class ExceptionCapture:
     def capture_exception(self, exception, metadata=None):
         try:
             distinct_id = metadata.get("distinct_id") if metadata else None
-            self.client.capture_exception(exception, distinct_id=distinct_id)
+            self.client.capture_exception(exception, distinct_id)
         except Exception as e:
             self.log.exception(f"Failed to capture exception: {e}")

posthog/exception_integrations/__init__.py

@@ -0,0 +1,5 @@
+class IntegrationEnablingError(Exception):
+    """
+    The integration could not be enabled due to a user error like
+    `django` not being installed for the `DjangoIntegration`.
+    """

posthog/exception_integrations/django.py

@@ -0,0 +1,91 @@
+# Portions of this file are derived from getsentry/sentry-javascript by Software, Inc. dba Sentry
+# Licensed under the MIT License
+
+# 💖open source (under MIT License)
+
+import re
+import sys
+from typing import TYPE_CHECKING
+
+from posthog.exception_integrations import IntegrationEnablingError
+
+try:
+    from django import VERSION as DJANGO_VERSION
+    from django.core import signals
+
+except ImportError:
+    raise IntegrationEnablingError("Django not installed")
+
+
+if TYPE_CHECKING:
+    from typing import Any, Dict  # noqa: F401
+
+    from django.core.handlers.wsgi import WSGIRequest  # noqa: F401
+
+
+class DjangoIntegration:
+    # TODO: Abstract integrations one we have more and can see patterns
+    """
+    Autocapture errors from a Django application.
+    """
+
+    identifier = "django"
+
+    def __init__(self, capture_exception_fn=None):
+        if DJANGO_VERSION < (4, 2):
+            raise IntegrationEnablingError("Django 4.2 or newer is required.")
+
+        # TODO: Right now this seems too complicated / overkill for us, but seems like we can automatically plug in middlewares
+        # which is great for users (they don't need to do this) and everything should just work.
+        # We should consider this in the future, but for now we can just use the middleware and signals handlers.
+        # See: https://github.com/getsentry/sentry-python/blob/269d96d6e9821122fbff280e6a26956e5ed03c0b/sentry_sdk/integrations/django/__init__.py
+
+        self.capture_exception_fn = capture_exception_fn
+
+        def _got_request_exception(request=None, **kwargs):
+            # type: (WSGIRequest, **Any) -> None
+
+            extra_props = {}
+            if request is not None:
+                # get headers metadata
+                extra_props = DjangoRequestExtractor(request).extract_person_data()
+
+            self.capture_exception_fn(sys.exc_info(), extra_props)
+
+        signals.got_request_exception.connect(_got_request_exception)
+
+    def uninstall(self):
+        pass
+
+
+class DjangoRequestExtractor:
+    def __init__(self, request):
+        # type: (Any) -> None
+        self.request = request
+
+    def extract_person_data(self):
+        headers = self.headers()
+
+        # Extract traceparent and tracestate headers
+        traceparent = headers.get("traceparent")
+        tracestate = headers.get("tracestate")
+
+        # Extract the distinct_id from tracestate
+        distinct_id = None
+        if tracestate:
+            # TODO: Align on the format of the distinct_id in tracestate
+            # We can't have comma or equals in header values here, so maybe we should base64 encode it?
+            match = re.search(r"posthog-distinct-id=([^,]+)", tracestate)
+            if match:
+                distinct_id = match.group(1)
+
+        return {
+            "distinct_id": distinct_id,
+            "ip": headers.get("X-Forwarded-For"),
+            "user_agent": headers.get("User-Agent"),
+            "traceparent": traceparent,
+        }
+
+    def headers(self):
+        # type: () -> Dict[str, str]
+        return dict(self.request.headers)

posthog/exception_utils.py

@@ -9,26 +9,8 @@ import linecache
 import os
 import re
 import sys
-import types
 from datetime import datetime
-from types import FrameType, TracebackType  # noqa: F401
-from typing import (  # noqa: F401
-    Any,
-    Dict,
-    Iterator,
-    List,
-    Literal,
-    Optional,
-    Set,
-    Tuple,
-    TypedDict,
-    TypeVar,
-    Union,
-    cast,
-    TYPE_CHECKING,
-)
-
-from posthog.args import ExcInfo, ExceptionArg  # noqa: F401
+from typing import TYPE_CHECKING
 
 try:
     # Python 3.11
@@ -40,61 +22,85 @@ except ImportError:
 
 DEFAULT_MAX_VALUE_LENGTH = 1024
 
-LogLevelStr = Literal["fatal", "critical", "error", "warning", "info", "debug"]
 
-Event = TypedDict(
-    "Event",
-    {
-        "breadcrumbs": Dict[
-            Literal["values"], List[Dict[str, Any]]
-        ],  # TODO: We can expand on this type
-        "check_in_id": str,
-        "contexts": Dict[str, Dict[str, object]],
-        "dist": str,
-        "duration": Optional[float],
-        "environment": str,
-        "errors": List[Dict[str, Any]],  # TODO: We can expand on this type
-        "event_id": str,
-        "exception": Dict[
-            Literal["values"], List[Dict[str, Any]]
-        ],  # TODO: We can expand on this type
-        # "extra": MutableMapping[str, object],
-        # "fingerprint": List[str],
-        "level": LogLevelStr,
-        # "logentry": Mapping[str, object],
-        "logger": str,
-        # "measurements": Dict[str, MeasurementValue],
-        "message": str,
-        "modules": Dict[str, str],
-        # "monitor_config": Mapping[str, object],
-        "monitor_slug": Optional[str],
-        "platform": Literal["python"],
-        "profile": object,
-        "release": str,
-        "request": Dict[str, object],
-        # "sdk": Mapping[str, object],
-        "server_name": str,
-        "spans": List[Dict[str, object]],
-        "stacktrace": Dict[
-            str, object
-        ],  # We access this key in the code, but I am unsure whether we ever set it
-        "start_timestamp": datetime,
-        "status": Optional[str],
-        # "tags": MutableMapping[
-        #     str, str
-        # ],  # Tags must be less than 200 characters each
-        "threads": Dict[
-            Literal["values"], List[Dict[str, Any]]
-        ],  # TODO: We can expand on this type
-        "timestamp": Optional[datetime],  # Must be set before sending the event
-        "transaction": str,
-        # "transaction_info": Mapping[str, Any],  # TODO: We can expand on this type
-        "type": Literal["check_in", "transaction"],
-        "user": Dict[str, object],
-        "_metrics_summary": Dict[str, object],
-    },
-    total=False,
-)
+if TYPE_CHECKING:
+    from types import FrameType, TracebackType
+    from typing import (  # noqa: F401
+        Any,
+        Callable,
+        Dict,
+        Iterator,
+        List,
+        Literal,
+        Optional,
+        Set,
+        Tuple,
+        Type,
+        TypedDict,
+        TypeVar,
+        Union,
+        cast,
+    )
+
+    ExcInfo = Union[
+        Tuple[Type[BaseException], BaseException, Optional[TracebackType]],
+        Tuple[None, None, None],
+    ]
+    LogLevelStr = Literal["fatal", "critical", "error", "warning", "info", "debug"]
+
+    Event = TypedDict(
+        "Event",
+        {
+            "breadcrumbs": Dict[
+                Literal["values"], List[Dict[str, Any]]
+            ],  # TODO: We can expand on this type
+            "check_in_id": str,
+            "contexts": Dict[str, Dict[str, object]],
+            "dist": str,
+            "duration": Optional[float],
+            "environment": str,
+            "errors": List[Dict[str, Any]],  # TODO: We can expand on this type
+            "event_id": str,
+            "exception": Dict[
+                Literal["values"], List[Dict[str, Any]]
+            ],  # TODO: We can expand on this type
+            # "extra": MutableMapping[str, object],
+            # "fingerprint": List[str],
+            "level": LogLevelStr,
+            # "logentry": Mapping[str, object],
+            "logger": str,
+            # "measurements": Dict[str, MeasurementValue],
+            "message": str,
+            "modules": Dict[str, str],
+            # "monitor_config": Mapping[str, object],
+            "monitor_slug": Optional[str],
+            "platform": Literal["python"],
+            "profile": object,  # Should be sentry_sdk.profiler.Profile, but we can't import that here due to circular imports
+            "release": str,
+            "request": Dict[str, object],
+            # "sdk": Mapping[str, object],
+            "server_name": str,
+            "spans": List[Dict[str, object]],
+            "stacktrace": Dict[
+                str, object
+            ],  # We access this key in the code, but I am unsure whether we ever set it
+            "start_timestamp": datetime,
+            "status": Optional[str],
+            # "tags": MutableMapping[
+            #     str, str
+            # ],  # Tags must be less than 200 characters each
+            "threads": Dict[
+                Literal["values"], List[Dict[str, Any]]
+            ],  # TODO: We can expand on this type
+            "timestamp": Optional[datetime],  # Must be set before sending the event
+            "transaction": str,
+            # "transaction_info": Mapping[str, Any],  # TODO: We can expand on this type
+            "type": Literal["check_in", "transaction"],
+            "user": Dict[str, object],
+            "_metrics_summary": Dict[str, object],
+        },
+        total=False,
+    )
 
 
 epoch = datetime(1970, 1, 1)
@@ -130,6 +136,9 @@ def event_hint_with_exc_info(exc_info=None):
 class AnnotatedValue:
     """
     Meta information for a data field in the event payload.
+    This is to tell Relay that we have tampered with the fields value.
+    See:
+    https://github.com/getsentry/relay/blob/be12cd49a0f06ea932ed9b9f93a655de5d6ad6d1/relay-general/src/types/meta.rs#L407-L423
     """
 
     __slots__ = ("value", "metadata")
@@ -355,9 +364,12 @@ def filename_for_module(module, abs_path):
 def serialize_frame(
     frame,
     tb_lineno=None,
+    include_local_variables=True,
+    include_source_context=True,
     max_value_length=None,
+    custom_repr=None,
 ):
-    # type: (FrameType, Optional[int], Optional[int]) -> Dict[str, Any]
+    # type: (FrameType, Optional[int], bool, bool, Optional[int], Optional[Callable[..., Optional[str]]]) -> Dict[str, Any]
     f_code = getattr(frame, "f_code", None)
     if not f_code:
         abs_path = None
@@ -382,13 +394,50 @@ def serialize_frame(
         "lineno": tb_lineno,
     }  # type: Dict[str, Any]
 
-    rv["pre_context"], rv["context_line"], rv["post_context"] = get_source_context(
-        frame, tb_lineno, max_value_length
-    )
+    if include_source_context:
+        rv["pre_context"], rv["context_line"], rv["post_context"] = get_source_context(
+            frame, tb_lineno, max_value_length
+        )
+
+    if include_local_variables:
+        # TODO(nk): Sort out this current invalid import
+        # from sentry_sdk.serializer import serialize
+
+        # rv["vars"] = serialize(
+        #     dict(frame.f_locals), is_vars=True, custom_repr=custom_repr
+        # )
+        pass
 
     return rv
 
 
+def current_stacktrace(
+    include_local_variables=True,  # type: bool
+    include_source_context=True,  # type: bool
+    max_value_length=None,  # type: Optional[int]
+):
+    # type: (...) -> Dict[str, Any]
+    __tracebackhide__ = True
+    frames = []
+
+    f = sys._getframe()  # type: Optional[FrameType]
+    while f is not None:
+        if not should_hide_frame(f):
+            frames.append(
+                serialize_frame(
+                    f,
+                    include_local_variables=include_local_variables,
+                    include_source_context=include_source_context,
+                    max_value_length=max_value_length,
+                )
+            )
+        f = f.f_back
+
+    frames.reverse()
+
+    return {"frames": frames, "type": "raw"}
+
+
 def get_errno(exc_value):
     # type: (BaseException) -> Optional[Any]
     return getattr(exc_value, "errno", None)
@@ -396,19 +445,18 @@ def get_errno(exc_value):
 
 def get_error_message(exc_value):
     # type: (Optional[BaseException]) -> str
-    message = (
+    return (
         getattr(exc_value, "message", "")
         or getattr(exc_value, "detail", "")
-        or exc_value
+        or safe_str(exc_value)
     )
 
-    return safe_str(message)
-
 
 def single_exception_from_error_tuple(
     exc_type,  # type: Optional[type]
     exc_value,  # type: Optional[BaseException]
     tb,  # type: Optional[TracebackType]
+    client_options=None,  # type: Optional[Dict[str, Any]]
     mechanism=None,  # type: Optional[Dict[str, Any]]
     exception_id=None,  # type: Optional[int]
     parent_id=None,  # type: Optional[int]
@@ -416,7 +464,10 @@ def single_exception_from_error_tuple(
 ):
     # type: (...) -> Dict[str, Any]
     """
-    Creates a dict that goes into the events `exception.values` list
+    Creates a dict that goes into the events `exception.values` list and is ingestible by Sentry.
+
+    See the Exception Interface documentation for more details:
+    https://develop.sentry.dev/sdk/event-payloads/exception/
     """
     exception_value = {}  # type: Dict[str, Any]
     exception_value["mechanism"] = (
@@ -456,13 +507,25 @@ def single_exception_from_error_tuple(
     exception_value["type"] = get_type_name(exc_type)
     exception_value["value"] = get_error_message(exc_value)
 
-    max_value_length = DEFAULT_MAX_VALUE_LENGTH  # fallback
+    if client_options is None:
+        include_local_variables = True
+        include_source_context = True
+        max_value_length = DEFAULT_MAX_VALUE_LENGTH  # fallback
+        custom_repr = None
+    else:
+        include_local_variables = client_options["include_local_variables"]
+        include_source_context = client_options["include_source_context"]
+        max_value_length = client_options["max_value_length"]
+        custom_repr = client_options.get("custom_repr")
 
     frames = [
         serialize_frame(
             tb.tb_frame,
             tb_lineno=tb.tb_lineno,
+            include_local_variables=include_local_variables,
+            include_source_context=include_source_context,
             max_value_length=max_value_length,
+            custom_repr=custom_repr,
         )
         for tb in iter_stacks(tb)
     ]
@@ -518,6 +581,7 @@ def exceptions_from_error(
     exc_type,  # type: Optional[type]
     exc_value,  # type: Optional[BaseException]
     tb,  # type: Optional[TracebackType]
+    client_options=None,  # type: Optional[Dict[str, Any]]
     mechanism=None,  # type: Optional[Dict[str, Any]]
     exception_id=0,  # type: int
     parent_id=0,  # type: int
@@ -527,12 +591,16 @@ def exceptions_from_error(
     """
     Creates the list of exceptions.
     This can include chained exceptions and exceptions from an ExceptionGroup.
+
+    See the Exception Interface documentation for more details:
+    https://develop.sentry.dev/sdk/event-payloads/exception/
     """
 
     parent = single_exception_from_error_tuple(
         exc_type=exc_type,
         exc_value=exc_value,
         tb=tb,
+        client_options=client_options,
         mechanism=mechanism,
         exception_id=exception_id,
         parent_id=parent_id,
@@ -560,6 +628,7 @@ def exceptions_from_error(
                 exc_type=type(cause),
                 exc_value=cause,
                 tb=getattr(cause, "__traceback__", None),
+                client_options=client_options,
                 mechanism=mechanism,
                 exception_id=exception_id,
                 source="__cause__",
@@ -580,6 +649,7 @@ def exceptions_from_error(
                 exc_type=type(context),
                 exc_value=context,
                 tb=getattr(context, "__traceback__", None),
+                client_options=client_options,
                 mechanism=mechanism,
                 exception_id=exception_id,
                 source="__context__",
@@ -594,6 +664,7 @@ def exceptions_from_error(
                 exc_type=type(e),
                 exc_value=e,
                 tb=getattr(e, "__traceback__", None),
+                client_options=client_options,
                 mechanism=mechanism,
                 exception_id=exception_id,
                 parent_id=parent_id,
@@ -606,6 +677,7 @@ def exceptions_from_error(
 
 def exceptions_from_error_tuple(
     exc_info,  # type: ExcInfo
+    client_options=None,  # type: Optional[Dict[str, Any]]
     mechanism=None,  # type: Optional[Dict[str, Any]]
 ):
     # type: (...) -> List[Dict[str, Any]]
@@ -620,6 +692,7 @@ def exceptions_from_error_tuple(
             exc_type=exc_type,
             exc_value=exc_value,
             tb=tb,
+            client_options=client_options,
             mechanism=mechanism,
             exception_id=0,
             parent_id=0,
@@ -629,7 +702,9 @@ def exceptions_from_error_tuple(
         exceptions = []
         for exc_type, exc_value, tb in walk_exception_chain(exc_info):
             exceptions.append(
-                single_exception_from_error_tuple(exc_type, exc_value, tb, mechanism)
+                single_exception_from_error_tuple(
+                    exc_type, exc_value, tb, client_options, mechanism
+                )
             )
 
     exceptions.reverse()
@@ -718,42 +793,11 @@ def set_in_app_in_frames(frames, in_app_exclude, in_app_include, project_root=No
     return frames
 
 
-def exception_is_already_captured(error):
-    # type: (ExceptionArg) -> bool
-    if isinstance(error, BaseException):
-        return hasattr(error, "__posthog_exception_captured")
-    # Autocaptured exceptions are passed as a tuple from our system hooks,
-    # the second item is the exception value (the first is the exception type)
-    elif isinstance(error, tuple) and len(error) > 1:
-        return error[1] is not None and hasattr(
-            error[1], "__posthog_exception_captured"
-        )
-    else:
-        return False  # type: ignore[unreachable]
-
-
-def mark_exception_as_captured(error, uuid):
-    # type: (ExceptionArg, str) -> None
-    if isinstance(error, BaseException):
-        setattr(error, "__posthog_exception_captured", True)
-        setattr(error, "__posthog_exception_uuid", uuid)
-    # Autocaptured exceptions are passed as a tuple from our system hooks,
-    # the second item is the exception value (the first is the exception type)
-    elif isinstance(error, tuple) and len(error) > 1:
-        if error[1] is not None:
-            setattr(error[1], "__posthog_exception_captured", True)
-            setattr(error[1], "__posthog_exception_uuid", uuid)
-
-
 def exc_info_from_error(error):
-    # type: (ExceptionArg) -> ExcInfo
+    # type: (Union[BaseException, ExcInfo]) -> ExcInfo
     if isinstance(error, tuple) and len(error) == 3:
         exc_type, exc_value, tb = error
     elif isinstance(error, BaseException):
-        try:
-            construct_artificial_traceback(error)
-        except Exception:
-            pass
         tb = getattr(error, "__traceback__", None)
         if tb is not None:
             exc_type = type(error)
@@ -778,29 +822,25 @@ def exc_info_from_error(error):
     return exc_info
 
 
-def construct_artificial_traceback(e):
-    # type: (BaseException) -> None
-    if getattr(e, "__traceback__", None) is not None:
-        return
-
-    depth = 0
-    frames = []
-    while True:
-        try:
-            frame = sys._getframe(depth)
-            depth += 1
-        except ValueError:
-            break
-
-        frames.append(frame)
-
-    frames.reverse()
-
-    tb = None
-    for frame in frames:
-        tb = types.TracebackType(tb, frame, frame.f_lasti, frame.f_lineno)
-
-    setattr(e, "__traceback__", tb)
+def event_from_exception(
+    exc_info,  # type: Union[BaseException, ExcInfo]
+    client_options=None,  # type: Optional[Dict[str, Any]]
+    mechanism=None,  # type: Optional[Dict[str, Any]]
+):
+    # type: (...) -> Tuple[Event, Dict[str, Any]]
+    exc_info = exc_info_from_error(exc_info)
+    hint = event_hint_with_exc_info(exc_info)
+    return (
+        {
+            "level": "error",
+            "exception": {
+                "values": exceptions_from_error_tuple(
+                    exc_info, client_options, mechanism
+                )
+            },
+        },
+        hint,
+    )
 
 
 def _module_in_list(name, items):

posthog/integrations/django.py

@@ -1,160 +0,0 @@
-from typing import TYPE_CHECKING, cast
-from posthog import contexts
-
-if TYPE_CHECKING:
-    from django.http import HttpRequest, HttpResponse  # noqa: F401
-    from typing import Callable, Dict, Any, Optional  # noqa: F401
-
-
-class PosthogContextMiddleware:
-    """Middleware to automatically track Django requests.
-
-    This middleware wraps all calls with a posthog context. It attempts to extract the following from the request headers:
-    - Session ID, (extracted from `X-POSTHOG-SESSION-ID`)
-    - Distinct ID, (extracted from `X-POSTHOG-DISTINCT-ID`)
-    - Request URL as $current_url
-    - Request Method as $request_method
-
-    The context will also auto-capture exceptions and send them to PostHog, unless you disable it by setting
-    `POSTHOG_MW_CAPTURE_EXCEPTIONS` to `False` in your Django settings.
-
-    The middleware behaviour is customisable through 3 additional functions:
-    - `POSTHOG_MW_EXTRA_TAGS`, which is a Callable[[HttpRequest], Dict[str, Any]] expected to return a dictionary of additional tags to be added to the context.
-    - `POSTHOG_MW_REQUEST_FILTER`, which is a Callable[[HttpRequest], bool] expected to return `False` if the request should not be tracked.
-    - `POSTHOG_MW_TAG_MAP`, which is a Callable[[Dict[str, Any]], Dict[str, Any]], which you can use to modify the tags before they're added to the context.
-
-    You can use the `POSTHOG_MW_TAG_MAP` function to remove any default tags you don't want to capture, or override them with your own values.
-
-    Context tags are automatically included as properties on all events captured within a context, including exceptions.
-    See the context documentation for more information. The extracted distinct ID and session ID, if found, are used to
-    associate all events captured in the middleware context with the same distinct ID and session as currently active on the
-    frontend. See the documentation for `set_context_session` and `identify_context` for more details.
-    """
-
-    def __init__(self, get_response):
-        # type: (Callable[[HttpRequest], HttpResponse]) -> None
-        self.get_response = get_response
-
-        from django.conf import settings
-
-        if hasattr(settings, "POSTHOG_MW_EXTRA_TAGS") and callable(
-            settings.POSTHOG_MW_EXTRA_TAGS
-        ):
-            self.extra_tags = cast(
-                "Optional[Callable[[HttpRequest], Dict[str, Any]]]",
-                settings.POSTHOG_MW_EXTRA_TAGS,
-            )
-        else:
-            self.extra_tags = None
-
-        if hasattr(settings, "POSTHOG_MW_REQUEST_FILTER") and callable(
-            settings.POSTHOG_MW_REQUEST_FILTER
-        ):
-            self.request_filter = cast(
-                "Optional[Callable[[HttpRequest], bool]]",
-                settings.POSTHOG_MW_REQUEST_FILTER,
-            )
-        else:
-            self.request_filter = None
-
-        if hasattr(settings, "POSTHOG_MW_TAG_MAP") and callable(
-            settings.POSTHOG_MW_TAG_MAP
-        ):
-            self.tag_map = cast(
-                "Optional[Callable[[Dict[str, Any]], Dict[str, Any]]]",
-                settings.POSTHOG_MW_TAG_MAP,
-            )
-        else:
-            self.tag_map = None
-
-        if hasattr(settings, "POSTHOG_MW_CAPTURE_EXCEPTIONS") and isinstance(
-            settings.POSTHOG_MW_CAPTURE_EXCEPTIONS, bool
-        ):
-            self.capture_exceptions = settings.POSTHOG_MW_CAPTURE_EXCEPTIONS
-        else:
-            self.capture_exceptions = True
-
-    def extract_tags(self, request):
-        # type: (HttpRequest) -> Dict[str, Any]
-        tags = {}
-
-        (user_id, user_email) = self.extract_request_user(request)
-
-        # Extract session ID from X-POSTHOG-SESSION-ID header
-        session_id = request.headers.get("X-POSTHOG-SESSION-ID")
-        if session_id:
-            contexts.set_context_session(session_id)
-
-        # Extract distinct ID from X-POSTHOG-DISTINCT-ID header or request user id
-        distinct_id = request.headers.get("X-POSTHOG-DISTINCT-ID") or user_id
-        if distinct_id:
-            contexts.identify_context(distinct_id)
-
-        # Extract user email
-        if user_email:
-            tags["email"] = user_email
-
-        # Extract current URL
-        absolute_url = request.build_absolute_uri()
-        if absolute_url:
-            tags["$current_url"] = absolute_url
-
-        # Extract request method
-        if request.method:
-            tags["$request_method"] = request.method
-
-        # Extract request path
-        if request.path:
-            tags["$request_path"] = request.path
-
-        # Extract IP address
-        ip_address = request.headers.get("X-Forwarded-For")
-        if ip_address:
-            tags["$ip_address"] = ip_address
-
-        # Extract user agent
-        user_agent = request.headers.get("User-Agent")
-        if user_agent:
-            tags["$user_agent"] = user_agent
-
-        # Apply extra tags if configured
-        if self.extra_tags:
-            extra = self.extra_tags(request)
-            if extra:
-                tags.update(extra)
-
-        # Apply tag mapping if configured
-        if self.tag_map:
-            tags = self.tag_map(tags)
-
-        return tags
-
-    def extract_request_user(self, request):
-        user_id = None
-        email = None
-
-        user = getattr(request, "user", None)
-
-        if user and getattr(user, "is_authenticated", False):
-            try:
-                user_id = str(user.pk)
-            except Exception:
-                pass
-
-            try:
-                email = str(user.email)
-            except Exception:
-                pass
-
-        return user_id, email
-
-    def __call__(self, request):
-        # type: (HttpRequest) -> HttpResponse
-        if self.request_filter and not self.request_filter(request):
-            return self.get_response(request)
-
-        with contexts.new_context(self.capture_exceptions):
-            for k, v in self.extract_tags(request).items():
-                contexts.tag(k, v)
-
-            return self.get_response(request)

posthog/scopes.py

@@ -0,0 +1,122 @@
+import contextvars
+from contextlib import contextmanager
+from typing import Any, Callable, Dict, TypeVar, cast
+
+_context_stack: contextvars.ContextVar[list] = contextvars.ContextVar(
+    "posthog_context_stack", default=[{}]
+)
+
+
+def _get_current_context() -> Dict[str, Any]:
+    return _context_stack.get()[-1]
+
+
+@contextmanager
+def new_context(fresh=False):
+    """
+     Create a new context scope that will be active for the duration of the with block.
+     Any tags set within this scope will be isolated to this context. Any exceptions raised
+    or events captured within the context will be tagged with the context tags.
+
+     Args:
+         fresh: Whether to start with a fresh context (default: False).
+                If False, inherits tags from parent context.
+                If True, starts with no tags.
+
+     Examples:
+         # Inherit parent context tags
+         with posthog.new_context():
+             posthog.tag("request_id", "123")
+             # Both this event and the exception will be tagged with the context tags
+             posthog.capture("event_name", {"property": "value"})
+             raise ValueError("Something went wrong")
+
+         # Start with fresh context (no inherited tags)
+         with posthog.new_context(fresh=True):
+             posthog.tag("request_id", "123")
+             # Both this event and the exception will be tagged with the context tags
+             posthog.capture("event_name", {"property": "value"})
+             raise ValueError("Something went wrong")
+
+    """
+    import posthog
+
+    current_tags = _get_current_context().copy()
+    current_stack = _context_stack.get()
+    new_stack = current_stack + [{}] if fresh else current_stack + [current_tags]
+    token = _context_stack.set(new_stack)
+
+    try:
+        yield
+    except Exception as e:
+        posthog.capture_exception(e)
+        raise
+    finally:
+        _context_stack.reset(token)
+
+
+def tag(key: str, value: Any) -> None:
+    """
+    Add a tag to the current context.
+
+    Args:
+        key: The tag key
+        value: The tag value
+
+    Example:
+        posthog.tag("user_id", "123")
+    """
+    _get_current_context()[key] = value
+
+
+def get_tags() -> Dict[str, Any]:
+    """
+    Get all tags from the current context. Note, modifying
+    the returned dictionary will not affect the current context.
+
+    Returns:
+        Dict of all tags in the current context
+    """
+    return _get_current_context().copy()
+
+
+def clear_tags() -> None:
+    """Clear all tags in the current context."""
+    _get_current_context().clear()
+
+
+F = TypeVar("F", bound=Callable[..., Any])
+
+
+def scoped(fresh=False):
+    """
+    Decorator that creates a new context for the function. Simply wraps
+    the function in a with posthog.new_context(): block.
+
+    Args:
+        fresh: Whether to start with a fresh context (default: False)
+
+    Example:
+        @posthog.scoped()
+        def process_payment(payment_id):
+            posthog.tag("payment_id", payment_id)
+            posthog.tag("payment_method", "credit_card")
+
+            # This event will be captured with tags
+            posthog.capture("payment_started")
+            # If this raises an exception, it will be captured with tags
+            # and then re-raised
+            some_risky_function()
+    """
+
+    def decorator(func: F) -> F:
+        from functools import wraps
+
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            with new_context(fresh=fresh):
+                return func(*args, **kwargs)
+
+        return cast(F, wrapper)
+
+    return decorator

posthog/sentry/__init__.py

@@ -0,0 +1 @@
+POSTHOG_ID_TAG = "posthog_distinct_id"

posthog/sentry/django.py

@@ -0,0 +1,28 @@
+from django.conf import settings
+from sentry_sdk import configure_scope
+
+from posthog.sentry import POSTHOG_ID_TAG
+
+GET_DISTINCT_ID = getattr(settings, "POSTHOG_DJANGO", {}).get("distinct_id")
+
+
+def get_distinct_id(request):
+    if not GET_DISTINCT_ID:
+        return None
+    try:
+        return GET_DISTINCT_ID(request)
+    except:  # noqa: E722
+        return None
+
+
+class PosthogDistinctIdMiddleware:
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        with configure_scope() as scope:
+            distinct_id = get_distinct_id(request)
+            if distinct_id:
+                scope.set_tag(POSTHOG_ID_TAG, distinct_id)
+            response = self.get_response(request)
+        return response

posthog/sentry/posthog_integration.py

@@ -0,0 +1,57 @@
+from sentry_sdk._types import MYPY
+from sentry_sdk.hub import Hub
+from sentry_sdk.integrations import Integration
+from sentry_sdk.scope import add_global_event_processor
+from sentry_sdk.utils import Dsn
+
+import posthog
+from posthog.request import DEFAULT_HOST
+from posthog.sentry import POSTHOG_ID_TAG
+
+if MYPY:
+    from typing import Optional  # noqa: F401
+
+    from sentry_sdk._types import Event, Hint  # noqa: F401
+
+
+class PostHogIntegration(Integration):
+    identifier = "posthog-python"
+    organization = None  # The Sentry organization, used to send a direct link from PostHog to Sentry
+    project_id = (
+        None  # The Sentry project id, used to send a direct link from PostHog to Sentry
+    )
+    prefix = "https://sentry.io/organizations/"  # URL of a hosted sentry instance (default: https://sentry.io/organizations/)
+
+    @staticmethod
+    def setup_once():
+        @add_global_event_processor
+        def processor(event, hint):
+            # type: (Event, Optional[Hint]) -> Optional[Event]
+            if Hub.current.get_integration(PostHogIntegration) is not None:
+                if event.get("level") != "error":
+                    return event
+
+                if event.get("tags", {}).get(POSTHOG_ID_TAG):
+                    posthog_distinct_id = event["tags"][POSTHOG_ID_TAG]
+                    event["tags"]["PostHog URL"] = (
+                        f"{posthog.host or DEFAULT_HOST}/person/{posthog_distinct_id}"
+                    )
+
+                    properties = {
+                        "$sentry_event_id": event["event_id"],
+                        "$sentry_exception": event["exception"],
+                    }
+
+                    if PostHogIntegration.organization:
+                        project_id = PostHogIntegration.project_id or (
+                            not not Hub.current.client.dsn
+                            and Dsn(Hub.current.client.dsn).project_id
+                        )
+                        if project_id:
+                            properties["$sentry_url"] = (
+                                f"{PostHogIntegration.prefix}{PostHogIntegration.organization}/issues/?project={project_id}&query={event['event_id']}"
+                            )
+
+                    posthog.capture(posthog_distinct_id, "$exception", properties)
+
+            return event

posthog/test/ai/langchain/test_callbacks.py

@@ -1378,11 +1378,11 @@ def test_langgraph_agent(mock_client):
     )
     graph.invoke(inputs, config={"callbacks": [cb]})
     calls = [call[1] for call in mock_client.capture.call_args_list]
-    assert len(calls) == 15
+    assert len(calls) == 21
     for call in calls:
         assert call["properties"]["$ai_trace_id"] == "test-trace-id"
     assert len([call for call in calls if call["event"] == "$ai_generation"]) == 2
-    assert len([call for call in calls if call["event"] == "$ai_span"]) == 12
+    assert len([call for call in calls if call["event"] == "$ai_span"]) == 18
     assert len([call for call in calls if call["event"] == "$ai_trace"]) == 1
 
 
@@ -1435,13 +1435,11 @@ def test_span_set_parent_ids_for_third_level_run(mock_client, trace_id):
 
     assert mock_client.capture.call_count == 3
 
-    calls = mock_client.capture.call_args_list
-    span_props_2 = calls[0][1]["properties"]
-    span_props_1 = calls[1][1]["properties"]
-    trace_props = calls[2][1]["properties"]
-
-    assert span_props_2["$ai_parent_id"] == span_props_1["$ai_span_id"]
-    assert span_props_1["$ai_parent_id"] == trace_props["$ai_trace_id"]
+    span2, span1, trace = [
+        call[1]["properties"] for call in mock_client.capture.call_args_list
+    ]
+    assert span2["$ai_parent_id"] == span1["$ai_span_id"]
+    assert span1["$ai_parent_id"] == trace["$ai_trace_id"]
 
 
 def test_captures_error_with_details_in_span(mock_client):
@@ -1480,250 +1478,3 @@ def test_captures_error_without_details_in_span(mock_client):
         == "ValueError"
     )
     assert mock_client.capture.call_args_list[1][1]["properties"]["$ai_is_error"]
-
-
-def test_openai_reasoning_tokens(mock_client):
-    """Test that OpenAI reasoning tokens are captured correctly."""
-    prompt = ChatPromptTemplate.from_messages(
-        [("user", "Think step by step about this problem")]
-    )
-
-    # Mock response with reasoning tokens in output_token_details
-    model = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="Let me think through this step by step...",
-                usage_metadata={
-                    "input_tokens": 10,
-                    "output_tokens": 25,
-                    "total_tokens": 35,
-                    "output_token_details": {"reasoning": 15},  # 15 reasoning tokens
-                },
-            )
-        ]
-    )
-
-    callbacks = [CallbackHandler(mock_client)]
-    chain = prompt | model
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "Let me think through this step by step..."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 10
-    assert generation_props["$ai_output_tokens"] == 25
-    assert generation_props["$ai_reasoning_tokens"] == 15
-
-
-def test_anthropic_cache_write_and_read_tokens(mock_client):
-    """Test that Anthropic cache creation and read tokens are captured correctly."""
-    prompt = ChatPromptTemplate.from_messages([("user", "Analyze this large document")])
-
-    # First call with cache creation
-    model_write = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="I've analyzed the document and cached the context.",
-                usage_metadata={
-                    "total_tokens": 1050,
-                    "input_tokens": 1000,
-                    "output_tokens": 50,
-                    "cache_creation_input_tokens": 800,  # Anthropic cache write
-                },
-            )
-        ]
-    )
-
-    callbacks = [CallbackHandler(mock_client)]
-    chain = prompt | model_write
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "I've analyzed the document and cached the context."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 1000
-    assert generation_props["$ai_output_tokens"] == 50
-    assert generation_props["$ai_cache_creation_input_tokens"] == 800
-    assert generation_props["$ai_cache_read_input_tokens"] == 0
-    assert generation_props["$ai_reasoning_tokens"] == 0
-
-    # Reset mock for second call
-    mock_client.reset_mock()
-
-    # Second call with cache read
-    model_read = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="Using cached analysis to provide quick response.",
-                usage_metadata={
-                    "input_tokens": 200,
-                    "output_tokens": 30,
-                    "total_tokens": 1030,
-                    "cache_read_input_tokens": 800,  # Anthropic cache read
-                },
-            )
-        ]
-    )
-
-    chain = prompt | model_read
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "Using cached analysis to provide quick response."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 200
-    assert generation_props["$ai_output_tokens"] == 30
-    assert generation_props["$ai_cache_creation_input_tokens"] == 0
-    assert generation_props["$ai_cache_read_input_tokens"] == 800
-    assert generation_props["$ai_reasoning_tokens"] == 0
-
-
-def test_openai_cache_read_tokens(mock_client):
-    """Test that OpenAI cache read tokens are captured correctly."""
-    prompt = ChatPromptTemplate.from_messages(
-        [("user", "Use the cached prompt for this request")]
-    )
-
-    # Mock response with cache read tokens in input_token_details
-    model = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="Response using cached prompt context.",
-                usage_metadata={
-                    "input_tokens": 150,
-                    "output_tokens": 40,
-                    "total_tokens": 190,
-                    "input_token_details": {
-                        "cache_read": 100,  # 100 tokens read from cache
-                        "cache_creation": 0,
-                    },
-                },
-            )
-        ]
-    )
-
-    callbacks = [CallbackHandler(mock_client)]
-    chain = prompt | model
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "Response using cached prompt context."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 150
-    assert generation_props["$ai_output_tokens"] == 40
-    assert generation_props["$ai_cache_read_input_tokens"] == 100
-    assert generation_props["$ai_cache_creation_input_tokens"] == 0
-    assert generation_props["$ai_reasoning_tokens"] == 0
-
-
-def test_openai_cache_creation_tokens(mock_client):
-    """Test that OpenAI cache creation tokens are captured correctly."""
-    prompt = ChatPromptTemplate.from_messages(
-        [("user", "Create a cache for this large prompt context")]
-    )
-
-    # Mock response with cache creation tokens in input_token_details
-    model = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="Created cache for the prompt context.",
-                usage_metadata={
-                    "input_tokens": 2000,
-                    "output_tokens": 25,
-                    "total_tokens": 2025,
-                    "input_token_details": {
-                        "cache_creation": 1500,  # 1500 tokens written to cache
-                        "cache_read": 0,
-                    },
-                },
-            )
-        ]
-    )
-
-    callbacks = [CallbackHandler(mock_client)]
-    chain = prompt | model
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "Created cache for the prompt context."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 2000
-    assert generation_props["$ai_output_tokens"] == 25
-    assert generation_props["$ai_cache_creation_input_tokens"] == 1500
-    assert generation_props["$ai_cache_read_input_tokens"] == 0
-    assert generation_props["$ai_reasoning_tokens"] == 0
-
-
-def test_combined_reasoning_and_cache_tokens(mock_client):
-    """Test that both reasoning tokens and cache tokens can be captured together."""
-    prompt = ChatPromptTemplate.from_messages(
-        [("user", "Think through this cached problem")]
-    )
-
-    # Mock response with both reasoning and cache tokens
-    model = FakeMessagesListChatModel(
-        responses=[
-            AIMessage(
-                content="Let me reason through this using cached context...",
-                usage_metadata={
-                    "input_tokens": 500,
-                    "output_tokens": 100,
-                    "total_tokens": 600,
-                    "input_token_details": {"cache_read": 300, "cache_creation": 0},
-                    "output_token_details": {"reasoning": 60},  # 60 reasoning tokens
-                },
-            )
-        ]
-    )
-
-    callbacks = [CallbackHandler(mock_client)]
-    chain = prompt | model
-    result = chain.invoke({}, config={"callbacks": callbacks})
-
-    assert result.content == "Let me reason through this using cached context..."
-    assert mock_client.capture.call_count == 3
-
-    generation_args = mock_client.capture.call_args_list[1][1]
-    generation_props = generation_args["properties"]
-
-    assert generation_args["event"] == "$ai_generation"
-    assert generation_props["$ai_input_tokens"] == 500
-    assert generation_props["$ai_output_tokens"] == 100
-    assert generation_props["$ai_cache_read_input_tokens"] == 300
-    assert generation_props["$ai_cache_creation_input_tokens"] == 0
-    assert generation_props["$ai_reasoning_tokens"] == 60
-
-
-@pytest.mark.skipif(not OPENAI_API_KEY, reason="OPENAI_API_KEY is not set")
-def test_openai_reasoning_tokens(mock_client):
-    model = ChatOpenAI(
-        api_key=OPENAI_API_KEY, model="o4-mini", max_completion_tokens=10
-    )
-    cb = CallbackHandler(
-        mock_client, trace_id="test-trace-id", distinct_id="test-distinct-id"
-    )
-    model.invoke("what is the weather in sf", config={"callbacks": [cb]})
-    call = mock_client.capture.call_args_list[0][1]
-    assert call["properties"]["$ai_reasoning_tokens"] is not None
-    assert call["properties"]["$ai_input_tokens"] is not None
-    assert call["properties"]["$ai_output_tokens"] is not None

posthog/test/ai/openai/test_openai.py

@@ -26,11 +26,6 @@ try:
         ResponseOutputMessage,
         ResponseOutputText,
         ResponseUsage,
-        ParsedResponse,
-    )
-    from openai.types.responses.parsed_response import (
-        ParsedResponseOutputMessage,
-        ParsedResponseOutputText,
     )
 
     from posthog.ai.openai import OpenAI
@@ -120,59 +115,6 @@ def mock_openai_response_with_responses_api():
     )
 
 
-@pytest.fixture
-def mock_parsed_response():
-    return ParsedResponse(
-        id="test",
-        model="gpt-4o-2024-08-06",
-        object="response",
-        created_at=1741476542,
-        status="completed",
-        error=None,
-        incomplete_details=None,
-        instructions=None,
-        max_output_tokens=None,
-        tools=[],
-        tool_choice="auto",
-        output=[
-            ParsedResponseOutputMessage(
-                id="msg_123",
-                type="message",
-                role="assistant",
-                status="completed",
-                content=[
-                    ParsedResponseOutputText(
-                        type="output_text",
-                        text='{"name": "Science Fair", "date": "Friday", "participants": ["Alice", "Bob"]}',
-                        annotations=[],
-                        parsed={
-                            "name": "Science Fair",
-                            "date": "Friday",
-                            "participants": ["Alice", "Bob"],
-                        },
-                    )
-                ],
-            )
-        ],
-        output_parsed={
-            "name": "Science Fair",
-            "date": "Friday",
-            "participants": ["Alice", "Bob"],
-        },
-        parallel_tool_calls=True,
-        previous_response_id=None,
-        usage=ResponseUsage(
-            input_tokens=15,
-            output_tokens=20,
-            input_tokens_details={"prompt_tokens": 15, "cached_tokens": 0},
-            output_tokens_details={"reasoning_tokens": 5},
-            total_tokens=35,
-        ),
-        user=None,
-        metadata={},
-    )
-
-
 @pytest.fixture
 def mock_embedding_response():
     return CreateEmbeddingResponse(
@@ -704,73 +646,3 @@ def test_responses_api(mock_client, mock_openai_response_with_responses_api):
         assert props["$ai_http_status"] == 200
         assert props["foo"] == "bar"
         assert isinstance(props["$ai_latency"], float)
-
-
-def test_responses_parse(mock_client, mock_parsed_response):
-    with patch(
-        "openai.resources.responses.Responses.parse",
-        return_value=mock_parsed_response,
-    ):
-        client = OpenAI(api_key="test-key", posthog_client=mock_client)
-        response = client.responses.parse(
-            model="gpt-4o-2024-08-06",
-            input=[
-                {"role": "system", "content": "Extract the event information."},
-                {
-                    "role": "user",
-                    "content": "Alice and Bob are going to a science fair on Friday.",
-                },
-            ],
-            text={
-                "format": {
-                    "type": "json_schema",
-                    "json_schema": {
-                        "name": "event",
-                        "schema": {
-                            "type": "object",
-                            "properties": {
-                                "name": {"type": "string"},
-                                "date": {"type": "string"},
-                                "participants": {
-                                    "type": "array",
-                                    "items": {"type": "string"},
-                                },
-                            },
-                            "required": ["name", "date", "participants"],
-                        },
-                    },
-                }
-            },
-            posthog_distinct_id="test-id",
-            posthog_properties={"foo": "bar"},
-        )
-
-        assert response == mock_parsed_response
-        assert mock_client.capture.call_count == 1
-
-        call_args = mock_client.capture.call_args[1]
-        props = call_args["properties"]
-
-        assert call_args["distinct_id"] == "test-id"
-        assert call_args["event"] == "$ai_generation"
-        assert props["$ai_provider"] == "openai"
-        assert props["$ai_model"] == "gpt-4o-2024-08-06"
-        assert props["$ai_input"] == [
-            {"role": "system", "content": "Extract the event information."},
-            {
-                "role": "user",
-                "content": "Alice and Bob are going to a science fair on Friday.",
-            },
-        ]
-        assert props["$ai_output_choices"] == [
-            {
-                "role": "assistant",
-                "content": '{"name": "Science Fair", "date": "Friday", "participants": ["Alice", "Bob"]}',
-            }
-        ]
-        assert props["$ai_input_tokens"] == 15
-        assert props["$ai_output_tokens"] == 20
-        assert props["$ai_reasoning_tokens"] == 5
-        assert props["$ai_http_status"] == 200
-        assert props["foo"] == "bar"
-        assert isinstance(props["$ai_latency"], float)

posthog/test/exception_integrations/__init__.py

@@ -0,0 +1,3 @@
+import pytest
+
+pytest.importorskip("django")

posthog/test/exception_integrations/test_django.py

@@ -0,0 +1,70 @@
+from posthog.exception_integrations.django import DjangoRequestExtractor
+
+DEFAULT_USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+
+
+def mock_request_factory(override_headers):
+    class Request:
+        META = {}
+        # TRICKY: Actual django request dict object has case insensitive matching, and strips http from the names
+        headers = {
+            "User-Agent": DEFAULT_USER_AGENT,
+            "Referrer": "http://example.com",
+            "X-Forwarded-For": "193.4.5.12",
+            **(override_headers or {}),
+        }
+
+    return Request()
+
+
+def test_request_extractor_with_no_trace():
+    request = mock_request_factory(None)
+    extractor = DjangoRequestExtractor(request)
+    assert extractor.extract_person_data() == {
+        "ip": "193.4.5.12",
+        "user_agent": DEFAULT_USER_AGENT,
+        "traceparent": None,
+        "distinct_id": None,
+    }
+
+
+def test_request_extractor_with_trace():
+    request = mock_request_factory(
+        {"traceparent": "00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01"}
+    )
+    extractor = DjangoRequestExtractor(request)
+    assert extractor.extract_person_data() == {
+        "ip": "193.4.5.12",
+        "user_agent": DEFAULT_USER_AGENT,
+        "traceparent": "00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01",
+        "distinct_id": None,
+    }
+
+
+def test_request_extractor_with_tracestate():
+    request = mock_request_factory(
+        {
+            "traceparent": "00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01",
+            "tracestate": "posthog-distinct-id=1234",
+        }
+    )
+    extractor = DjangoRequestExtractor(request)
+    assert extractor.extract_person_data() == {
+        "ip": "193.4.5.12",
+        "user_agent": DEFAULT_USER_AGENT,
+        "traceparent": "00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01",
+        "distinct_id": "1234",
+    }
+
+
+def test_request_extractor_with_complicated_tracestate():
+    request = mock_request_factory(
+        {"tracestate": "posthog-distinct-id=alohaMountainsXUYZ,rojo=00f067aa0ba902b7"}
+    )
+    extractor = DjangoRequestExtractor(request)
+    assert extractor.extract_person_data() == {
+        "ip": "193.4.5.12",
+        "user_agent": DEFAULT_USER_AGENT,
+        "traceparent": None,
+        "distinct_id": "alohaMountainsXUYZ",
+    }

posthog/test/integrations/test_middleware.py

@@ -1,173 +0,0 @@
-from posthog.contexts import (
-    new_context,
-    get_context_session_id,
-    get_context_distinct_id,
-)
-import unittest
-from unittest.mock import Mock
-
-from posthog.integrations.django import PosthogContextMiddleware
-
-
-class MockRequest:
-    """Mock Django HttpRequest object"""
-
-    def __init__(
-        self,
-        headers=None,
-        method="GET",
-        path="/test",
-        host="example.com",
-        is_secure=False,
-    ):
-        self.headers = headers or {}
-        self.method = method
-        self.path = path
-        self._host = host
-        self._is_secure = is_secure
-
-    def build_absolute_uri(self):
-        scheme = "https" if self._is_secure else "http"
-        return f"{scheme}://{self._host}{self.path}"
-
-
-class TestPosthogContextMiddleware(unittest.TestCase):
-    def create_middleware(
-        self,
-        extra_tags=None,
-        request_filter=None,
-        tag_map=None,
-        capture_exceptions=True,
-    ):
-        """Helper to create middleware instance without calling __init__"""
-        middleware = PosthogContextMiddleware.__new__(PosthogContextMiddleware)
-        middleware.get_response = Mock()
-        middleware.extra_tags = extra_tags
-        middleware.request_filter = request_filter
-        middleware.tag_map = tag_map
-        middleware.capture_exceptions = capture_exceptions
-        return middleware
-
-    def test_extract_tags_basic(self):
-        with new_context():
-            """Test basic tag extraction from request"""
-            middleware = self.create_middleware()
-            request = MockRequest(
-                headers={
-                    "X-POSTHOG-SESSION-ID": "session-123",
-                    "X-POSTHOG-DISTINCT-ID": "user-456",
-                },
-                method="POST",
-                path="/api/test",
-                host="example.com",
-                is_secure=True,
-            )
-
-            tags = middleware.extract_tags(request)
-
-            self.assertEqual(get_context_session_id(), "session-123")
-            self.assertEqual(get_context_distinct_id(), "user-456")
-            self.assertEqual(tags["$current_url"], "https://example.com/api/test")
-            self.assertEqual(tags["$request_method"], "POST")
-
-    def test_extract_tags_missing_headers(self):
-        """Test tag extraction when PostHog headers are missing"""
-
-        with new_context():
-            middleware = self.create_middleware()
-            request = MockRequest(headers={}, method="GET", path="/home")
-
-            tags = middleware.extract_tags(request)
-
-            self.assertIsNone(get_context_session_id())
-            self.assertIsNone(get_context_distinct_id())
-            self.assertEqual(tags["$current_url"], "http://example.com/home")
-            self.assertEqual(tags["$request_method"], "GET")
-
-    def test_extract_tags_partial_headers(self):
-        """Test tag extraction with only some PostHog headers present"""
-
-        with new_context():
-            middleware = self.create_middleware()
-            request = MockRequest(
-                headers={"X-POSTHOG-SESSION-ID": "session-only"}, method="PUT"
-            )
-
-            tags = middleware.extract_tags(request)
-
-            self.assertEqual(get_context_session_id(), "session-only")
-            self.assertIsNone(get_context_distinct_id())
-            self.assertEqual(tags["$request_method"], "PUT")
-
-    def test_extract_tags_with_extra_tags(self):
-        """Test tag extraction with extra_tags function"""
-
-        def extra_tags_func(request):
-            return {"custom_tag": "custom_value", "user_id": "789"}
-
-        with new_context():
-            middleware = self.create_middleware(extra_tags=extra_tags_func)
-            request = MockRequest(
-                headers={"X-POSTHOG-SESSION-ID": "session-123"}, method="GET"
-            )
-
-            tags = middleware.extract_tags(request)
-
-            self.assertEqual(get_context_session_id(), "session-123")
-            self.assertEqual(tags["custom_tag"], "custom_value")
-            self.assertEqual(tags["user_id"], "789")
-
-    def test_extract_tags_with_tag_map(self):
-        """Test tag extraction with tag_map function"""
-
-        def extra_tags_func(request):
-            return {"custom_tag": "custom_value", "user_id": "789"}
-
-        def tag_map_func(tags):
-            if "custom_tag" in tags:
-                tags["mapped_custom_tag"] = f"mapped_{tags['custom_tag']}"
-                del tags["custom_tag"]
-            return tags
-
-        with new_context():
-            middleware = self.create_middleware(
-                tag_map=tag_map_func, extra_tags=extra_tags_func
-            )
-            request = MockRequest(
-                headers={"X-POSTHOG-SESSION-ID": "session-123"}, method="GET"
-            )
-
-            tags = middleware.extract_tags(request)
-
-            self.assertEqual(tags["mapped_custom_tag"], "mapped_custom_value")
-
-    def test_extract_tags_extra_tags_returns_none(self):
-        """Test tag extraction when extra_tags returns None"""
-
-        def extra_tags_func(request):
-            return None
-
-        middleware = self.create_middleware(extra_tags=extra_tags_func)
-        request = MockRequest(method="GET")
-
-        tags = middleware.extract_tags(request)
-
-        self.assertEqual(tags["$request_method"], "GET")
-        # Should not crash when extra_tags returns None
-
-    def test_extract_tags_extra_tags_returns_empty_dict(self):
-        """Test tag extraction when extra_tags returns empty dict"""
-
-        def extra_tags_func(request):
-            return {}
-
-        middleware = self.create_middleware(extra_tags=extra_tags_func)
-        request = MockRequest(method="PATCH")
-
-        tags = middleware.extract_tags(request)
-
-        self.assertEqual(tags["$request_method"], "PATCH")
-
-
-if __name__ == "__main__":
-    unittest.main()

posthog/test/test_before_send.py

@@ -1,218 +0,0 @@
-import unittest
-
-import mock
-
-from posthog.client import Client
-from posthog.test.test_utils import FAKE_TEST_API_KEY
-
-
-class TestClient(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls):
-        # This ensures no real HTTP POST requests are made
-        cls.client_post_patcher = mock.patch("posthog.client.batch_post")
-        cls.consumer_post_patcher = mock.patch("posthog.consumer.batch_post")
-        cls.client_post_patcher.start()
-        cls.consumer_post_patcher.start()
-
-    @classmethod
-    def tearDownClass(cls):
-        cls.client_post_patcher.stop()
-        cls.consumer_post_patcher.stop()
-
-    def set_fail(self, e, batch):
-        """Mark the failure handler"""
-        print("FAIL", e, batch)  # noqa: T201
-        self.failed = True
-
-    def setUp(self):
-        self.failed = False
-        self.client = Client(FAKE_TEST_API_KEY, on_error=self.set_fail)
-
-    def test_before_send_callback_modifies_event(self):
-        """Test that before_send callback can modify events."""
-        processed_events = []
-
-        def my_before_send(event):
-            processed_events.append(event.copy())
-            if "properties" not in event:
-                event["properties"] = {}
-            event["properties"]["processed_by_before_send"] = True
-            return event
-
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=my_before_send,
-                sync_mode=True,
-            )
-            msg_uuid = client.capture(
-                "test_event", distinct_id="user1", properties={"original": "value"}
-            )
-
-            self.assertIsNotNone(msg_uuid)
-
-            # Get the enqueued message from the mock
-            mock_post.assert_called_once()
-            batch_data = mock_post.call_args[1]["batch"]
-            enqueued_msg = batch_data[0]
-
-            self.assertEqual(
-                enqueued_msg["properties"]["processed_by_before_send"], True
-            )
-            self.assertEqual(enqueued_msg["properties"]["original"], "value")
-            self.assertEqual(len(processed_events), 1)
-            self.assertEqual(processed_events[0]["event"], "test_event")
-
-    def test_before_send_callback_drops_event(self):
-        """Test that before_send callback can drop events by returning None."""
-
-        def drop_test_events(event):
-            if event.get("event") == "test_drop_me":
-                return None
-            return event
-
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=drop_test_events,
-                sync_mode=True,
-            )
-
-            # Event should be dropped
-            msg_uuid = client.capture("test_drop_me", distinct_id="user1")
-            self.assertIsNone(msg_uuid)
-
-            # Event should go through
-            msg_uuid = client.capture("keep_me", distinct_id="user1")
-            self.assertIsNotNone(msg_uuid)
-
-            # Check the enqueued message
-            mock_post.assert_called_once()
-            batch_data = mock_post.call_args[1]["batch"]
-            enqueued_msg = batch_data[0]
-            self.assertEqual(enqueued_msg["event"], "keep_me")
-
-    def test_before_send_callback_handles_exceptions(self):
-        """Test that exceptions in before_send don't crash the client."""
-
-        def buggy_before_send(event):
-            raise ValueError("Oops!")
-
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=buggy_before_send,
-                sync_mode=True,
-            )
-            msg_uuid = client.capture("robust_event", distinct_id="user1")
-
-            # Event should still be sent despite the exception
-            self.assertIsNotNone(msg_uuid)
-
-            # Check the enqueued message
-            mock_post.assert_called_once()
-            batch_data = mock_post.call_args[1]["batch"]
-            enqueued_msg = batch_data[0]
-            self.assertEqual(enqueued_msg["event"], "robust_event")
-
-    def test_before_send_callback_works_with_all_event_types(self):
-        """Test that before_send works with capture, set, etc."""
-
-        def add_marker(event):
-            if "properties" not in event:
-                event["properties"] = {}
-            event["properties"]["marked"] = True
-            return event
-
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=add_marker,
-                sync_mode=True,
-            )
-
-            # Test capture
-            msg_uuid = client.capture("event", distinct_id="user1")
-            self.assertIsNotNone(msg_uuid)
-
-            # Test set
-            msg_uuid = client.set(distinct_id="user1", properties={"prop": "value"})
-            self.assertIsNotNone(msg_uuid)
-
-            # Check all events were marked
-            self.assertEqual(mock_post.call_count, 2)
-            for call in mock_post.call_args_list:
-                batch_data = call[1]["batch"]
-                enqueued_msg = batch_data[0]
-                self.assertTrue(enqueued_msg["properties"]["marked"])
-
-    def test_before_send_callback_disabled_when_none(self):
-        """Test that client works normally when before_send is None."""
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=None,
-                sync_mode=True,
-            )
-            msg_uuid = client.capture("normal_event", distinct_id="user1")
-            self.assertIsNotNone(msg_uuid)
-
-            # Check the event was sent normally
-            mock_post.assert_called_once()
-            batch_data = mock_post.call_args[1]["batch"]
-            enqueued_msg = batch_data[0]
-            self.assertEqual(enqueued_msg["event"], "normal_event")
-
-    def test_before_send_callback_pii_scrubbing_example(self):
-        """Test a realistic PII scrubbing use case."""
-
-        def scrub_pii(event):
-            properties = event.get("properties", {})
-
-            # Mask email but keep domain
-            if "email" in properties:
-                email = properties["email"]
-                if "@" in email:
-                    domain = email.split("@")[1]
-                    properties["email"] = f"***@{domain}"
-                else:
-                    properties["email"] = "***"
-
-            # Remove credit card
-            properties.pop("credit_card", None)
-
-            return event
-
-        with mock.patch("posthog.client.batch_post") as mock_post:
-            client = Client(
-                FAKE_TEST_API_KEY,
-                on_error=self.set_fail,
-                before_send=scrub_pii,
-                sync_mode=True,
-            )
-            msg_uuid = client.capture(
-                "form_submit",
-                distinct_id="user1",
-                properties={
-                    "email": "user@example.com",
-                    "credit_card": "1234-5678-9012-3456",
-                    "form_name": "contact",
-                },
-            )
-
-            self.assertIsNotNone(msg_uuid)
-
-            # Check the enqueued message was scrubbed
-            mock_post.assert_called_once()
-            batch_data = mock_post.call_args[1]["batch"]
-            enqueued_msg = batch_data[0]
-
-            self.assertEqual(enqueued_msg["properties"]["email"], "***@example.com")
-            self.assertNotIn("credit_card", enqueued_msg["properties"])
-            self.assertEqual(enqueued_msg["properties"]["form_name"], "contact")

posthog/test/test_contexts.py

@@ -1,207 +0,0 @@
-import unittest
-from unittest.mock import patch
-
-from posthog.contexts import (
-    get_tags,
-    new_context,
-    scoped,
-    tag,
-    identify_context,
-    set_context_session,
-    get_context_session_id,
-    get_context_distinct_id,
-)
-
-
-class TestContexts(unittest.TestCase):
-    def test_tag_and_get_tags(self):
-        with new_context(fresh=True):
-            tag("key1", "value1")
-            tag("key2", 2)
-
-            tags = get_tags()
-            assert tags["key1"] == "value1"
-            assert tags["key2"] == 2
-
-    def test_new_context_isolation(self):
-        with new_context(fresh=True):
-            # Set tag in outer context
-            tag("outer", "value")
-
-            with new_context(fresh=True):
-                # Inner context should start empty
-                assert get_tags() == {}
-
-                # Set tag in inner context
-                tag("inner", "value")
-                assert get_tags()["inner"] == "value"
-
-                # Outer tag should not be visible
-                self.assertNotIn("outer", get_tags())
-
-            with new_context(fresh=False):
-                # Inner context should inherit outer tag
-                assert get_tags() == {"outer": "value"}
-
-            # After exiting context, inner tag should be gone
-            self.assertNotIn("inner", get_tags())
-
-            # Outer tag should still be there
-            assert get_tags()["outer"] == "value"
-
-    def test_nested_contexts(self):
-        with new_context(fresh=True):
-            tag("level1", "value1")
-
-            with new_context(fresh=True):
-                tag("level2", "value2")
-
-                with new_context(fresh=True):
-                    tag("level3", "value3")
-                    assert get_tags() == {"level3": "value3"}
-
-                # Back to level 2
-                assert get_tags() == {"level2": "value2"}
-
-            # Back to level 1
-            assert get_tags() == {"level1": "value1"}
-
-    @patch("posthog.capture_exception")
-    def test_scoped_decorator_success(self, mock_capture):
-        @scoped()
-        def successful_function(x, y):
-            tag("x", x)
-            tag("y", y)
-            return x + y
-
-        result = successful_function(1, 2)
-
-        # Function should execute normally
-        assert result == 3
-
-        # No exception should be captured
-        mock_capture.assert_not_called()
-
-        # Context should be cleared after function execution
-        assert get_tags() == {}
-
-    @patch("posthog.capture_exception")
-    def test_scoped_decorator_exception(self, mock_capture):
-        test_exception = ValueError("Test exception")
-
-        def check_context_on_capture(exception, **kwargs):
-            # Assert tags are available when capture_exception is called
-            current_tags = get_tags()
-            assert current_tags.get("important_context") == "value"
-
-        mock_capture.side_effect = check_context_on_capture
-
-        @scoped()
-        def failing_function():
-            tag("important_context", "value")
-            raise test_exception
-
-        # Function should raise the exception
-        with self.assertRaises(ValueError):
-            failing_function()
-
-        # Verify capture_exception was called
-        mock_capture.assert_called_once_with(test_exception)
-
-        # Context should be cleared after function execution
-        assert get_tags() == {}
-
-    @patch("posthog.capture_exception")
-    def test_new_context_exception_handling(self, mock_capture):
-        test_exception = RuntimeError("Context exception")
-
-        def check_context_on_capture(exception, **kwargs):
-            # Assert inner context tags are available when capture_exception is called
-            current_tags = get_tags()
-            assert current_tags.get("inner_context") == "inner_value"
-
-        mock_capture.side_effect = check_context_on_capture
-
-        # Set up outer context
-        with new_context():
-            tag("outer_context", "outer_value")
-
-            try:
-                with new_context():
-                    tag("inner_context", "inner_value")
-                    raise test_exception
-            except RuntimeError:
-                pass  # Expected exception
-
-            # Outer context should still be intact
-            assert get_tags()["outer_context"] == "outer_value"
-
-        # Verify capture_exception was called
-        mock_capture.assert_called_once_with(test_exception)
-
-    def test_identify_context(self):
-        with new_context(fresh=True):
-            # Initially no distinct ID
-            assert get_context_distinct_id() is None
-
-            # Set distinct ID
-            identify_context("user123")
-            assert get_context_distinct_id() == "user123"
-
-    def test_set_context_session(self):
-        with new_context(fresh=True):
-            # Initially no session ID
-            assert get_context_session_id() is None
-
-            # Set session ID
-            set_context_session("session456")
-            assert get_context_session_id() == "session456"
-
-    def test_context_inheritance_fresh_context(self):
-        with new_context(fresh=True):
-            identify_context("user123")
-            set_context_session("session456")
-
-            with new_context(fresh=True):
-                # Fresh context should not inherit
-                assert get_context_distinct_id() is None
-                assert get_context_session_id() is None
-
-            # Original context should still have values
-            assert get_context_distinct_id() == "user123"
-            assert get_context_session_id() == "session456"
-
-    def test_context_inheritance_non_fresh_context(self):
-        with new_context(fresh=True):
-            identify_context("user123")
-            set_context_session("session456")
-
-            with new_context(fresh=False):
-                # Non-fresh context should inherit
-                assert get_context_distinct_id() == "user123"
-                assert get_context_session_id() == "session456"
-
-                # Override in child context
-                identify_context("user789")
-                set_context_session("session999")
-                assert get_context_distinct_id() == "user789"
-                assert get_context_session_id() == "session999"
-
-            # Original context should still have original values
-            assert get_context_distinct_id() == "user123"
-            assert get_context_session_id() == "session456"
-
-    def test_scoped_decorator_with_context_ids(self):
-        @scoped()
-        def function_with_context():
-            identify_context("user456")
-            set_context_session("session789")
-            return get_context_distinct_id(), get_context_session_id()
-
-        distinct_id, session_id = function_with_context()
-        assert distinct_id == "user456"
-        assert session_id == "session789"
-
-        # Context should be cleared after function execution
-        assert get_context_distinct_id() is None
-        assert get_context_session_id() is None

posthog/test/test_exception_capture.py

@@ -32,3 +32,32 @@ def test_excepthook(tmpdir):
         b'"$exception_list": [{"mechanism": {"type": "generic", "handled": true}, "module": null, "type": "ZeroDivisionError", "value": "division by zero", "stacktrace": {"frames": [{"platform": "python", "filename": "app.py", "abs_path"'
         in output
     )
+
+
+def test_trying_to_use_django_integration(tmpdir):
+    app = tmpdir.join("app.py")
+    app.write(
+        dedent(
+            """
+    from posthog import Posthog, Integrations
+    posthog = Posthog('phc_x', host='https://eu.i.posthog.com', enable_exception_autocapture=True, exception_autocapture_integrations=[Integrations.Django], debug=True, on_error=lambda e, batch: print('error handling batch: ', e, batch))
+
+    # frame_value = "LOL"
+
+    1/0
+    """
+        )
+    )
+
+    with pytest.raises(subprocess.CalledProcessError) as excinfo:
+        subprocess.check_output([sys.executable, str(app)], stderr=subprocess.STDOUT)
+
+    output = excinfo.value.output
+
+    assert b"ZeroDivisionError" in output
+    assert b"LOL" in output
+    assert b"DEBUG:posthog:data uploaded successfully" in output
+    assert (
+        b'"$exception_list": [{"mechanism": {"type": "generic", "handled": true}, "module": null, "type": "ZeroDivisionError", "value": "division by zero", "stacktrace": {"frames": [{"platform": "python", "filename": "app.py", "abs_path"'
+        in output
+    )

posthog/test/test_feature_flag_result.py

@@ -229,9 +229,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
         self.assertEqual(flag_result.variant, None)
         self.assertEqual(flag_result.payload, 300)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -283,9 +283,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
         self.assertEqual(flag_result.payload, {"some": "value"})
 
         patch_capture.assert_called_with(
+            "distinct_id",
             "$feature_flag_called",
-            distinct_id="distinct_id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": "variant-1",
                 "locally_evaluated": True,
@@ -305,9 +305,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
         self.assertIsNone(another_flag_result.payload)
 
         patch_capture.assert_called_with(
+            "another-distinct-id",
             "$feature_flag_called",
-            distinct_id="another-distinct-id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": "variant-2",
                 "locally_evaluated": True,
@@ -345,9 +345,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
         self.assertEqual(flag_result.variant, None)
         self.assertEqual(flag_result.payload, 300)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": False,
@@ -388,9 +388,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
         self.assertEqual(flag_result.get_value(), "variant-1")
         self.assertEqual(flag_result.payload, [1, 2, 3])
         patch_capture.assert_called_with(
+            "distinct_id",
             "$feature_flag_called",
-            distinct_id="distinct_id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": "variant-1",
                 "locally_evaluated": False,
@@ -431,9 +431,9 @@ class TestGetFeatureFlagResult(unittest.TestCase):
 
         self.assertIsNone(flag_result)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "no-person-flag",
                 "$feature_flag_response": None,
                 "locally_evaluated": False,

posthog/test/test_feature_flags.py

@@ -2695,9 +2695,9 @@ class TestCaptureCalls(unittest.TestCase):
         )
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "complex-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -2729,9 +2729,9 @@ class TestCaptureCalls(unittest.TestCase):
         )
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id2",
             "$feature_flag_called",
-            distinct_id="some-distinct-id2",
-            properties={
+            {
                 "$feature_flag": "complex-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -2767,9 +2767,9 @@ class TestCaptureCalls(unittest.TestCase):
         self.assertEqual(patch_flags.call_count, 1)
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id2",
             "$feature_flag_called",
-            distinct_id="some-distinct-id2",
-            properties={
+            {
                 "$feature_flag": "decide-flag",
                 "$feature_flag_response": "decide-value",
                 "locally_evaluated": False,
@@ -2820,9 +2820,9 @@ class TestCaptureCalls(unittest.TestCase):
         )
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "decide-flag",
                 "$feature_flag_response": "decide-variant",
                 "locally_evaluated": False,
@@ -2871,9 +2871,9 @@ class TestCaptureCalls(unittest.TestCase):
         )
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "decide-flag-with-payload",
                 "$feature_flag_response": True,
                 "locally_evaluated": False,
@@ -2948,9 +2948,7 @@ class TestCaptureCalls(unittest.TestCase):
             "featureFlags": {"person-flag": True},
             "featureFlagPayloads": {"person-flag": 300},
         }
-        client = Client(
-            project_api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY
-        )
+        client = Client(api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY)
 
         client.feature_flags = [
             {
@@ -2979,9 +2977,9 @@ class TestCaptureCalls(unittest.TestCase):
         # Assert that capture was called once, with the correct parameters
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -3014,9 +3012,9 @@ class TestCaptureCalls(unittest.TestCase):
 
         self.assertEqual(patch_capture.call_count, 1)
         patch_capture.assert_called_with(
+            "some-distinct-id2",
             "$feature_flag_called",
-            distinct_id="some-distinct-id2",
-            properties={
+            {
                 "$feature_flag": "person-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -3060,9 +3058,9 @@ class TestCaptureCalls(unittest.TestCase):
         )
 
         patch_capture.assert_called_with(
+            "some-distinct-id",
             "$feature_flag_called",
-            distinct_id="some-distinct-id",
-            properties={
+            {
                 "$feature_flag": "complex-flag",
                 "$feature_flag_response": True,
                 "locally_evaluated": True,
@@ -3104,9 +3102,9 @@ class TestCaptureCalls(unittest.TestCase):
                 person_properties={"region": "USA", "name": "Aloha"},
             )
             patch_capture.assert_called_with(
+                distinct_id,
                 "$feature_flag_called",
-                distinct_id=distinct_id,
-                properties={
+                {
                     "$feature_flag": "complex-flag",
                     "$feature_flag_response": True,
                     "locally_evaluated": True,
@@ -5231,9 +5229,7 @@ class TestConsistency(unittest.TestCase):
             "featureFlags": {}
         }  # Ensure decide returns empty flags
 
-        client = Client(
-            project_api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY
-        )
+        client = Client(api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY)
         client.feature_flags = [
             {
                 "id": 1,
@@ -5257,9 +5253,7 @@ class TestConsistency(unittest.TestCase):
             "featureFlagPayloads": {"Beta-Feature": {"some": "value"}},
         }
 
-        client = Client(
-            project_api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY
-        )
+        client = Client(api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY)
         client.feature_flags = [
             {
                 "id": 1,
@@ -5288,9 +5282,7 @@ class TestConsistency(unittest.TestCase):
             "featureFlagPayloads": {"Beta-Feature": {"some": "value"}},
         }
 
-        client = Client(
-            project_api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY
-        )
+        client = Client(api_key=FAKE_TEST_API_KEY, personal_api_key=FAKE_TEST_API_KEY)
         client.feature_flags = [
             {
                 "id": 1,

posthog/test/test_module.py

@@ -7,7 +7,8 @@ class TestModule(unittest.TestCase):
     posthog = None
 
     def _assert_enqueue_result(self, result):
-        self.assertEqual(type(result[0]), str)
+        self.assertEqual(type(result[0]), bool)
+        self.assertEqual(type(result[1]), dict)
 
     def failed(self):
         self.failed = True
@@ -27,7 +28,12 @@ class TestModule(unittest.TestCase):
         self.assertRaises(Exception, self.posthog.capture)
 
     def test_track(self):
-        res = self.posthog.capture("python module event", distinct_id="distinct_id")
+        res = self.posthog.capture("distinct_id", "python module event")
+        self._assert_enqueue_result(res)
+        self.posthog.flush()
+
+    def test_identify(self):
+        res = self.posthog.identify("distinct_id", {"email": "user@email.com"})
         self._assert_enqueue_result(res)
         self.posthog.flush()
 
@@ -36,5 +42,9 @@ class TestModule(unittest.TestCase):
         self._assert_enqueue_result(res)
         self.posthog.flush()
 
+    def test_page(self):
+        self.posthog.page("distinct_id", "https://posthog.com/contact")
+        self.posthog.flush()
+
     def test_flush(self):
         self.posthog.flush()

posthog/test/test_scopes.py

@@ -0,0 +1,138 @@
+import unittest
+from unittest.mock import patch
+
+from posthog.scopes import clear_tags, get_tags, new_context, scoped, tag
+
+
+class TestScopes(unittest.TestCase):
+    def setUp(self):
+        # Reset any context between tests
+        clear_tags()
+
+    def test_tag_and_get_tags(self):
+        tag("key1", "value1")
+        tag("key2", 2)
+
+        tags = get_tags()
+        assert tags["key1"] == "value1"
+        assert tags["key2"] == 2
+
+    def test_clear_tags(self):
+        tag("key1", "value1")
+        assert get_tags()["key1"] == "value1"
+
+        clear_tags()
+        assert get_tags() == {}
+
+    def test_new_context_isolation(self):
+        # Set tag in outer context
+        tag("outer", "value")
+
+        with new_context(fresh=True):
+            # Inner context should start empty
+            assert get_tags() == {}
+
+            # Set tag in inner context
+            tag("inner", "value")
+            assert get_tags()["inner"] == "value"
+
+            # Outer tag should not be visible
+            self.assertNotIn("outer", get_tags())
+
+        with new_context(fresh=False):
+            # Inner context should start empty
+            assert get_tags() == {"outer": "value"}
+
+        # After exiting context, inner tag should be gone
+        self.assertNotIn("inner", get_tags())
+
+        # Outer tag should still be there
+        assert get_tags()["outer"] == "value"
+
+    def test_nested_contexts(self):
+        tag("level1", "value1")
+
+        with new_context(fresh=True):
+            tag("level2", "value2")
+
+            with new_context(fresh=True):
+                tag("level3", "value3")
+                assert get_tags() == {"level3": "value3"}
+
+            # Back to level 2
+            assert get_tags() == {"level2": "value2"}
+
+        # Back to level 1
+        assert get_tags() == {"level1": "value1"}
+
+    @patch("posthog.capture_exception")
+    def test_scoped_decorator_success(self, mock_capture):
+        @scoped()
+        def successful_function(x, y):
+            tag("x", x)
+            tag("y", y)
+            return x + y
+
+        result = successful_function(1, 2)
+
+        # Function should execute normally
+        assert result == 3
+
+        # No exception should be captured
+        mock_capture.assert_not_called()
+
+        # Context should be cleared after function execution
+        assert get_tags() == {}
+
+    @patch("posthog.capture_exception")
+    def test_scoped_decorator_exception(self, mock_capture):
+        test_exception = ValueError("Test exception")
+
+        def check_context_on_capture(exception, **kwargs):
+            # Assert tags are available when capture_exception is called
+            current_tags = get_tags()
+            assert current_tags.get("important_context") == "value"
+
+        mock_capture.side_effect = check_context_on_capture
+
+        @scoped()
+        def failing_function():
+            tag("important_context", "value")
+            raise test_exception
+
+        # Function should raise the exception
+        with self.assertRaises(ValueError):
+            failing_function()
+
+        # Verify capture_exception was called
+        mock_capture.assert_called_once_with(test_exception)
+
+        # Context should be cleared after function execution
+        assert get_tags() == {}
+
+    @patch("posthog.capture_exception")
+    def test_new_context_exception_handling(self, mock_capture):
+        test_exception = RuntimeError("Context exception")
+
+        def check_context_on_capture(exception, **kwargs):
+            # Assert inner context tags are available when capture_exception is called
+            current_tags = get_tags()
+            assert current_tags.get("inner_context") == "inner_value"
+
+        mock_capture.side_effect = check_context_on_capture
+
+        # Set up outer context
+        tag("outer_context", "outer_value")
+
+        try:
+            with new_context():
+                tag("inner_context", "inner_value")
+                raise test_exception
+        except RuntimeError:
+            pass  # Expected exception
+
+        # Verify capture_exception was called
+        mock_capture.assert_called_once_with(test_exception)
+
+        # Outer context should still be intact
+        assert get_tags()["outer_context"] == "outer_value"

posthog/types.py

@@ -1,13 +1,9 @@
 import json
 from dataclasses import dataclass
-from typing import Any, Callable, List, Optional, TypedDict, Union, cast
+from typing import Any, List, Optional, TypedDict, Union, cast
 
 FlagValue = Union[bool, str]
 
-# Type alias for the before_send callback function
-# Takes an event dictionary and returns the modified event or None to drop it
-BeforeSendCallback = Callable[[dict[str, Any]], Optional[dict[str, Any]]]
-
 
 @dataclass(frozen=True)
 class FlagReason:

posthog/utils.py

@@ -7,9 +7,6 @@ from datetime import date, datetime, timezone
 from decimal import Decimal
 from typing import Any, Optional
 from uuid import UUID
-import sys
-import platform
-import distro  # For Linux OS detection
 
 import six
 from dateutil.tz import tzlocal, tzutc
@@ -201,57 +198,3 @@ def str_iequals(value, comparand):
         False
     """
     return str(value).casefold() == str(comparand).casefold()
-
-
-def get_os_info():
-    """
-    Returns standardized OS name and version information.
-    Similar to how user agent parsing works in JS.
-    """
-    os_name = ""
-    os_version = ""
-
-    platform_name = sys.platform
-
-    if platform_name.startswith("win"):
-        os_name = "Windows"
-        if hasattr(platform, "win32_ver"):
-            win_version = platform.win32_ver()[0]
-            if win_version:
-                os_version = win_version
-
-    elif platform_name == "darwin":
-        os_name = "Mac OS X"
-        if hasattr(platform, "mac_ver"):
-            mac_version = platform.mac_ver()[0]
-            if mac_version:
-                os_version = mac_version
-
-    elif platform_name.startswith("linux"):
-        os_name = "Linux"
-        linux_info = distro.info()
-        if linux_info["version"]:
-            os_version = linux_info["version"]
-
-    elif platform_name.startswith("freebsd"):
-        os_name = "FreeBSD"
-        if hasattr(platform, "release"):
-            os_version = platform.release()
-
-    else:
-        os_name = platform_name
-        if hasattr(platform, "release"):
-            os_version = platform.release()
-
-    return os_name, os_version
-
-
-def system_context() -> dict[str, Any]:
-    os_name, os_version = get_os_info()
-
-    return {
-        "$python_runtime": platform.python_implementation(),
-        "$python_version": "%s.%s.%s" % (sys.version_info[:3]),
-        "$os": os_name,
-        "$os_version": os_version,
-    }

posthog/version.py

@@ -1,4 +1,4 @@
-VERSION = "6.0.1"
+VERSION = "4.3.3"
 
 if __name__ == "__main__":
     print(VERSION, end="")  # noqa: T201

pyproject.toml

@@ -8,14 +8,13 @@ dynamic = ["version"]
 description = "Integrate PostHog into any python application."
 authors = [{ name = "PostHog", email = "hey@posthog.com" }]
 maintainers = [{ name = "PostHog", email = "hey@posthog.com" }]
-license = { text = "MIT" }
+license = { text = "MIT License" }
 readme = "README.md"
 requires-python = ">=3.9"
 classifiers = [
     "Development Status :: 5 - Production/Stable",
     "Intended Audience :: Developers",
     "Operating System :: OS Independent",
-    "License :: OSI Approved :: MIT License",
     "Programming Language :: Python",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
@@ -29,7 +28,6 @@ dependencies = [
     "python-dateutil>=2.2",
     "backoff>=1.10.0",
     "distro>=1.5.0",
-    "typing-extensions>=4.2.0",
 ]
 
 [project.urls]
@@ -37,7 +35,6 @@ Homepage = "https://github.com/posthog/posthog-python"
 Repository = "https://github.com/posthog/posthog-python"
 
 [project.optional-dependencies]
-langchain = ["langchain>=0.2.0"]
 dev = [
     "django-stubs",
     "lxml",
@@ -51,12 +48,6 @@ dev = [
     "pre-commit",
     "pydantic",
     "ruff",
-    "setuptools",
-    "packaging",
-    "wheel",
-    "twine",
-    "tomli",
-    "tomli_w",
 ]
 test = [
     "mock>=2.0.0",
@@ -68,15 +59,16 @@ test = [
     "django",
     "openai",
     "anthropic",
-    "langgraph>=0.4.8",
-    "langchain-core>=0.3.65",
-    "langchain-community>=0.3.25",
-    "langchain-openai>=0.3.22",
-    "langchain-anthropic>=0.3.15",
+    "langgraph",
+    "langchain-community>=0.2.0",
+    "langchain-openai>=0.2.0",
+    "langchain-anthropic>=0.2.0",
     "google-genai",
     "pydantic",
     "parameterized>=0.8.1",
 ]
+sentry = ["sentry-sdk", "django"]
+langchain = ["langchain>=0.2.0"]
 
 [tool.setuptools]
 packages = [
@@ -87,9 +79,12 @@ packages = [
     "posthog.ai.anthropic",
     "posthog.ai.gemini",
     "posthog.test",
-    "posthog.integrations",
+    "posthog.sentry",
+    "posthog.exception_integrations",
 ]
 
+license-files = []
+
 [tool.setuptools.dynamic]
 version = { attr = "posthog.version.VERSION" }
 

sentry_django_example/manage.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+"""Django's command-line utility for administrative tasks."""
+
+import os
+import sys
+
+
+def main():
+    """Run administrative tasks."""
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "sentry_django_example.settings")
+    try:
+        from django.core.management import execute_from_command_line
+    except ImportError as exc:
+        raise ImportError(
+            "Couldn't import Django. Are you sure it's installed and "
+            "available on your PYTHONPATH environment variable? Did you "
+            "forget to activate a virtual environment?"
+        ) from exc
+    execute_from_command_line(sys.argv)
+
+
+if __name__ == "__main__":
+    main()

sentry_django_example/sentry_django_example/asgi.py

@@ -0,0 +1,16 @@
+"""
+ASGI config for sentry_django_example project.
+
+It exposes the ASGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/3.2/howto/deployment/asgi/
+"""
+
+import os
+
+from django.core.asgi import get_asgi_application
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "sentry_django_example.settings")
+
+application = get_asgi_application()

sentry_django_example/sentry_django_example/settings.py

@@ -0,0 +1,171 @@
+"""
+Django settings for sentry_django_example project.
+
+Generated by 'django-admin startproject' using Django 3.2.2.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/3.2/topics/settings/
+
+For the full list of settings and their values, see
+https://docs.djangoproject.com/en/3.2/ref/settings/
+"""
+
+from pathlib import Path
+from uuid import uuid4
+
+# Build paths inside the project like this: BASE_DIR / 'subdir'.
+BASE_DIR = Path(__file__).resolve().parent.parent
+
+
+# Quick-start development settings - unsuitable for production
+# See https://docs.djangoproject.com/en/3.2/howto/deployment/checklist/
+
+# SECURITY WARNING: keep the secret key used in production secret!
+SECRET_KEY = "django-insecure-4kzfiq7vb(t0+jbl#vq)u=%06ouf)n*=l%730c8=tk(wkm9i9o"
+
+# SECURITY WARNING: don't run with debug turned on in production!
+DEBUG = True
+
+ALLOWED_HOSTS = []
+
+
+# PostHog Setup (can be a separate app)
+import posthog  # noqa: E402
+
+# You can find this key on the /setup page in PostHog
+posthog.api_key = (
+    "LXP6nQXvo-2TCqGVrWvPah8uJIyVykoMmhnEkEBi5PA"  # TODO: replace with your api key
+)
+
+posthog.personal_api_key = ""
+
+# Where you host PostHog, with no trailing /.
+# You can remove this line if you're using posthog.com
+posthog.host = "http://127.0.0.1:8000"
+
+from posthog.sentry.posthog_integration import PostHogIntegration  # noqa: E402
+
+PostHogIntegration.organization = "posthog"  # TODO: your sentry organization
+# PostHogIntegration.prefix = # TODO: your self hosted Sentry url. (default: https://sentry.io/organizations/)
+
+# Since Sentry doesn't allow Integrations configuration (see https://github.com/getsentry/sentry-python/blob/master/sentry_sdk/integrations/__init__.py#L171-L183)
+# we work around this by setting static class variables beforehand
+
+# Sentry Setup
+import sentry_sdk  # noqa: E402
+from sentry_sdk.integrations.django import DjangoIntegration  # noqa: E402
+
+sentry_sdk.init(
+    dsn="https://27ac54f7f4cf484abf1335436b0c52e5@o344752.ingest.sentry.io/5624115",  # TODO: your Sentry DSN here
+    integrations=[DjangoIntegration(), PostHogIntegration()],
+    # Set traces_sample_rate to 1.0 to capture 100%
+    # of transactions for performance monitoring.
+    # We recommend adjusting this value in production.
+    traces_sample_rate=1.0,
+    # If you wish to associate users to errors (assuming you are using
+    # django.contrib.auth) you may enable sending PII data.
+    send_default_pii=True,
+)
+
+POSTHOG_DJANGO = {
+    "distinct_id": lambda request: str(
+        uuid4()
+    )  # TODO: your logic for generating unique ID, given the request object
+}
+
+# Application definition
+
+INSTALLED_APPS = [
+    "django.contrib.admin",
+    "django.contrib.auth",
+    "django.contrib.contenttypes",
+    "django.contrib.sessions",
+    "django.contrib.messages",
+    "django.contrib.staticfiles",
+]
+
+MIDDLEWARE = [
+    "django.middleware.security.SecurityMiddleware",
+    "django.contrib.sessions.middleware.SessionMiddleware",
+    "django.middleware.common.CommonMiddleware",
+    "django.middleware.csrf.CsrfViewMiddleware",
+    "django.contrib.auth.middleware.AuthenticationMiddleware",
+    "django.contrib.messages.middleware.MessageMiddleware",
+    "django.middleware.clickjacking.XFrameOptionsMiddleware",
+    "posthog.sentry.django.PosthogDistinctIdMiddleware",
+]
+
+ROOT_URLCONF = "sentry_django_example.urls"
+
+TEMPLATES = [
+    {
+        "BACKEND": "django.template.backends.django.DjangoTemplates",
+        "DIRS": [],
+        "APP_DIRS": True,
+        "OPTIONS": {
+            "context_processors": [
+                "django.template.context_processors.debug",
+                "django.template.context_processors.request",
+                "django.contrib.auth.context_processors.auth",
+                "django.contrib.messages.context_processors.messages",
+            ],
+        },
+    },
+]
+
+WSGI_APPLICATION = "sentry_django_example.wsgi.application"
+
+
+# Database
+# https://docs.djangoproject.com/en/3.2/ref/settings/#databases
+
+DATABASES = {
+    "default": {
+        "ENGINE": "django.db.backends.sqlite3",
+        "NAME": BASE_DIR / "db.sqlite3",
+    }
+}
+
+
+# Password validation
+# https://docs.djangoproject.com/en/3.2/ref/settings/#auth-password-validators
+
+AUTH_PASSWORD_VALIDATORS = [
+    {
+        "NAME": "django.contrib.auth.password_validation.UserAttributeSimilarityValidator",
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.MinimumLengthValidator",
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.CommonPasswordValidator",
+    },
+    {
+        "NAME": "django.contrib.auth.password_validation.NumericPasswordValidator",
+    },
+]
+
+
+# Internationalization
+# https://docs.djangoproject.com/en/3.2/topics/i18n/
+
+LANGUAGE_CODE = "en-us"
+
+TIME_ZONE = "UTC"
+
+USE_I18N = True
+
+USE_L10N = True
+
+USE_TZ = True
+
+
+# Static files (CSS, JavaScript, Images)
+# https://docs.djangoproject.com/en/3.2/howto/static-files/
+
+STATIC_URL = "/static/"
+
+# Default primary key field type
+# https://docs.djangoproject.com/en/3.2/ref/settings/#default-auto-field
+
+DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"

sentry_django_example/sentry_django_example/urls.py

@@ -0,0 +1,28 @@
+"""sentry_django_example URL Configuration
+
+The `urlpatterns` list routes URLs to views. For more information please see:
+    https://docs.djangoproject.com/en/3.2/topics/http/urls/
+Examples:
+Function views
+    1. Add an import:  from my_app import views
+    2. Add a URL to urlpatterns:  path('', views.home, name='home')
+Class-based views
+    1. Add an import:  from other_app.views import Home
+    2. Add a URL to urlpatterns:  path('', Home.as_view(), name='home')
+Including another URLconf
+    1. Import the include() function: from django.urls import include, path
+    2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
+"""
+
+from django.contrib import admin
+from django.urls import path
+
+
+def trigger_error(request):
+    division_by_zero = 1 / 0
+
+
+urlpatterns = [
+    path("admin/", admin.site.urls),
+    path("sentry-debug/", trigger_error),
+]

sentry_django_example/sentry_django_example/wsgi.py

@@ -0,0 +1,16 @@
+"""
+WSGI config for sentry_django_example project.
+
+It exposes the WSGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/3.2/howto/deployment/wsgi/
+"""
+
+import os
+
+from django.core.wsgi import get_wsgi_application
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "sentry_django_example.settings")
+
+application = get_wsgi_application()

setup.cfg

@@ -0,0 +1,12 @@
+[bdist_wheel]
+universal = 1
+
+[tool:pytest]
+asyncio_mode = auto
+asyncio_default_fixture_loop_scope = function
+
+[flake8]
+# ignore E501 for line length
+# ignore W503 for line break before binary operator
+ignore = E501,W503
+max-line-length = 120

setup.py

@@ -28,6 +28,7 @@ setup(
     author_email="hey@posthog.com",
     maintainer="PostHog",
     maintainer_email="hey@posthog.com",
+    test_suite="posthog.test.all",
     license="MIT License",
     description="Integrate PostHog into any python application.",
     long_description=long_description,

setup_analytics.py

@@ -1,8 +1,5 @@
 import os
 import sys
-import tomli
-import tomli_w
-import shutil
 
 try:
     from setuptools import setup
@@ -10,39 +7,9 @@ except ImportError:
     from distutils.core import setup
 
 # Don't import analytics-python module here, since deps may not be installed
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), "posthoganalytics"))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "posthog"))
 from version import VERSION  # noqa: E402
 
-
-# Copy the original pyproject.toml as backup
-shutil.copy("pyproject.toml", "pyproject.toml.backup")
-
-# Read the original pyproject.toml
-with open("pyproject.toml", "rb") as f:
-    config = tomli.load(f)
-
-# Override specific values
-config["project"]["name"] = "posthoganalytics"
-config["tool"]["setuptools"]["dynamic"]["version"] = {
-    "attr": "posthoganalytics.version.VERSION"
-}
-
-# Rename packages from posthog.* to posthoganalytics.*
-if "packages" in config["tool"]["setuptools"]:
-    new_packages = []
-    for package in config["tool"]["setuptools"]["packages"]:
-        if package == "posthog":
-            new_packages.append("posthoganalytics")
-        elif package.startswith("posthog."):
-            new_packages.append(package.replace("posthog.", "posthoganalytics.", 1))
-        else:
-            new_packages.append(package)
-    config["tool"]["setuptools"]["packages"] = new_packages
-
-# Overwrite the original pyproject.toml
-with open("pyproject.toml", "wb") as f:
-    tomli_w.dump(config, f)
-
 long_description = """
 PostHog is developer-friendly, self-hosted product analytics.
 posthog-python is the python package.
@@ -61,6 +28,7 @@ setup(
     author_email="hey@posthog.com",
     maintainer="PostHog",
     maintainer_email="hey@posthog.com",
+    test_suite="posthog.test.all",
     license="MIT License",
     description="Integrate PostHog into any python application.",
     long_description=long_description,

simulator.py

@@ -0,0 +1,112 @@
+import argparse
+import json
+import logging
+
+import posthog
+
+__name__ = "simulator.py"
+__version__ = "0.0.1"
+__description__ = "scripting simulator"
+
+
+def json_hash(str):
+    if str:
+        return json.loads(str)
+
+
+# posthog -method=<method> -posthog-write-key=<posthogWriteKey> [options]
+
+
+parser = argparse.ArgumentParser(description="send a posthog message")
+
+parser.add_argument("--writeKey", help="the posthog writeKey")
+parser.add_argument("--type", help="The posthog message type")
+
+parser.add_argument("--distinct_id", help="the user id to send the event as")
+parser.add_argument("--anonymousId", help="the anonymous user id to send the event as")
+
+parser.add_argument("--event", help="the event name to send with the event")
+parser.add_argument("--properties", help="the event properties to send (JSON-encoded)")
+
+parser.add_argument(
+    "--name", help="name of the screen or page to send with the message"
+)
+
+parser.add_argument("--traits", help="the identify/group traits to send (JSON-encoded)")
+
+parser.add_argument("--groupId", help="the group id")
+
+options = parser.parse_args()
+
+
+def failed(status, msg):
+    raise Exception(msg)
+
+
+def capture():
+    posthog.capture(
+        options.distinct_id,
+        options.event,
+        anonymous_id=options.anonymousId,
+        properties=json_hash(options.properties),
+    )
+
+
+def page():
+    posthog.page(
+        options.distinct_id,
+        name=options.name,
+        anonymous_id=options.anonymousId,
+        properties=json_hash(options.properties),
+    )
+
+
+def identify():
+    posthog.identify(
+        options.distinct_id,
+        anonymous_id=options.anonymousId,
+        traits=json_hash(options.traits),
+    )
+
+
+def set_once():
+    posthog.set_once(
+        options.distinct_id,
+        properties=json_hash(options.traits),
+    )
+
+
+def set():
+    posthog.set(
+        options.distinct_id,
+        properties=json_hash(options.traits),
+    )
+
+
+def unknown():
+    print()
+
+
+posthog.api_key = options.writeKey
+posthog.on_error = failed
+posthog.debug = True
+
+log = logging.getLogger("posthog")
+ch = logging.StreamHandler()
+ch.setLevel(logging.DEBUG)
+log.addHandler(ch)
+
+switcher = {
+    "capture": capture,
+    "page": page,
+    "identify": identify,
+    "set_once": set_once,
+    "set": set,
+}
+
+func = switcher.get(options.type)
+if func:
+    func()
+    posthog.shutdown()
+else:
+    print("Invalid Message Type " + options.type)