Skip to main content

Runtime PII leak detector for Django — find personal data leaking from your API responses and third-party calls

Project description

Wiregraph

Detect and visualize live PII flow inside running Django apps.

Your view returns more than you think. Your OpenAI call ships customer emails to a third party. Your Stripe webhook echoes an SSN into a log line. Static analysis won't catch it. APM won't flag it.

Wiregraph is a runtime PII leak detector that sits inside your Django app and watches the traffic you're actually serving — inbound, outbound, and egress to third parties.

PII leak in the dashboard

The alarm moment

A user hits /api/v1/support/ticket/. Your view enriches the response with an OpenAI summary. Wiregraph sees this:

[egress] POST api.openai.com/v1/chat/completions
  └─ EMAIL_ADDRESS  (confidence 0.99)  asset: request.body.messages[1].content
  └─ US_SSN         (confidence 0.95)  asset: request.body.messages[1].content

[response] 200 /api/v1/support/ticket/
  └─ PHONE_NUMBER   (confidence 0.90)  asset: response.body.ticket.notes

You didn't write code to log that. You didn't know it was happening. Now you do.

Why runtime detection?

PII leaks happen at runtime, not in source:

  • Serializers mutate fields you didn't expect to expose
  • LLM calls forward whatever the prompt builder concatenates
  • Middleware rewrites responses after your view returns
  • Third-party SDKs ship payloads you never see in your repo
  • Dynamic queries return columns that weren't in the original spec

Grep won't find any of these. APM tells you a request was slow. Logs tell you what you decided to log. Wiregraph tells you what your app actually sent.

What you get

  • Runtime visibility into every PII-bearing field crossing your app's boundary — inbound, outbound, egress
  • Audit trail of which endpoints leak what, by tenant
  • Compliance evidence — exportable PDF/JSON reports of observed flows
  • Zero raw PII at rest — detections are hashed, masked, or truncated before storage

Quick start

1. Install and configure

pip install wiregraph
# settings.py
import wiregraph

INSTALLED_APPS = [*INSTALLED_APPS, *wiregraph.INSTALLED_APPS]
MIDDLEWARE = wiregraph.setup(MIDDLEWARE)
WIREGRAPH = {"ENABLED": True}

Install and configure

2. Migrate and attach a superuser to a tenant

python manage.py migrate
python manage.py wiregraph_doctor   # sanity-check config

# create a superuser if you don't already have one
export DJANGO_SUPERUSER_PASSWORD=admin
python manage.py createsuperuser --noinput --username admin --email a@a.com

# attach your admin user to a tenant so events get attributed
python manage.py shell -c "from django.contrib.auth import get_user_model; from wiregraph_apps.tenants.models import Tenant, TenantMembership; u = get_user_model().objects.filter(is_superuser=True).first(); t, _ = Tenant.objects.get_or_create(name='Demo Co', defaults={'slug': 'demo'}); TenantMembership.objects.get_or_create(user=u, tenant=t)"

Migrate and create tenant

3. Hit an endpoint and watch the leak

Hit any endpoint, then open /admin/wiregraph/dashboard/ to see the flow graph. (For the JSON API at /api/v1/, install wiregraph[drf] — see below.)

Wiregraph skips the Django admin URL prefix by default — otherwise the DataEvent admin would re-detect its own contents on refresh. Override with AUTO_EXCLUDE_ADMIN=False.

Architecture

   ┌─────────────┐
   │  Request    │
   └──────┬──────┘
          ▼
   ┌─────────────────────┐
   │  Wiregraph          │  ← scans request body, headers, query
   │  middleware         │
   └──────┬──────────────┘
          ▼
   ┌─────────────┐         ┌──────────────────┐
   │  Your view  │────────▶│  Egress hook     │ ← scans outbound
   └──────┬──────┘         │  (OpenAI, Stripe)│   third-party calls
          ▼                └──────────────────┘
   ┌─────────────────────┐
   │  Response scanner   │  ← scans response body
   └──────┬──────────────┘
          ▼
   ┌─────────────────┐
   │  DataEvent      │  → classified, redacted, stored
   │  (hashed PII)   │  → /admin/wiregraph/dashboard/
   └─────────────────┘

Classification

Not every PII hit is a leak. An email to api.stripe.com is the product working; the same email to api.openai.com is an incident. Wiregraph classifies every DataEvent on what × where × policy and writes the verdict to DataEvent.outcome:

Outcome Meaning
expected Asset flowed to a sink that accepts it (Stripe ← email)
acceptable Trusted sink, asset not on its accept-list but not dangerous
suspicious Unknown sink, new flow, or known sink receiving an unexpected asset
prohibited Sensitive asset to a sink that should never receive it

A built-in catalog (~15 vendors: Stripe, OpenAI, Anthropic, Bedrock, Twilio, SendGrid, Segment, S3, Auth0, …) means a fresh install gets meaningful outcomes with zero setup. Override per-host via SINK_OVERRIDES (settings) or SinkCatalogOverride (DB, tenant-scoped). LLM strictness is tunable: LLM_POLICY = "strict" (default — medium+ PII to LLM is prohibited) or "relaxed".

Detection

Regex out of the box: emails, phones, SSNs, credit cards, and more. Add your own:

WIREGRAPH = {
    "ENABLED": True,
    "CUSTOM_PATTERNS": [
        {"name": "emp_id", "regex": r"\bEMP-\d{6}\b", "confidence": 0.9},
    ],
}

Optional Presidio (ML NER for names, addresses, IBANs, 50+ entity types) runs async via Celery so the request path stays fast:

WIREGRAPH = {"ENABLED": True, "ENABLE_PRESIDIO": True}

See Installing Presidio. Presidio matches that overlap a regex hit on the same asset are deduped (regex wins).

Security guarantee

Wiregraph never persists raw PII. Every detection is redacted (hash / mask / truncate, configurable) before it touches storage. The matched value lives in memory only long enough to classify and redact.

Install extras

Wiregraph ships in slices so you only pull in what you need.

Install line What you get
pip install wiregraph Core middleware + bundled admin dashboard at /admin/wiregraph/dashboard/. No DRF.
pip install wiregraph[drf] Adds the JSON API at /api/v1/ (viewsets, JWT auth, OpenAPI schema). Required by the React wiregraph-dashboard consumer.
pip install wiregraph[presidio] ML-based detection. Also run python -m spacy download en_core_web_lg.
pip install wiregraph[export] PDF/JSON compliance reports.
pip install wiregraph[postgres] psycopg[binary] for Postgres backends.
pip install wiregraph[all] Everything above plus dev tooling.

If you skip [drf], the /api/v1/ routes are not registered — the admin dashboard still works.

Tenant resolution

By default Wiregraph walks request.user.tenant_memberships. Point TENANT_RESOLVER at your own callable if your project stores tenancy differently (FK, subdomain, header).

Retention purge

Delete events older than DATA_RETENTION_DAYS:

python manage.py wiregraph_purge [--dry-run]

Or via Celery Beat:

import wiregraph.celery as wg_celery
CELERY_BEAT_SCHEDULE = {**wg_celery.schedule(hour=3, minute=0)}

API

Requires pip install wiregraph[drf]. Without the extra, /api/v1/ routes are not registered.

Versioned under /api/v1/, JWT Bearer auth (obtain via /api/v1/auth/token/). OpenAPI at /api/v1/schema/, Swagger at /api/v1/schema/docs/.

Method Path Description
POST /api/v1/auth/token/ Obtain access + refresh token
POST /api/v1/auth/token/refresh/ Rotate refresh token
GET /api/v1/detection/events/ List events (filter: direction, data_asset, endpoint, timestamp__gte, timestamp__lte)
GET /api/v1/detection/events/{id}/ Retrieve event
GET /api/v1/detection/assets/ PII categories seen in traffic
GET /api/v1/detection/stats/summary/ Dashboard counts

See docs/settings.md for all config keys.

Requirements

Python ≥ 3.10 · Django ≥ 4.2 · Celery + Redis (async detection) · PostgreSQL recommended

Roadmap

A standalone React dashboard is in development — interactive graph exploration, historical flow analysis, and per-tenant views beyond what the bundled Django admin dashboard offers today.

Wiregraph React dashboard preview

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

wiregraph-0.1.0.post3.tar.gz (67.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

wiregraph-0.1.0.post3-py3-none-any.whl (107.1 kB view details)

Uploaded Python 3

File details

Details for the file wiregraph-0.1.0.post3.tar.gz.

File metadata

  • Download URL: wiregraph-0.1.0.post3.tar.gz
  • Upload date:
  • Size: 67.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.12

File hashes

Hashes for wiregraph-0.1.0.post3.tar.gz
Algorithm Hash digest
SHA256 33e65213047e1b93bcd9abf27986286fe17c8ff772a82e1165804b3d59b5911b
MD5 92fc8d70c28ff72fb5c432b12b877338
BLAKE2b-256 b88b54f70247d9fcbab98b2fe742a966f11bcd29df389d0aa2f62a6bdbd96f74

See more details on using hashes here.

File details

Details for the file wiregraph-0.1.0.post3-py3-none-any.whl.

File metadata

File hashes

Hashes for wiregraph-0.1.0.post3-py3-none-any.whl
Algorithm Hash digest
SHA256 0ad5f6488be635380650f81c43d95d458822352addd14e591266e23239369a32
MD5 ec8864d54bd600cbd6a8cac818790f56
BLAKE2b-256 6e485f19d0370e9118005531c67c9d5d21d943ae193a3e81f8fb65428242dff4

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page