from metadata.sdk import configure
from metadata.sdk.entities import Tables

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

table_id = "a1b2c3d4-e5f6-7890-abcd-ef1234567890"

# Add join information
Tables.add_joins(table_id, {
    "startDate": "2024-01-01",
    "dayCount": 30,
    "columnJoins": [
        {
            "columnName": "id",
            "joinedWith": [
                {
                    "fullyQualifiedName": "snowflake_prod.analytics.public.orders.customer_id",
                    "joinCount": 1542
                }
            ]
        }
    ]
})

# Set data model
Tables.set_data_model(table_id, {
    "modelType": "DBT",
    "description": "Customer dimension table built from raw events",
    "sql": "SELECT id, name, email, created_at FROM raw.customers WHERE active = true",
    "columns": [
        {"name": "id", "description": "Primary key from source"},
        {"name": "name", "description": "Customer full name"},
        {"name": "email", "description": "Verified email address"},
        {"name": "created_at", "description": "First seen timestamp"}
    ]
})

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

String tableId = "a1b2c3d4-e5f6-7890-abcd-ef1234567890";

// Add join information
Tables.addJoins(tableId, Map.of(
    "startDate", "2024-01-01",
    "dayCount", 30,
    "columnJoins", List.of(
        Map.of(
            "columnName", "id",
            "joinedWith", List.of(
                Map.of(
                    "fullyQualifiedName", "snowflake_prod.analytics.public.orders.customer_id",
                    "joinCount", 1542
                )
            )
        )
    )
));

// Set data model
Tables.setDataModel(tableId, Map.of(
    "modelType", "DBT",
    "description", "Customer dimension table built from raw events",
    "sql", "SELECT id, name, email, created_at FROM raw.customers WHERE active = true"
));

# Add join information
curl -X PUT "{base_url}/api/v1/tables/455e3d9d-dbbf-455e-b3be-7191daa825f3/joins" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '{
    "startDate": "2024-01-01",
    "dayCount": 30,
    "columnJoins": [
      {
        "columnName": "agent_id",
        "joinedWith": [
          {
            "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_agent.agent_id",
            "joinCount": 1542
          }
        ]
      }
    ]
  }'

# Set data model
curl -X PUT "{base_url}/api/v1/tables/455e3d9d-dbbf-455e-b3be-7191daa825f3/dataModel" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '{
    "modelType": "DBT",
    "description": "Agent performance summary built from utilization detail tables",
    "sql": "SELECT agent_id, AVG(score) as performance_score FROM raw.agent_scores GROUP BY agent_id"
  }'

# Get columns with pagination
curl "{base_url}/api/v1/tables/455e3d9d-dbbf-455e-b3be-7191daa825f3/columns?limit=10&offset=0" \
  -H "Authorization: Bearer {access_token}"

{
  "startDate": "2024-01-01",
  "dayCount": 30,
  "columnJoins": [
    {
      "columnName": "agent_id",
      "joinedWith": [
        {
          "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_agent.agent_id",
          "joinCount": 1542
        }
      ]
    }
  ]
}