Skip to main content

dagster-rocky

Dagster integration for Rocky, the typed graph between your code and your warehouse.

dagster-rocky wraps the rocky CLI as a Dagster ConfigurableResource and exposes Rocky-managed tables as materializable Dagster assets. Compile-time contracts, column-level lineage, schema-drift detection, quality check results, and per-model cost surface as native Dagster events, so the guarantees Rocky enforces at compile time appear directly in the Dagster asset graph.

Install

uv add dagster-rocky

You'll also need the Rocky CLI on your $PATH: https://github.com/rocky-data/rocky/releases?q=engine

Quick start (component)

Add a defs.yaml next to your other Dagster definitions:

type: dagster_rocky.RockyComponent
attributes:
  binary_path: rocky
  config_path: config/rocky.toml
  models_dir: models

Dagster's component loader will:

  1. Run rocky discover (and rocky compile, when models are present) and cache the result.
  2. Build one subset-aware multi_asset per Rocky group, with declared row_count / column_match / freshness checks per table.
  3. On materialization, shell out to rocky run --filter <key>=<value> for the selected subset and yield MaterializeResult + AssetCheckResult events with rich metadata.

Quick start (resource)

import dagster as dg
from dagster_rocky import RockyResource

rocky = RockyResource(
    binary_path="rocky",
    config_path="config/rocky.toml",
    timeout_seconds=3600,
)

defs = dg.Definitions(resources={"rocky": rocky})

Then in an asset:

@dg.asset
def acme_orders(rocky: RockyResource) -> dg.MaterializeResult:
    result = rocky.run(filter="tenant=acme")
    return dg.MaterializeResult(
        metadata={"tables_copied": result.tables_copied, "duration_ms": result.duration_ms},
    )

Public API

Symbol Purpose
RockyResource ConfigurableResource wrapping the Rocky CLI
RockyComponent State-backed Dagster component that loads Rocky tables as assets
RockyDagsterTranslator Subclass to customize asset key / group / tag mapping
RockyMetadataSet Namespaced metadata (source_id, strategy, watermark, …)
load_rocky_assets() Functional helper that returns AssetSpec for each Rocky table
emit_materializations() / emit_check_results() Convert a RunResult into Dagster events
check_metadata() Build a metadata mapping for a single Rocky CheckResult
cost_metadata_from_optimize() Extract per-model cost recommendations from OptimizeResult
parse_rocky_output() Auto-detect a Rocky JSON payload and return the matching Pydantic model

Documentation

Related projects

License

Apache 2.0

Download files

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

Source Distribution

dagster_rocky-1.54.1.tar.gz (355.1 kB view details)

Uploaded Source

Built Distribution

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

dagster_rocky-1.54.1-py3-none-any.whl (119.9 kB view details)

Uploaded Python 3

File details

Details for the file dagster_rocky-1.54.1.tar.gz.

File metadata

  • Download URL: dagster_rocky-1.54.1.tar.gz
  • Upload date:
  • Size: 355.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for dagster_rocky-1.54.1.tar.gz
Algorithm Hash digest
SHA256 ba69527820b923e460e8d0e4afcf5f77a3d84e9e4e0aea1d4234cd026bd1dff0
MD5 0dfd55b78ec2c9573ad6ba4a56e7baf0
BLAKE2b-256 9a48dc0fbfe394075990f05d32f4a962fdda2196c7b214bc066585a19e0c8440

See more details on using hashes here.

Provenance

The following attestation bundles were made for dagster_rocky-1.54.1.tar.gz:

Publisher: dagster-release.yml on rocky-data/rocky

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file dagster_rocky-1.54.1-py3-none-any.whl.

File metadata

  • Download URL: dagster_rocky-1.54.1-py3-none-any.whl
  • Upload date:
  • Size: 119.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for dagster_rocky-1.54.1-py3-none-any.whl
Algorithm Hash digest
SHA256 4a44c94ca59c86c8262839a9560219b75f3e1e2b41b8a57903084fd8acebfe2c
MD5 20a51a21cbb25481bac75f8bb01014eb
BLAKE2b-256 b3d540ed18f43139da2148442f27893b26801824c6df57d79c8f68826e26638c

See more details on using hashes here.

Provenance

The following attestation bundles were made for dagster_rocky-1.54.1-py3-none-any.whl:

Publisher: dagster-release.yml on rocky-data/rocky

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

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