Skip to content

Lightning-fast PII detection and anonymization library with 190x performance advantage - detect emails, SSNs, names, and more in <2MB package

License

Notifications You must be signed in to change notification settings

DataFog/datafog-python

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DataFog logo

Comprehensive PII Detection & Anonymization
Intelligent Engine Selection • Lightweight • Production Ready

PyPi Version PyPI pyversions GitHub stars PyPi downloads Tests Lint Benchmarks Discord Code style: black codecov GitHub Issues

DataFog is a comprehensive open-source library for detecting and anonymizing personally identifiable information (PII) in unstructured data. Built for production workloads, it delivers intelligent engine selection to handle both structured identifiers and contextual entities across different industries and use cases.

⚡ Why Choose DataFog?

đź§  Intelligent Engine Selection

  • Automatically chooses the best detection approach for your data
  • Pattern-based engine for structured PII (emails, phones, SSNs, credit cards)
  • NLP-based engine for contextual entities (names, organizations, locations)
  • Industry-optimized detection across financial, healthcare, legal, and enterprise domains

📦 Lightweight & Modular

  • Core package under 2MB (vs 800MB+ alternatives)
  • Install only what you need: datafog[nlp], datafog[ocr], datafog[all]
  • Zero ML model downloads for basic usage

🎯 Production Ready

  • Comprehensive PII coverage for diverse enterprise needs
  • Battle-tested detection patterns with high precision
  • Comprehensive test suite with 99.4% coverage
  • CLI tools and Python SDK for any workflow

đź”§ Developer Friendly

  • Simple API: detect("Contact john@example.com")
  • Multiple anonymization methods: redact, replace, hash
  • OCR support for images and documents

Installation

DataFog can be installed via pip:

pip install datafog

CLI

📚 Quick Reference

Command Description
scan-text Analyze text for PII
scan-image Extract and analyze text from images
redact-text Redact PII in text
replace-text Replace PII with anonymized values
hash-text Hash PII in text
health Check service status
show-config Display current settings
download-model Get a specific spaCy model
list-spacy-models Show available models
list-entities View supported PII entities

🔍 Detailed Usage

Scanning Text

To scan and annotate text for PII entities:

datafog scan-text "Your text here"

Example:

datafog scan-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"

Scanning Images

To extract text from images and optionally perform PII annotation:

datafog scan-image "path/to/image.png" --operations extract

Example:

datafog scan-image "nokia-statement.png" --operations extract

To extract text and annotate PII:

datafog scan-image "nokia-statement.png" --operations scan

Redacting Text

To redact PII in text:

datafog redact-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"

which should output:

[REDACTED] is the CEO of [REDACTED] and is based out of [REDACTED], [REDACTED]

Replacing Text

To replace detected PII:

datafog replace-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"

which should return something like:

[PERSON_B86CACE6] is the CEO of [UNKNOWN_445944D7] and is based out of [UNKNOWN_32BA5DCA], [UNKNOWN_B7DF4969]

Note: a unique randomly generated identifier is created for each detected entity

Hashing Text

You can select from SHA256, SHA3-256, and MD5 hashing algorithms to hash detected PII. Currently the hashed output does not match the length of the original entity, for privacy-preserving purposes. The default is SHA256.

datafog hash-text "Tim Cook is the CEO of Apple and is based out of Cupertino, California"

generating an output which looks like this:

5738a37f0af81594b8a8fd677e31b5e2cabd6d7791c89b9f0a1c233bb563ae39 is the CEO of f223faa96f22916294922b171a2696d868fd1f9129302eb41a45b2a2ea2ebbfd and is based out of ab5f41f04096cf7cd314357c4be26993eeebc0c094ca668506020017c35b7a9c, cad0535decc38b248b40e7aef9a1cfd91ce386fa5c46f05ea622649e7faf18fb

Utility Commands

🏥 Health Check

datafog health

⚙️ Show Configuration

datafog show-config

📥 Download Model

datafog download-model en_core_web_sm

đź“‚ Show Model Directory

datafog show-spacy-model-directory en_core_web_sm

đź“‹ List Models

datafog list-spacy-models

🏷️ List Entities

datafog list-entities

⚠️ Important Notes

  • For scan-image and scan-text commands, use --operations to specify different operations. Default is scan.
  • Process multiple images or text strings in a single command by providing multiple arguments.
  • Ensure proper permissions and configuration of the DataFog service before running commands.

