from metadata.sdk import configure
from metadata.sdk.entities import DataContracts

configure(
    host="https://your-company.getcollate.io/api",
    jwt_token="your-jwt-token"
)

# Create a basic data contract for a table
contract = DataContracts.create(
    name="sales-orders-contract",
    entity={
        "id": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
        "type": "table"
    },
    description="Data contract for the sales orders table",
    entityStatus="Approved",
    schema=[
        {"name": "order_id", "dataType": "INT", "constraint": "PRIMARY_KEY"},
        {"name": "customer_id", "dataType": "INT", "constraint": "NOT_NULL"},
        {"name": "order_date", "dataType": "TIMESTAMP", "constraint": "NOT_NULL"},
        {"name": "total_amount", "dataType": "DECIMAL"}
    ],
    semantics=[
        {"name": "hasOwner"},
        {"name": "hasDescription"},
        {"name": "hasTags"}
    ],
    sla={
        "refreshFrequency": {"interval": 1, "unit": "hour"},
        "availabilityTime": "09:00 UTC"
    }
)
print(f"Created contract: {contract.fullyQualifiedName}")

import static org.openmetadata.sdk.fluent.DataContracts.*;

// client is an initialized OpenMetadataClient.
setDefaultClient(client);

// Create a data contract with fluent API
DataContract contract = create()
    .name("sales-orders-contract")
    .forEntity(new EntityReference()
        .withId(tableId)
        .withType("table"))
    .withDescription("Data contract for the sales orders table")
    .withStatus(EntityStatus.APPROVED)
    .withSchema(List.of(
        new Column().withName("order_id").withDataType(ColumnDataType.INT),
        new Column().withName("customer_id").withDataType(ColumnDataType.INT),
        new Column().withName("order_date").withDataType(ColumnDataType.TIMESTAMP)
    ))
    .withSemanticRule(new SemanticsRule().withName("hasOwner"))
    .withSemanticRule(new SemanticsRule().withName("hasDescription"))
    .execute();

curl -X POST "{base_url}/api/v1/dataContracts" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '{
    "name": "sales-orders-contract",
    "entity": {
      "id": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
      "type": "table"
    },
    "description": "Data contract for the sales orders table",
    "entityStatus": "Approved",
    "schema": [
      {"name": "order_id", "dataType": "INT", "constraint": "PRIMARY_KEY"},
      {"name": "customer_id", "dataType": "INT", "constraint": "NOT_NULL"},
      {"name": "order_date", "dataType": "TIMESTAMP", "constraint": "NOT_NULL"},
      {"name": "total_amount", "dataType": "DECIMAL"}
    ],
    "semantics": [
      {"name": "hasOwner"},
      {"name": "hasDescription"},
      {"name": "hasTags"}
    ],
    "sla": {
      "refreshFrequency": {"interval": 1, "unit": "hour"},
      "availabilityTime": "09:00 UTC"
    }
  }'

{
  "id": "f7a1b2c3-d4e5-6789-0abc-def123456789",
  "name": "sales-orders-contract",
  "fullyQualifiedName": "sample_data.ecommerce_db.shopify.sales_orders.dataContract_sales-orders-contract",
  "displayName": "sales-orders-contract",
  "description": "Data contract for the sales orders table",
  "entityStatus": "Approved",
  "entity": {
    "id": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
    "type": "table",
    "name": "sales_orders",
    "fullyQualifiedName": "sample_data.ecommerce_db.shopify.sales_orders"
  },
  "schema": [
    {"name": "order_id", "dataType": "INT", "constraint": "PRIMARY_KEY"},
    {"name": "customer_id", "dataType": "INT", "constraint": "NOT_NULL"},
    {"name": "order_date", "dataType": "TIMESTAMP", "constraint": "NOT_NULL"},
    {"name": "total_amount", "dataType": "DECIMAL"}
  ],
  "semantics": [
    {"name": "hasOwner"},
    {"name": "hasDescription"},
    {"name": "hasTags"}
  ],
  "sla": {
    "refreshFrequency": {"interval": 1, "unit": "hour"},
    "availabilityTime": "09:00 UTC"
  },
  "version": 0.1,
  "updatedAt": 1769982800000,
  "updatedBy": "admin"
}