Data Contract CLI
The datacontract
CLI is an open-source command-line tool for working with data contracts.
It uses data contract YAML files as Data Contract Specification or ODCS to lint the data contract, connect to data sources and execute schema and quality tests, detect breaking changes, and export to different formats. The tool is written in Python. It can be used as a standalone CLI tool, in a CI/CD pipeline, or directly as a Python library.
Getting started
Let's look at this data contract:
https://datacontract.com/examples/orders-latest/datacontract.yaml
We have a servers section with endpoint details to the S3 bucket, models for the structure of the data, servicelevels and quality attributes that describe the expected freshness and number of rows.
This data contract contains all information to connect to S3 and check that the actual data meets the defined schema and quality requirements. We can use this information to test if the actual data product in S3 is compliant to the data contract.
Let's use pip to install the CLI (or use the Docker image),
$ python3 -m pip install datacontract-cli[all]
now, let's run the tests:
$ datacontract test https://datacontract.com/examples/orders-latest/datacontract.yaml
Testing https://datacontract.com/examples/orders-latest/datacontract.yaml
╭────────┬─────────────────────────────────────────────────────────────────────┬───────────────────────────────┬─────────╮
│ Result │ Check │ Field │ Details │
├────────┼─────────────────────────────────────────────────────────────────────┼───────────────────────────────┼─────────┤
│ passed │ Check that JSON has valid schema │ orders │ │
│ passed │ Check that JSON has valid schema │ line_items │ │
│ passed │ Check that field order_id is present │ orders │ │
│ passed │ Check that field order_timestamp is present │ orders │ │
│ passed │ Check that field order_total is present │ orders │ │
│ passed │ Check that field customer_id is present │ orders │ │
│ passed │ Check that field customer_email_address is present │ orders │ │
│ passed │ row_count >= 5000 │ orders │ │
│ passed │ Check that required field order_id has no null values │ orders.order_id │ │
│ passed │ Check that unique field order_id has no duplicate values │ orders.order_id │ │
│ passed │ duplicate_count(order_id) = 0 │ orders.order_id │ │
│ passed │ Check that required field order_timestamp has no null values │ orders.order_timestamp │ │
│ passed │ freshness(order_timestamp) < 24h │ orders.order_timestamp │ │
│ passed │ Check that required field order_total has no null values │ orders.order_total │ │
│ passed │ Check that required field customer_email_address has no null values │ orders.customer_email_address │ │
│ passed │ Check that field lines_item_id is present │ line_items │ │
│ passed │ Check that field order_id is present │ line_items │ │
│ passed │ Check that field sku is present │ line_items │ │
│ passed │ values in (order_id) must exist in orders (order_id) │ line_items.order_id │ │
│ passed │ row_count >= 5000 │ line_items │ │
│ passed │ Check that required field lines_item_id has no null values │ line_items.lines_item_id │ │
│ passed │ Check that unique field lines_item_id has no duplicate values │ line_items.lines_item_id │ │
╰────────┴─────────────────────────────────────────────────────────────────────┴───────────────────────────────┴─────────╯
🟢 data contract is valid. Run 22 checks. Took 6.739514 seconds.
Voilà, the CLI tested that the datacontract.yaml itself is valid, all records comply with the schema, and all quality attributes are met.
We can also use the datacontract.yaml to export in many formats, e.g., to generate a SQL DDL:
$ datacontract export --format sql https://datacontract.com/examples/orders-latest/datacontract.yaml
-- Data Contract: urn:datacontract:checkout:orders-latest
-- SQL Dialect: snowflake
CREATE TABLE orders (
order_id TEXT not null primary key,
order_timestamp TIMESTAMP_TZ not null,
order_total NUMBER not null,
customer_id TEXT,
customer_email_address TEXT not null,
processed_timestamp TIMESTAMP_TZ not null
);
CREATE TABLE line_items (
lines_item_id TEXT not null primary key,
order_id TEXT,
sku TEXT
);
Or generate an HTML export:
$ datacontract export --format html https://datacontract.com/examples/orders-latest/datacontract.yaml > datacontract.html
which will create this HTML export.
Usage
$ datacontract init datacontract.yaml
$ datacontract lint datacontract.yaml
$ datacontract test datacontract.yaml
$ datacontract test --examples datacontract.yaml
$ datacontract export --format html datacontract.yaml > datacontract.html
$ datacontract import --format avro --source avro_schema.avsc
$ datacontract diff datacontract-v1.yaml datacontract-v2.yaml
$ datacontract changelog datacontract-v1.yaml datacontract-v2.yaml
$ datacontract breaking datacontract-v1.yaml datacontract-v2.yaml
Programmatic (Python)
from datacontract.data_contract import DataContract
data_contract = DataContract(data_contract_file="datacontract.yaml")
run = data_contract.test()
if not run.has_passed():
print("Data quality validation failed.")
Installation
Choose the most appropriate installation method for your needs:
pip
Python 3.10, 3.11, and 3.12 are supported. We recommend to use Python 3.11.
python3 -m pip install datacontract-cli[all]
pipx
pipx installs into an isolated environment.
pipx install datacontract-cli[all]
Docker
You can also use our Docker image to run the CLI tool. It is also convenient for CI/CD pipelines.
docker pull datacontract/cli
docker run --rm -v ${PWD}:/home/datacontract datacontract/cli
You can create an alias for the Docker command to make it easier to use:
alias datacontract='docker run --rm -v "${PWD}:/home/datacontract" datacontract/cli:latest'
Note: The output of Docker command line messages is limited to 80 columns and may include line breaks. Don't pipe docker output to files if you want to export code. Use the --output
option instead.
Optional Dependencies
The CLI tool defines several optional dependencies (also known as extras) that can be installed for using with specific servers types.
With all, all server dependencies are included.
pip install datacontract-cli[all]
A list of available extras:
Dependency | Installation Command |
---|
Avro Support | pip install datacontract-cli[avro] |
Google BigQuery | pip install datacontract-cli[bigquery] |
Databricks Integration | pip install datacontract-cli[databricks] |
Iceberg | pip install datacontract-cli[iceberg] |
Kafka Integration | pip install datacontract-cli[kafka] |
PostgreSQL Integration | pip install datacontract-cli[postgres] |
S3 Integration | pip install datacontract-cli[s3] |
Snowflake Integration | pip install datacontract-cli[snowflake] |
Microsoft SQL Server | pip install datacontract-cli[sqlserver] |
Trino | pip install datacontract-cli[trino] |
Dbt | pip install datacontract-cli[dbt] |
Dbml | pip install datacontract-cli[dbml] |
Parquet | pip install datacontract-cli[parquet] |
Documentation
Commands
init
Usage: datacontract init [OPTIONS] [LOCATION]
Download a datacontract.yaml template and write it to file.
╭─ Arguments ──────────────────────────────────────────────────────────────────────────────────╮
│ location [LOCATION] The location (url or path) of the data contract yaml to create. │
│ [default: datacontract.yaml] │
╰──────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ────────────────────────────────────────────────────────────────────────────────────╮
│ --template TEXT URL of a template or data contract │
│ [default: │
│ https://datacontract.com/datacontract.init.yaml] │
│ --overwrite --no-overwrite Replace the existing datacontract.yaml │
│ [default: no-overwrite] │
│ --help Show this message and exit. │
╰──────────────────────────────────────────────────────────────────────────────────────────────╯
lint
Usage: datacontract lint [OPTIONS] [LOCATION]
Validate that the datacontract.yaml is correctly formatted.
╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ location [LOCATION] The location (url or path) of the data contract yaml. [default: datacontract.yaml] │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --schema TEXT The location (url or path) of the Data Contract Specification JSON Schema │
│ [default: https://datacontract.com/datacontract.schema.json] │
│ --help Show this message and exit. │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
test
Usage: datacontract test [OPTIONS] [LOCATION]
Run schema and quality tests on configured servers.
╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ location [LOCATION] The location (url or path) of the data contract yaml. [default: datacontract.yaml] │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --schema TEXT The location (url or path) of the Data Contract │
│ Specification JSON Schema │
│ [default: │
│ https://datacontract.com/datacontract.schema.json] │
│ --server TEXT The server configuration to run the schema and quality │
│ tests. Use the key of the server object in the data │
│ contract yaml file to refer to a server, e.g., │
│ `production`, or `all` for all servers (default). │
│ [default: all] │
│ --examples --no-examples Run the schema and quality tests on the example data │
│ within the data contract. │
│ [default: no-examples] │
│ --publish TEXT The url to publish the results after the test │
│ [default: None] │
│ --publish-to-opentelemetry --no-publish-to-opentelemetry Publish the results to opentelemetry. Use environment │
│ variables to configure the OTLP endpoint, headers, etc. │
│ [default: no-publish-to-opentelemetry] │
│ --logs --no-logs Print logs [default: no-logs] │
│ --help Show this message and exit. │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Data Contract CLI connects to a data source and runs schema and quality tests to verify that the data contract is valid.
$ datacontract test --server production datacontract.yaml
To connect to the databases the server
block in the datacontract.yaml is used to set up the connection.
In addition, credentials, such as username and passwords, may be defined with environment variables.
The application uses different engines, based on the server type
.
Internally, it connects with DuckDB, Spark, or a native connection and executes the most tests with soda-core and fastjsonschema.
Credentials are provided with environment variables.
Supported server types:
Supported formats:
- parquet
- json
- csv
- delta
- iceberg (coming soon)
Feel free to create an issue, if you need support for an additional type and formats.
S3
Data Contract CLI can test data that is stored in S3 buckets or any S3-compliant endpoints in various formats.
- CSV
- JSON
- Delta
- Parquet
- Iceberg (coming soon)
Examples
JSON
datacontract.yaml
servers:
production:
type: s3
endpointUrl: https://minio.example.com
location: s3://bucket-name/path/*/*.json
format: json
delimiter: new_line
Delta Tables
datacontract.yaml
servers:
production:
type: s3
endpointUrl: https://minio.example.com
location: s3://bucket-name/path/table.delta
format: delta
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_S3_REGION | eu-central-1 | Region of S3 bucket |
DATACONTRACT_S3_ACCESS_KEY_ID | AKIAXV5Q5QABCDEFGH | AWS Access Key ID |
DATACONTRACT_S3_SECRET_ACCESS_KEY | 93S7LRrJcqLaaaa/XXXXXXXXXXXXX | AWS Secret Access Key |
DATACONTRACT_S3_SESSION_TOKEN | AQoDYXdzEJr... | AWS temporary session token (optional) |
Google Cloud Storage (GCS)
The S3 integration also works with files on Google Cloud Storage through its interoperability.
Use https://storage.googleapis.com
as the endpoint URL.
Example
datacontract.yaml
servers:
production:
type: s3
endpointUrl: https://storage.googleapis.com
location: s3://bucket-name/path/*/*.json
format: json
delimiter: new_line
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_S3_ACCESS_KEY_ID | GOOG1EZZZ... | The GCS HMAC Key Key ID |
DATACONTRACT_S3_SECRET_ACCESS_KEY | PDWWpb... | The GCS HMAC Key Secret |
BigQuery
We support authentication to BigQuery using Service Account Key. The used Service Account should include the roles:
- BigQuery Job User
- BigQuery Data Viewer
Example
datacontract.yaml
servers:
production:
type: bigquery
project: datameshexample-product
dataset: datacontract_cli_test_dataset
models:
datacontract_cli_test_table:
type: table
fields: ...
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_BIGQUERY_ACCOUNT_INFO_JSON_PATH | ~/service-access-key.json | Service Access key as saved on key creation by BigQuery. If this environment variable isn't set, the cli tries to use GOOGLE_APPLICATION_CREDENTIALS as a fallback, so if you have that set for using their Python library anyway, it should work seamlessly. |
Azure
Data Contract CLI can test data that is stored in Azure Blob storage or Azure Data Lake Storage (Gen2) (ADLS) in various formats.
Example
datacontract.yaml
servers:
production:
type: azure
location: abfss://datameshdatabricksdemo.dfs.core.windows.net/dataproducts/inventory_events/*.parquet
format: parquet
Environment Variables
Authentication works with an Azure Service Principal (SPN) aka App Registration with a secret.
Environment Variable | Example | Description |
---|
DATACONTRACT_AZURE_TENANT_ID | 79f5b80f-10ff-40b9-9d1f-774b42d605fc | The Azure Tenant ID |
DATACONTRACT_AZURE_CLIENT_ID | 3cf7ce49-e2e9-4cbc-a922-4328d4a58622 | The ApplicationID / ClientID of the app registration |
DATACONTRACT_AZURE_CLIENT_SECRET | yZK8Q~GWO1MMXXXXXXXXXXXXX | The Client Secret value |
Sqlserver
Data Contract CLI can test data in MS SQL Server (including Azure SQL, Synapse Analytics SQL Pool).
Example
datacontract.yaml
servers:
production:
type: sqlserver
host: localhost
port: 5432
database: tempdb
schema: dbo
driver: ODBC Driver 18 for SQL Server
models:
my_table_1:
type: table
fields:
my_column_1:
type: varchar
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_SQLSERVER_USERNAME | root | Username |
DATACONTRACT_SQLSERVER_PASSWORD | toor | Password |
DATACONTRACT_SQLSERVER_TRUSTED_CONNECTION | True | Use windows authentication, instead of login |
DATACONTRACT_SQLSERVER_TRUST_SERVER_CERTIFICATE | True | Trust self-signed certificate |
DATACONTRACT_SQLSERVER_ENCRYPTED_CONNECTION | True | Use SSL |
Databricks
Works with Unity Catalog and Hive metastore.
Needs a running SQL warehouse or compute cluster.
Example
datacontract.yaml
servers:
production:
type: databricks
host: dbc-abcdefgh-1234.cloud.databricks.com
catalog: acme_catalog_prod
schema: orders_latest
models:
orders:
type: table
fields: ...
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_DATABRICKS_TOKEN | dapia00000000000000000000000000000 | The personal access token to authenticate |
DATACONTRACT_DATABRICKS_HTTP_PATH | /sql/1.0/warehouses/b053a3ffffffff | The HTTP path to the SQL warehouse or compute cluster |
Databricks (programmatic)
Works with Unity Catalog and Hive metastore.
When running in a notebook or pipeline, the provided spark
session can be used.
An additional authentication is not required.
Requires a Databricks Runtime with Python >= 3.10.
Example
datacontract.yaml
servers:
production:
type: databricks
host: dbc-abcdefgh-1234.cloud.databricks.com
catalog: acme_catalog_prod
schema: orders_latest
models:
orders:
type: table
fields: ...
Notebook
%pip install datacontract-cli[databricks]
dbutils.library.restartPython()
from datacontract.data_contract import DataContract
data_contract = DataContract(
data_contract_file="/Volumes/acme_catalog_prod/orders_latest/datacontract/datacontract.yaml",
spark=spark)
run = data_contract.test()
run.result
Dataframe (programmatic)
Works with Spark DataFrames.
DataFrames need to be created as named temporary views.
Multiple temporary views are supported if your data contract contains multiple models.
Testing DataFrames is useful to test your datasets in a pipeline before writing them to a data source.
Example
datacontract.yaml
servers:
production:
type: dataframe
models:
my_table:
type: table
fields: ...
Example code
from datacontract.data_contract import DataContract
df.createOrReplaceTempView("my_table")
data_contract = DataContract(
data_contract_file="datacontract.yaml",
spark=spark,
)
run = data_contract.test()
assert run.result == "passed"
Snowflake
Data Contract CLI can test data in Snowflake.
Example
datacontract.yaml
servers:
snowflake:
type: snowflake
account: abcdefg-xn12345
database: ORDER_DB
schema: ORDERS_PII_V2
models:
my_table_1:
type: table
fields:
my_column_1:
type: varchar
Environment Variables
All parameters supported by Soda, uppercased and prepended by DATACONTRACT_SNOWFLAKE_
prefix.
For example:
Soda parameter | Environment Variable |
---|
username | DATACONTRACT_SNOWFLAKE_USERNAME |
password | DATACONTRACT_SNOWFLAKE_PASSWORD |
warehouse | DATACONTRACT_SNOWFLAKE_WAREHOUSE |
role | DATACONTRACT_SNOWFLAKE_ROLE |
connection_timeout | DATACONTRACT_SNOWFLAKE_CONNECTION_TIMEOUT |
Beware, that parameters:
are obtained from the servers
section of the YAML-file.
E.g. from the example above:
servers:
snowflake:
account: abcdefg-xn12345
database: ORDER_DB
schema: ORDERS_PII_V2
Kafka
Kafka support is currently considered experimental.
Example
datacontract.yaml
servers:
production:
type: kafka
host: abc-12345.eu-central-1.aws.confluent.cloud:9092
topic: my-topic-name
format: json
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_KAFKA_SASL_USERNAME | xxx | The SASL username (key). |
DATACONTRACT_KAFKA_SASL_PASSWORD | xxx | The SASL password (secret). |
DATACONTRACT_KAFKA_SASL_MECHANISM | PLAIN | Default PLAIN . Other supported mechanisms: SCRAM-SHA-256 and SCRAM-SHA-512 |
Postgres
Data Contract CLI can test data in Postgres or Postgres-compliant databases (e.g., RisingWave).
Example
datacontract.yaml
servers:
postgres:
type: postgres
host: localhost
port: 5432
database: postgres
schema: public
models:
my_table_1:
type: table
fields:
my_column_1:
type: varchar
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_POSTGRES_USERNAME | postgres | Username |
DATACONTRACT_POSTGRES_PASSWORD | mysecretpassword | Password |
Trino
Data Contract CLI can test data in Trino.
Example
datacontract.yaml
servers:
trino:
type: trino
host: localhost
port: 8080
catalog: my_catalog
schema: my_schema
models:
my_table_1:
type: table
fields:
my_column_1:
type: varchar
my_column_2:
type: object
config:
trinoType: row(en_us varchar, pt_br varchar)
Environment Variables
Environment Variable | Example | Description |
---|
DATACONTRACT_TRINO_USERNAME | trino | Username |
DATACONTRACT_TRINO_PASSWORD | mysecretpassword | Password |
export
Usage: datacontract export [OPTIONS] [LOCATION]
Convert data contract to a specific format. Saves to file specified by `output` option if present, otherwise prints to stdout.
╭─ Arguments ────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ location [LOCATION] The location (url or path) of the data contract yaml. [default: datacontract.yaml] │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * --format [jsonschema|pydantic-model|sodacl|dbt|dbt-sources|db The export format. [default: None] [required] │
│ t-staging-sql|odcs|odcs_v2|odcs_v3|rdf|avro|protobuf │
│ |great-expectations|terraform|avro-idl|sql|sql-query │
│ |html|go|bigquery|dbml|spark|sqlalchemy|data-caterer │
│ |dcs] │
│ --output PATH Specify the file path where the exported data will be │
│ saved. If no path is provided, the output will be │
│ printed to stdout. │
│ [default: None] │
│ --server TEXT The server name to export. [default: None] │
│ --model TEXT Use the key of the model in the data contract yaml │
│ file to refer to a model, e.g., `orders`, or `all` │
│ for all models (default). │
│ [default: all] │
│ --schema TEXT The location (url or path) of the Data Contract │
│ Specification JSON Schema │
│ [default: │
│ https://datacontract.com/datacontract.schema.json] │
│ --engine TEXT [engine] The engine used for great expection run. │
│ [default: None] │
│ --help Show this message and exit. │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ RDF Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --rdf-base TEXT [rdf] The base URI used to generate the RDF graph. [default: None] │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ SQL Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --sql-server-type TEXT [sql] The server type to determine the sql dialect. By default, it uses 'auto' to automatically │
│ detect the sql dialect via the specified servers in the data contract. │
│ [default: auto] │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
datacontract export --format html --output datacontract.html
Available export options:
Type | Description | Status |
---|
html | Export to HTML | ✅ |
jsonschema | Export to JSON Schema | ✅ |
odcs_v2 | Export to Open Data Contract Standard (ODCS) V2 | ✅ |
odcs_v3 | Export to Open Data Contract Standard (ODCS) V3 | ✅ |
odcs | Export to Open Data Contract Standard (ODCS) V3 | ✅ |
sodacl | Export to SodaCL quality checks in YAML format | ✅ |
dbt | Export to dbt models in YAML format | ✅ |
dbt-sources | Export to dbt sources in YAML format | ✅ |
dbt-staging-sql | Export to dbt staging SQL models | ✅ |
rdf | Export data contract to RDF representation in N3 format | ✅ |
avro | Export to AVRO models | ✅ |
protobuf | Export to Protobuf | ✅ |
terraform | Export to terraform resources | ✅ |
sql | Export to SQL DDL | ✅ |
sql-query | Export to SQL Query | ✅ |
great-expectations | Export to Great Expectations Suites in JSON Format | ✅ |
bigquery | Export to BigQuery Schemas | ✅ |
go | Export to Go types | ✅ |
pydantic-model | Export to pydantic models | ✅ |
DBML | Export to a DBML Diagram description | ✅ |
spark | Export to a Spark StructType | ✅ |
sqlalchemy | Export to SQLAlchemy Models | ✅ |
data-caterer | Export to Data Caterer in YAML format | ✅ |
dcs | Export to Data Contract Specification in YAML format | ✅ |
Missing something? | Please create an issue on GitHub | TBD |
Great Expectations
The export
function transforms a specified data contract into a comprehensive Great Expectations JSON suite.
If the contract includes multiple models, you need to specify the names of the model you wish to export.
datacontract export datacontract.yaml --format great-expectations --model orders
The export creates a list of expectations by utilizing:
- The data from the Model definition with a fixed mapping
- The expectations provided in the quality field for each model (find here the expectations gallery: Great Expectations Gallery)
Additional Arguments
To further customize the export, the following optional arguments are available:
-
suite_name
: The name of the expectation suite. This suite groups all generated expectations and provides a convenient identifier within Great Expectations. If not provided, a default suite name will be generated based on the model name(s).
-
engine
: Specifies the engine used to run Great Expectations checks. Accepted values are:
pandas
— Use this when working with in-memory data frames through the Pandas library.spark
— Use this for working with Spark dataframes.sql
— Use this for working with SQL databases.
-
sql_server_type
: Specifies the type of SQL server to connect with when engine
is set to sql
.
Providing sql_server_type
ensures that the appropriate SQL dialect and connection settings are applied during the expectation validation.
RDF
The export function converts a given data contract into a RDF representation. You have the option to
add a base_url which will be used as the default prefix to resolve relative IRIs inside the document.
datacontract export --format rdf --rdf-base https://www.example.com/ datacontract.yaml
The data contract is mapped onto the following concepts of a yet to be defined Data Contract
Ontology named https://datacontract.com/DataContractSpecification/ :
Having the data contract inside an RDF Graph gives us access the following use cases:
- Interoperability with other data contract specification formats
- Store data contracts inside a knowledge graph
- Enhance a semantic search to find and retrieve data contracts
- Linking model elements to already established ontologies and knowledge
- Using full power of OWL to reason about the graph structure of data contracts
- Apply graph algorithms on multiple data contracts (Find similar data contracts, find "gatekeeper"
data products, find the true domain owner of a field attribute)
DBML
The export function converts the logical data types of the datacontract into the specific ones of a concrete Database
if a server is selected via the --server
option (based on the type
of that server). If no server is selected, the
logical data types are exported.
Spark
The export function converts the data contract specification into a StructType Spark schema. The returned value is a Python code picture of the model schemas.
Spark DataFrame schema is defined as StructType. For more details about Spark Data Types please see the spark documentation
Avro
The export function converts the data contract specification into an avro schema. It supports specifying custom avro properties for logicalTypes and default values.
Custom Avro Properties
We support a config map on field level. A config map may include any additional key-value pairs and support multiple server type bindings.
To specify custom Avro properties in your data contract, you can define them within the config
section of your field definition. Below is an example of how to structure your YAML configuration to include custom Avro properties, such as avroLogicalType
and avroDefault
.
NOTE: At this moment, we just support logicalType and default
Data Caterer
The export function converts the data contract to a data generation task in YAML format that can be
ingested by Data Caterer. This gives you the
ability to generate production-like data in any environment based off your data contract.
datacontract export datacontract.yaml --format data-caterer --model orders
You can further customise the way data is generated via adding
additional metadata in the YAML
to suit your needs.
Example Configuration
models:
orders:
fields:
my_field_1:
description: Example for AVRO with Timestamp (microsecond precision) https://avro.apache.org/docs/current/spec.html#Local+timestamp+%28microsecond+precision%29
type: long
example: 1672534861000000
required: true
config:
avroLogicalType: local-timestamp-micros
avroDefault: 1672534861000000
Explanation
- models: The top-level key that contains different models (tables or objects) in your data contract.
- orders: A specific model name. Replace this with the name of your model.
- fields: The fields within the model. Each field can have various properties defined.
- my_field_1: The name of a specific field. Replace this with your field name.
- description: A textual description of the field.
- type: The data type of the field. In this example, it is
long
. - example: An example value for the field.
- required: Is this a required field (as opposed to optional/nullable).
- config: Section to specify custom Avro properties.
- avroLogicalType: Specifies the logical type of the field in Avro. In this example, it is
local-timestamp-micros
. - avroDefault: Specifies the default value for the field in Avro. In this example, it is 1672534861000000 which corresponds to
2023-01-01 01:01:01 UTC
.
import
Usage: datacontract import [OPTIONS]
Create a data contract from the given source location. Saves to file specified by `output` option if present, otherwise
prints to stdout.
╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * --format [sql|avro|dbt|dbml|glue|jsonschema|bigquery The format of the source file. │
│ |odcs|unity|spark|iceberg|parquet] [default: None] │
│ [required] │
│ --output PATH Specify the file path where the Data │
│ Contract will be saved. If no path is │
│ provided, the output will be printed to │
│ stdout. │
│ [default: None] │
│ --source TEXT The path to the file or Glue Database that │
│ should be imported. │
│ [default: None] │
│ --glue-table TEXT List of table ids to import from the Glue │
│ Database (repeat for multiple table ids, │
│ leave empty for all tables in the dataset). │
│ [default: None] │
│ --bigquery-project TEXT The bigquery project id. [default: None] │
│ --bigquery-dataset TEXT The bigquery dataset id. [default: None] │
│ --bigquery-table TEXT List of table ids to import from the │
│ bigquery API (repeat for multiple table ids, │
│ leave empty for all tables in the dataset). │
│ [default: None] │
│ --unity-table-full-name TEXT Full name of a table in the unity catalog │
│ [default: None] │
│ --dbt-model TEXT List of models names to import from the dbt │
│ manifest file (repeat for multiple models │
│ names, leave empty for all models in the │
│ dataset). │
│ [default: None] │
│ --dbml-schema TEXT List of schema names to import from the DBML │
│ file (repeat for multiple schema names, │
│ leave empty for all tables in the file). │
│ [default: None] │
│ --dbml-table TEXT List of table names to import from the DBML │
│ file (repeat for multiple table names, leave │
│ empty for all tables in the file). │
│ [default: None] │
│ --iceberg-table TEXT Table name to assign to the model created │
│ from the Iceberg schema. │
│ [default: None] │
│ --help Show this message and exit. │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Example:
datacontract import --format sql --source my_ddl.sql
datacontract import --format sql --source my_ddl.sql --output datacontract.yaml
Available import options:
Type | Description | Status |
---|
sql | Import from SQL DDL | ✅ |
avro | Import from AVRO schemas | ✅ |
glue | Import from AWS Glue DataCatalog | ✅ |
jsonschema | Import from JSON Schemas | ✅ |
bigquery | Import from BigQuery Schemas | ✅ |
unity | Import from Databricks Unity Catalog | partial |
dbt | Import from dbt models | ✅ |
odcs | Import from Open Data Contract Standard (ODCS) | ✅ |
spark | Import from Spark StructTypes | ✅ |
dbml | Import from DBML models | ✅ |
protobuf | Import from Protobuf schemas | TBD |
iceberg | Import from an Iceberg JSON Schema Definition | partial |
parquet | Import from Parquet File Metadta | ✅ |
Missing something? | Please create an issue on GitHub | TBD |
ODCS
Import from Open Data Contract Standard (ODCS) v2 or v3.
The importer automatically detects the ODCS version and imports the data contract.
Examples:
datacontract import --format odcs --source my_data_contract.odcs.yaml
BigQuery
BigQuery data can either be imported off of JSON Files generated from the table descriptions or directly from the Bigquery API. In case you want to use JSON Files, specify the source
parameter with a path to the JSON File.
To import from the Bigquery API, you have to omit source
and instead need to provide bigquery-project
and bigquery-dataset
. Additionally you may specify bigquery-table
to enumerate the tables that should be imported. If no tables are given, all available tables of the dataset will be imported.
For providing authentication to the Client, please see the google documentation or the one about authorizing client libraries.
Examples:
datacontract import --format bigquery --source my_bigquery_table.json
datacontract import --format bigquery --bigquery-project <project_id> --bigquery-dataset <dataset_id> --bigquery-table <tableid_1> --bigquery-table <tableid_2> --bigquery-table <tableid_3>
datacontract import --format bigquery --bigquery-project <project_id> --bigquery-dataset <dataset_id>
Unity Catalog
datacontract import --format unity --source my_unity_table.json
export DATABRICKS_IMPORT_INSTANCE="https://xyz.cloud.databricks.com"
export DATABRICKS_IMPORT_ACCESS_TOKEN=<token>
datacontract import --format unity --unity-table-full-name <table_full_name>
dbt
Importing from dbt manifest file.
You may give the dbt-model
parameter to enumerate the tables that should be imported. If no tables are given, all available tables of the database will be imported.
Examples:
datacontract import --format dbt --source <manifest_path> --dbt-model <model_name_1> --dbt-model <model_name_2> --dbt-model <model_name_3>
datacontract import --format dbt --source <manifest_path>
Glue
Importing from Glue reads the necessary Data directly off of the AWS API.
You may give the glue-table
parameter to enumerate the tables that should be imported. If no tables are given, all available tables of the database will be imported.
Examples:
datacontract import --format glue --source <database_name> --glue-table <table_name_1> --glue-table <table_name_2> --glue-table <table_name_3>
datacontract import --format glue --source <database_name>
Spark
Importing from Spark table or view these must be created or accessible in the Spark context. Specify tables list in source
parameter.
Example:
datacontract import --format spark --source "users,orders"
DBML
Importing from DBML Documents.
NOTE: Since DBML does not have strict requirements on the types of columns, this import may create non-valid datacontracts, as not all types of fields can be properly mapped. In this case you will have to adapt the generated document manually.
We also assume, that the description for models and fields is stored in a Note within the DBML model.
You may give the dbml-table
or dbml-schema
parameter to enumerate the tables or schemas that should be imported.
If no tables are given, all available tables of the source will be imported. Likewise, if no schema is given, all schemas are imported.
Examples:
datacontract import --format dbml --source <file_path>
datacontract import --format dbml --source <file_path> --dbml-schema <schema_1> --dbml-schema <schema_2>
datacontract import --format dbml --source <file_path> --dbml-table <table_name_1> --dbml-table <table_name_2>
datacontract import --format dbml --source <file_path> --dbml-table <table_name_1> --dbml-schema <schema_1>
Iceberg
Importing from an Iceberg Table Json Schema Definition. Specify location of json files using the source
parameter.
Examples:
datacontract import --format iceberg --source ./tests/fixtures/iceberg/simple_schema.json --iceberg-table test-table
breaking
Usage: datacontract breaking [OPTIONS] LOCATION_OLD LOCATION_NEW
Identifies breaking changes between data contracts. Prints to stdout.
╭─ Arguments ───────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * location_old TEXT The location (url or path) of the old data contract yaml. [default: None] [required] │
│ * location_new TEXT The location (url or path) of the new data contract yaml. [default: None] [required] │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help Show this message and exit. │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
changelog
Usage: datacontract changelog [OPTIONS] LOCATION_OLD LOCATION_NEW
Generate a changelog between data contracts. Prints to stdout.
╭─ Arguments ───────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * location_old TEXT The location (url or path) of the old data contract yaml. [default: None] [required] │
│ * location_new TEXT The location (url or path) of the new data contract yaml. [default: None] [required] │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help Show this message and exit. │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
diff
Usage: datacontract diff [OPTIONS] LOCATION_OLD LOCATION_NEW
PLACEHOLDER. Currently works as 'changelog' does.
╭─ Arguments ───────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * location_old TEXT The location (url or path) of the old data contract yaml. [default: None] [required] │
│ * location_new TEXT The location (url or path) of the new data contract yaml. [default: None] [required] │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help Show this message and exit. │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
catalog
Usage: datacontract catalog [OPTIONS]
Create an html catalog of data contracts.
╭─ Options ────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --files TEXT Glob pattern for the data contract files to include in the catalog. [default: *.yaml] │
│ --output TEXT Output directory for the catalog html files. [default: catalog/] │
│ --help Show this message and exit. │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Examples:
# create a catalog right in the current folder
datacontract catalog --output "."
# Create a catalog based on a filename convention
datacontract catalog --files "*.odcs.yaml"
Publish
Usage: datacontract publish [OPTIONS] [LOCATION]
Publish the data contract to the Data Mesh Manager.
╭─ Arguments ────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ location [LOCATION] The location (url or path) of the data contract yaml. [default: datacontract.yaml] │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help Show this message and exit. │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Integrations
Integration | Option | Description |
---|
Data Mesh Manager | --publish | Push full results to the Data Mesh Manager API |
Data Contract Manager | --publish | Push full results to the Data Contract Manager API |
OpenTelemetry | --publish-to-opentelemetry | Push result as gauge metrics |
Integration with Data Mesh Manager
If you use Data Mesh Manager or Data Contract Manager, you can use the data contract URL and append the --publish
option to send and display the test results. Set an environment variable for your API key.
$ EXPORT DATAMESH_MANAGER_API_KEY=xxx
$ datacontract test https://demo.datamesh-manager.com/demo279750347121/datacontracts/4df9d6ee-e55d-4088-9598-b635b2fdcbbc/datacontract.yaml \
--server production \
--publish https://api.datamesh-manager.com/api/test-results
Integration with OpenTelemetry
If you use OpenTelemetry, you can use the data contract URL and append the --publish-to-opentelemetry
option to send the test results to your OLTP-compatible instance, e.g., Prometheus.
The metric name is "datacontract.cli.test.result" and it uses the following encoding for the result:
datacontract.cli.test.result | Description |
---|
0 | test run passed, no warnings |
1 | test run has warnings |
2 | test run failed |
3 | test run not possible due to an error |
4 | test status unknown |
$ EXPORT OTEL_SERVICE_NAME=datacontract-cli
$ EXPORT OTEL_EXPORTER_OTLP_ENDPOINT=https://YOUR_ID.apm.westeurope.azure.elastic-cloud.com:443
$ EXPORT OTEL_EXPORTER_OTLP_HEADERS=Authorization=Bearer%20secret
$ EXPORT OTEL_EXPORTER_OTLP_PROTOCOL=http/protobuf
$ datacontract test https://demo.datamesh-manager.com/demo279750347121/datacontracts/4df9d6ee-e55d-4088-9598-b635b2fdcbbc/datacontract.yaml --server production --publish-to-opentelemetry
Current limitations:
- currently, only ConsoleExporter and OTLP Exporter
- Metrics only, no logs yet (but loosely planned)
Best Practices
We share best practices in using the Data Contract CLI.
Data-first Approach
Create a data contract based on the actual data. This is the fastest way to get started and to get feedback from the data consumers.
-
Use an existing physical schema (e.g., SQL DDL) as a starting point to define your logical data model in the contract. Double check right after the import whether the actual data meets the imported logical data model. Just to be sure.
$ datacontract import --format sql --source ddl.sql
$ datacontract test
-
Add examples to the datacontract.yaml
. If you can, use actual data and anonymize. Make sure that the examples match the imported logical data model.
$ datacontract test --examples
-
Add quality checks and additional type constraints one by one to the contract and make sure the examples and the actual data still adheres to the contract. Check against examples for a very fast feedback loop.
$ datacontract test --examples
$ datacontract test
-
Make sure that all the best practices for a datacontract.yaml
are met using the linter. You probably forgot to document some fields and add the terms and conditions.
$ datacontract lint
-
Set up a CI pipeline that executes daily and reports the results to the Data Mesh Manager. Or to some place else. You can even publish to any opentelemetry compatible system.
$ datacontract test --publish https://api.datamesh-manager.com/api/test-results
Contract-First
Create a data contract based on the requirements from use cases.
-
Start with a datacontract.yaml
template.
$ datacontract init
-
Add examples to the datacontract.yaml
. Do not start with the data model, although you are probably tempted to do that. Examples are the fastest way to get feedback from everybody and not loose someone in the discussion.
-
Create the model based on the examples. Test the model against the examples to double-check whether the model matches the examples.
$ datacontract test --examples
-
Add quality checks and additional type constraints one by one to the contract and make sure the examples and the actual data still adheres to the contract. Check against examples for a very fast feedback loop.
$ datacontract test --examples
-
Fill in the terms, descriptions, etc. Make sure you follow all best practices for a datacontract.yaml
using the linter.
$ datacontract lint
-
Set up a CI pipeline that lints and tests the examples so you make sure that any changes later do not decrease the quality of the contract.
$ datacontract lint
$ datacontract test --examples
-
Use the export function to start building the providing data product as well as the integration into the consuming data products.
$ datacontract export --format dbt
$ datacontract export --format dbt-sources
$ datacontract export --format dbt-staging-sql
Schema Evolution
Non-breaking Changes
Examples: adding models or fields
Breaking Changes
Examples: Removing or renaming models and fields.
Customizing Exporters and Importers
Custom Exporter
Using the exporter factory to add a new custom exporter
from datacontract.data_contract import DataContract
from datacontract.export.exporter import Exporter
from datacontract.export.exporter_factory import exporter_factory
class CustomExporter(Exporter):
def export(self, data_contract, model, server, sql_server_type, export_args) -> dict:
result = {
"title": data_contract.info.title,
"version": data_contract.info.version,
"description": data_contract.info.description,
"email": data_contract.info.contact.email,
"url": data_contract.info.contact.url,
"model": model,
"model_columns": ", ".join(list(data_contract.models.get(model).fields.keys())),
"export_args": export_args,
"custom_args": export_args.get("custom_arg", ""),
}
return result
exporter_factory.register_exporter("custom", CustomExporter)
if __name__ == "__main__":
data_contract = DataContract(
data_contract_file="/path/datacontract.yaml"
)
result = data_contract.export(
export_format="custom", model="orders", server="production", custom_arg="my_custom_arg"
)
print(result)
Output
{
'title': 'Orders Unit Test',
'version': '1.0.0',
'description': 'The orders data contract',
'email': 'team-orders@example.com',
'url': 'https://wiki.example.com/teams/checkout',
'model': 'orders',
'model_columns': 'order_id, order_total, order_status',
'export_args': {'server': 'production', 'custom_arg': 'my_custom_arg'},
'custom_args': 'my_custom_arg'
}
Custom Importer
Using the importer factory to add a new custom importer
from datacontract.model.data_contract_specification import DataContractSpecification, Field, Model
from datacontract.data_contract import DataContract
from datacontract.imports.importer import Importer
from datacontract.imports.importer_factory import importer_factory
import json
class CustomImporter(Importer):
def import_source(
self, data_contract_specification: DataContractSpecification, source: str, import_args: dict
) -> dict:
source_dict = json.loads(source)
data_contract_specification.id = source_dict.get("id_custom")
data_contract_specification.info.title = source_dict.get("title")
data_contract_specification.info.version = source_dict.get("version")
data_contract_specification.info.description = source_dict.get("description_from_app")
for model in source_dict.get("models", []):
fields = {}
for column in model.get('columns'):
field = Field(
description=column.get('column_description'),
type=column.get('type')
)
fields[column.get('name')] = field
dc_model = Model(
description=model.get('description'),
fields= fields
)
data_contract_specification.models[model.get('name')] = dc_model
return data_contract_specification
importer_factory.register_importer("custom_company_importer", CustomImporter)
if __name__ == "__main__":
json_from_custom_app = '''
{
"id_custom": "uuid-custom",
"version": "0.0.2",
"title": "my_custom_imported_data",
"description_from_app": "Custom contract description",
"models": [
{
"name": "model1",
"description": "model description from app",
"columns": [
{
"name": "columnA",
"type": "varchar",
"column_description": "my_column description"
},
{
"name": "columnB",
"type": "varchar",
"column_description": "my_columnB description"
}
]
}
]
}
'''
data_contract = DataContract()
result = data_contract.import_from_source(
format="custom_company_importer",
data_contract_specification=DataContract.init(),
source=json_from_custom_app
)
print(result.to_yaml() )
Output
dataContractSpecification: 1.1.0
id: uuid-custom
info:
title: my_custom_imported_data
version: 0.0.2
description: Custom contract description
models:
model1:
fields:
columnA:
type: varchar
description: my_column description
columnB:
type: varchar
description: my_columnB description
Development Setup
Python base interpreter should be 3.11.x (unless working on 3.12 release candidate).
python3 -m venv venv
source venv/bin/activate
pip install --upgrade pip setuptools wheel
pip install -e '.[dev]'
pre-commit install
pre-commit run --all-files
pytest
Docker Build
docker build -t datacontract/cli .
docker run --rm -v ${PWD}:/home/datacontract datacontract/cli
Docker compose integration
We've included a docker-compose.yml configuration to simplify the build, test, and deployment of the image.
Building the Image with Docker Compose
To build the Docker image using Docker Compose, run the following command:
docker compose build
This command utilizes the docker-compose.yml
to build the image, leveraging predefined settings such as the build context and Dockerfile location. This approach streamlines the image creation process, avoiding the need for manual build specifications each time.
Testing the Image
After building the image, you can test it directly with Docker Compose:
docker compose run --rm datacontract --version
This command runs the container momentarily to check the version of the datacontract
CLI. The --rm
flag ensures that the container is automatically removed after the command executes, keeping your environment clean.
Use with pre-commit
To run datacontract-cli
as part of a pre-commit workflow, add something like the below to the repos
list in the project's .pre-commit-config.yaml
:
repos:
- repo: https://github.com/datacontract/datacontract-cli
rev: "v0.10.9"
hooks:
- id: datacontract-lint
- id: datacontract-test
args: ["--server", "production"]
Available Hook IDs
Hook ID | Description | Dependency |
---|
datacontract-lint | Runs the lint subcommand. | Python3 |
datacontract-test | Runs the test subcommand. Please look at | Python3 |
| test section for all available arguments. | |
Release Steps
- Update the version in
pyproject.toml
- Have a look at the
CHANGELOG.md
- Create release commit manually
- Execute
./release
- Wait until GitHub Release is created
- Add the release notes to the GitHub Release
Contribution
We are happy to receive your contributions. Propose your change in an issue or directly create a pull request with your improvements.
Companies using this tool
Related Tools
- Data Contract Manager is a commercial tool to manage data contracts. It contains a web UI, access management, and data governance for a full enterprise data marketplace.
- Data Contract GPT is a custom GPT that can help you write data contracts.
- Data Contract Editor is an editor for Data Contracts, including a live html preview.
- Data Contract Playground allows you to validate and export your data contract to different formats within your browser.
License
MIT License
Credits
Created by Stefan Negele and Jochen Christ.
.github-corner:hover .octo-arm{animation:octocat-wave 560ms ease-in-out}@keyframes octocat-wave{0%,100%{transform:rotate(0)}20%,60%{transform:rotate(-25deg)}40%,80%{transform:rotate(10deg)}}@media (max-width:500px){.github-corner:hover .octo-arm{animation:none}.github-corner .octo-arm{animation:octocat-wave 560ms ease-in-out}}