đź’ˇ Tip: For more detailed information on each command, use the --help option, e.g., datafog scan-text --help.

Python SDK

Getting Started

To use DataFog, you'll need to create a DataFog client with the desired operations. Here's a basic setup:

from datafog import DataFog

# For text annotation
client = DataFog(operations="scan")

# For OCR (Optical Character Recognition)
ocr_client = DataFog(operations="extract")

Engine Selection

DataFog now supports multiple annotation engines through the TextService class. You can choose between different engines for PII detection:

from datafog.services.text_service import TextService

# Use fast engine only (fastest, pattern-based detection)
fast_service = TextService(engine="regex")

# Use spaCy engine only (more comprehensive NLP-based detection)
spacy_service = TextService(engine="spacy")

# Use auto mode (default) - tries fast engine first, falls back to spaCy if no entities found
auto_service = TextService()  # engine="auto" is the default

Each engine targets different PII detection needs:

  • regex: Pattern-based detection optimized for structured identifiers like emails, phone numbers, credit cards, SSNs, and IP addresses
  • spacy: NLP-based entity recognition for contextual entities like names, organizations, locations, dates, and monetary amounts
  • auto: Intelligent selection - tries pattern-based detection first, falls back to NLP for comprehensive contextual analysis

Text PII Annotation

Here's an example of how to annotate PII in a text document:

import requests

# Fetch sample medical record
doc_url = "https://gist.githubusercontent.com/sidmohan0/b43b72693226422bac5f083c941ecfdb/raw/b819affb51796204d59987893f89dee18428ed5d/note1.txt"
response = requests.get(doc_url)
text_lines = [line for line in response.text.splitlines() if line.strip()]

# Run annotation
annotations = client.run_text_pipeline_sync(str_list=text_lines)
print(annotations)

OCR PII Annotation

For OCR capabilities, you can use the following:

import asyncio
import nest_asyncio

nest_asyncio.apply()


async def run_ocr_pipeline_demo():
    image_url = "https://s3.amazonaws.com/thumbnails.venngage.com/template/dc377004-1c2d-49f2-8ddf-d63f11c8d9c2.png"
    results = await ocr_client.run_ocr_pipeline(image_urls=[image_url])
    print("OCR Pipeline Results:", results)


loop = asyncio.get_event_loop()
loop.run_until_complete(run_ocr_pipeline_demo())

Note: The DataFog library uses asynchronous programming for OCR, so make sure to use the async/await syntax when calling the appropriate methods.

Text Anonymization

DataFog provides various anonymization techniques to protect sensitive information. Here are examples of how to use them:

Redacting Text

To redact PII in text:

from datafog import DataFog
from datafog.config import OperationType

client = DataFog(operations=[OperationType.SCAN, OperationType.REDACT])

text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
redacted_text = client.run_text_pipeline_sync([text])[0]
print(redacted_text)

Output:

[REDACTED] is the CEO of [REDACTED] and is based out of [REDACTED], [REDACTED]

Replacing Text

To replace detected PII with unique identifiers:

from datafog import DataFog
from datafog.config import OperationType

client = DataFog(operations=[OperationType.SCAN, OperationType.REPLACE])

text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
replaced_text = client.run_text_pipeline_sync([text])[0]
print(replaced_text)

Output:

[PERSON_B86CACE6] is the CEO of [UNKNOWN_445944D7] and is based out of [UNKNOWN_32BA5DCA], [UNKNOWN_B7DF4969]

Hashing Text

To hash detected PII:

from datafog import DataFog
from datafog.config import OperationType
from datafog.models.anonymizer import HashType

client = DataFog(operations=[OperationType.SCAN, OperationType.HASH], hash_type=HashType.SHA256)

text = "Tim Cook is the CEO of Apple and is based out of Cupertino, California"
hashed_text = client.run_text_pipeline_sync([text])[0]
print(hashed_text)

Output:

5738a37f0af81594b8a8fd677e31b5e2cabd6d7791c89b9f0a1c233bb563ae39 is the CEO of f223faa96f22916294922b171a2696d868fd1f9129302eb41a45b2a2ea2ebbfd and is based out of ab5f41f04096cf7cd314357c4be26993eeebc0c094ca668506020017c35b7a9c, cad0535decc38b248b40e7aef9a1cfd91ce386fa5c46f05ea622649e7faf18fb

