Update to soda contracts v3

Author: andrewrjones

checks.py

@@ -1,13 +1,11 @@
 #!/usr/bin/env python
 
-from soda.contracts.connection import Connection, SodaException
-from soda.contracts.contract import Contract, ContractResult
+from soda.contracts.contract_verification import ContractVerification, ContractVerificationResult
 
-try:
- with Connection.from_yaml_file("./connection.yml") as connection:
- contract: Contract = Contract.from_yaml_file("./customers.contract.yml")
- contract_result: ContractResult = contract.verify(connection)
- 
- print("Soda checks passed")
-except SodaException as e:
- print(e.contract_result)
+contract_verification_result: ContractVerificationResult = (
+ ContractVerification.builder()
+ .with_contract_yaml_file('./customers.contract.yml')
+ .with_warehouse_yaml_file('./connection.yml')
+ .execute()
+ .assert_ok()
+)

connection.yml

@@ -1,5 +1,7 @@
- type: postgres
- host: localhost
- database: postgres
- username: postgres
- password: secret
+name: local_postgres
+type: postgres
+connection:
+ host: localhost
+ database: postgres
+ username: postgres
+ password: secret

customers.contract.yml

@@ -1,5 +1,6 @@
 dataset: customers
-owner: product-team@data-contracts.com
+owner:
+ email: product-team@data-contracts.com
 description: All active customers of our product.
 version: 1
 
@@ -23,3 +24,9 @@ columns:
  - name: distance
  description: The distance the customer is from our shop
  data_type: INTEGER
+
+checks:
+- type: metric_expression
+ metric: retention_period
+ expression_sql: COUNT(CASE WHEN created < NOW() - interval '1 year' THEN 1 END)
+ must_be: 0

lib/data_contract.py

@@ -40,7 +40,7 @@ def to_datacontract_cli(contract):
  title = contract["dataset"],
  version = str(contract["version"]),
  description = contract["description"],
- owner = contract["owner"],
+ owner = contract["owner"]["email"],
  ),
  servers = dict(
  workshop = dict(

main.py

@@ -6,12 +6,10 @@
 from data_contract import *
 
 import yaml
-from soda.contracts.contract import Contract
 
 with open("./customers.contract.yml", "r") as stream:
  contents = stream.read()
  contract = yaml.safe_load(contents)
- soda_contract = Contract.from_yaml_str(contents)
 
 print(
  f'Successfully parsed the `{contract["dataset"]}` contract, which is owned by `{contract["owner"]}`.')

requirements.txt

@@ -1,4 +1,4 @@
 pyyaml==6.0
-soda-core==3.3.2
-soda-core-contracts==3.3.2
-soda-core-postgres==3.3.2
+soda-core==3.3.5
+soda-core-contracts==3.3.5
+soda-core-postgres==3.3.5