All Scaffoldings

Load Squirro data in Python using dltHub

Build a Squirro-to-database pipeline in Python using dlt with AI Workbench support for Claude Code, Cursor, and Codex.

Last updated:

In this guide, we'll set up a complete Squirro data pipeline from API credentials to your first data load in just 10 minutes. You'll end up with a fully declarative Python pipeline based on dlt's REST API connector, like in the partial example code below:

Example code
@dlt.source
def squirro_source(access_token=dlt.secrets.value):
    config: RESTAPIConfig = {
        "client": {
            "base_url": "https://localhost:9200/_nodes/stats",
            "auth": {
                "type": "bearer",
                "token": access_token,
            }
        },
        "resources": [
            "os", "indices"
        ],
    }
    [...]
    yield from rest_api_resources(config)


def get_data() -> None:
    # Connect to destination
    pipeline = dlt.pipeline(
        pipeline_name='squirro_pipeline',
        destination='duckdb',
        dataset_name='squirro_data', 
    )
    # Load the data
    load_info = pipeline.run(squirro_source())
    print(load_info)

Why use dlt to generate Python pipelines?

  • Accelerate pipeline development with AI-native context
  • Debug pipelines, validate schemas and data with the integrated Pipeline Dashboard
  • Build Python notebooks for end users of your data
  • Low maintenance thanks to schema evolution with type inference, resilience and self-documenting REST API connectors. A shallow learning curve makes the pipeline easy to extend by any team member
  • dlt is the tool of choice for Pythonic Iceberg Lakehouses, bringing mature data loading to Iceberg with or without catalogs

What you’ll do

We’ll show you how to generate a readable and easily maintainable Python script that fetches data from Squirro's API and loads it into Iceberg, DataFrames, files, or a database of your choice. Here are some of the endpoints you can load:

  • Snapshot Repository: Endpoints related to managing and interacting with snapshot repositories for data backup.

    • /_snapshot/my_repository: Access or manage the specified snapshot repository.

  • Node Information: Endpoints that provide details about the nodes in the cluster.

    • /_nodes?pretty: Get a list of nodes with detailed information in a readable format.
    • /_nodes/stats?pretty: Retrieve statistics about the nodes in a friendly format.
    • /_nodes/stats/jvm?pretty: Get JVM statistics for each node in the cluster.
    • /_nodes/stats/fs?pretty: Access file system statistics for the nodes.

  • Cluster Settings and Health: Endpoints for managing cluster settings and health checks.

    • /_cluster/settings: View or modify cluster-wide settings.
    • /_cluster/health?pretty=true: Check the health status of the cluster in a readable format.
    • /_cluster/reroute?retry_failed: Reroute shards that have failed to be allocated.

  • Shards and Indices: Endpoints that provide information about shards and indices in the cluster.

    • /_cat/shards?v: List all shards in the cluster with verbose output.
    • /_cat/shards: Get a concise list of all shards.
    • /_cat/indices?v: Display information about all indices in the cluster.

  • Template Management: Endpoints related to index templates and component templates.

    • /_index_template/squirro_v9?pretty: Access or manage a specific index template.
    • /_index_template?pretty: Retrieve all index templates in a readable format.
    • /_component_template/metadata_property: Manage component templates related to metadata properties.

  • Settings and Stats: Endpoints for retrieving settings and statistics of various components.

    • /_settings: Access settings for various indices or the cluster.
    • /_stats?pretty: Get overall statistics for indices in a readable format.
    • /_nodes/stats/indices/fielddata?pretty: Retrieve field data statistics for indices in a friendly format.

You will then debug the Squirro pipeline using our Pipeline Dashboard tool to ensure it is copying the data correctly, before building a Notebook to explore your data and build reports.

Setup & steps to follow

💡
Before getting started, set up a virtual environment (instructions) and install the dlt workspace:


uv venv && source .venv/bin/activate
uv pip install "dlt[workspace]"

Now you're ready to get started!

  1. Install the dlt AI Workbench

    Configure the workbench for your coding assistant:

    dlt ai init --agent <your-agent> # <agent>: claude | cursor | codex

    This installs project rules, a secrets management skill, appropriate ignore files, and configures the dlt MCP server for your agent.

    Learn more about the dltHub AI Workbench and setup details for each assistant →

  2. Install the rest-api-pipeline toolkit

    The AI Workbench provides different toolkits for each phase of the data engineering lifecycle. To start you need to install the rest-api-pipeline toolkit:

    dlt ai toolkit rest-api-pipeline install

    This loads different skills and contexts about dlt the agent uses to build the pipeline iteratively, efficiently, and safely. Importantly, it does not need to ask you for credentials directly. In dlt, API credentials are provided via a secrets.toml file (learn more about secrets management →), and the agent should use the MCP tools to see their shape and detect misconfigurations. It never needs to access the file directly.

    Learn more about the rest-api-pipeline toolkit →

  3. Start LLM-assisted coding

    Here's a prompt to get you started:

    Prompt
    Use /find-source to load data from the Squirro API into DuckDB.

    The AI Workbench rest-api-pipeline toolkit takes over from here — it reads relevant API documentation, presents you with options for which endpoints to load, and then follows a structured workflow to scaffold, debug, and validate the pipeline step by step.

  4. View the result

    After the rest-api-pipeline workflow has finished, you will end up with a working REST API source with validated endpoints and a pipeline that writes data into a local dataset you have inspected and verified.

    > python squirro_pipeline.py
Pipeline squirro load step completed in 0.26 seconds
1 load package(s) were loaded to destination duckdb and into dataset squirro_data
The duckdb destination used duckdb:/squirro.duckdb location to store data
Load package 1749667187.541553 is LOADED and contains no failed jobs

    By launching the Pipeline Dashboard, you can see various information about the pipeline and the loaded data

    • Pipeline overview: State, load metrics
    • Data's schema: tables, columns, types, hints
    • You can query the data itself
    dlt pipeline squirro_pipeline show

Next steps

You can go to the next phases of your data engineering journey by handing over to other toolkits of the dltHub AI Workbench:

  • data-exploration — Build custom notebooks, charts, and dashboards for deeper analysis with marimo notebooks.
  • dlthub-runtime — Deploy, schedule, and monitor your pipeline in production
dlt ai toolkit data-exploration install
dlt ai toolkit dlthub-runtime install

Or explore the following resources for more information:

Was this page helpful?

Community Hub

Need more dlt context for Squirro?

Request dlt skills, commands, AGENT.md files, and AI-native context.

Request more contextSubmit context