You can choose from SHA256 (default), SHA3-256, and MD5 hashing algorithms by specifying the hash_type parameter

PII Detection Capabilities

DataFog provides multiple annotation engines designed for different PII detection scenarios:

Engine Selection

The TextService class supports three engine modes:

# Use regex engine for structured identifiers
regex_service = TextService(engine="regex")

# Use spaCy engine for contextual entities
spacy_service = TextService(engine="spacy")

# Use auto mode (default) - intelligent engine selection
auto_service = TextService()  # engine="auto" is the default

PII Coverage by Engine

Different engines excel at detecting different types of personally identifiable information:

Engine PII Types Detected Best For
Regex EMAIL, PHONE, SSN, CREDIT_CARD, IP_ADDRESS, DOB, ZIP Financial services, healthcare, compliance
SpaCy PERSON, ORG, GPE, CARDINAL, DATE, TIME, MONEY, PRODUCT Legal documents, communication monitoring, general text
Auto All of the above (context-dependent) Mixed data sources, unknown content types

Industry-Specific Use Cases

Financial Services & Healthcare:

  • Primary need: Structured identifiers (SSNs, credit cards, account numbers)
  • Recommended: regex engine for high precision on regulatory requirements
  • Common PII: ~60% structured identifiers, ~40% names/addresses

Legal & Document Review:

  • Primary need: Names, organizations, locations in unstructured text
  • Recommended: spacy engine for comprehensive entity recognition
  • Common PII: ~30% structured identifiers, ~70% contextual entities

Enterprise Communication & Mixed Content:

  • Primary need: Both structured and contextual PII detection
  • Recommended: auto engine for intelligent selection
  • Benefits from both engines depending on content type

When to Use Each Engine

Regex Engine: Choose when you need to detect specific, well-formatted identifiers:

  • Processing structured databases or forms
  • Compliance scanning for specific regulatory requirements (GDPR, HIPAA)
  • High-volume processing where deterministic results are important
  • Financial data with credit cards, SSNs, account numbers

SpaCy Engine: Choose when you need contextual understanding:

  • Analyzing unstructured documents, emails, or communications
  • Legal eDiscovery where names and organizations are key
  • Content where entities don't follow standard patterns
  • Multi-language support requirements

Auto Engine: Choose for general-purpose PII detection:

  • Unknown or mixed content types
  • Applications serving multiple industries
  • When you want comprehensive coverage without manual engine selection
  • Default choice for most production applications

Running Detection Tests

You can test the different engines locally using pytest:

pip install pytest-benchmark
pytest tests/benchmark_text_service.py -v

Examples

For more detailed examples, check out our Jupyter notebooks in the examples/ directory:

  • text_annotation_example.ipynb: Demonstrates text PII annotation
  • image_processing.ipynb: Shows OCR capabilities and text extraction from images

These notebooks provide step-by-step guides on how to use DataFog for various tasks.

Dev Notes

For local development:

  1. Clone the repository.
  2. Navigate to the project directory:
    cd datafog-python
    
  3. Create a new virtual environment (using .venv is recommended as it is hardcoded in the justfile):
    python -m venv .venv
    
  4. Activate the virtual environment:
    • On Windows:
      .venv\Scripts\activate
      
    • On macOS/Linux:
      source .venv/bin/activate
      
  5. Install the package in editable mode:
    pip install -r requirements-dev.txt
    
  6. Set up the project:
    just setup
    

Now, you can develop and run the project locally.

Important Actions:

  • Format the code:
    just format
    
    This runs isort to sort imports.
  • Lint the code:
    just lint
    
    This runs flake8 to check for linting errors.
  • Generate coverage report:
    just coverage-html
    
    This runs pytest and generates a coverage report in the htmlcov/ directory.

We use pre-commit to run checks locally before committing changes. Once installed, you can run:

pre-commit run --all-files

Dependencies

For OCR, we use Tesseract, which is incorporated into the build step. You can find the relevant configurations under .github/workflows/ in the following files:

  • dev-cicd.yml
  • feature-cicd.yml
  • main-cicd.yml

Testing

  • Python 3.10

License

This software is published under the MIT license.

About

Lightning-fast PII detection and anonymization library with 190x performance advantage - detect emails, SSNs, names, and more in <2MB package

Topics

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages