from metadata.generated.schema.type.entityReference import EntityReference
from metadata.sdk import configure
from metadata.sdk.entities import DataProducts

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

# Identify the data product by its name, which is also its fully qualified name, and each asset by its UUID.

# Add input ports
DataProducts.add_input_ports(
    "CustomerInsights",
    [EntityReference(id="table-uuid-1", type="table")]
)

# Remove input ports
DataProducts.remove_input_ports(
    "CustomerInsights",
    [EntityReference(id="table-uuid-1", type="table")]
)

# Add output ports
DataProducts.add_output_ports(
    "CustomerInsights",
    [EntityReference(id="dashboard-uuid-1", type="dashboard")]
)

# Remove output ports
DataProducts.remove_output_ports(
    "CustomerInsights",
    [EntityReference(id="dashboard-uuid-1", type="dashboard")]
)

# Add assets
DataProducts.add_assets(
    "CustomerInsights",
    [EntityReference(id="table-uuid-2", type="table")]
)

# Remove assets
DataProducts.remove_assets(
    "CustomerInsights",
    [EntityReference(id="table-uuid-2", type="table")]
)

import java.util.List;
import java.util.UUID;
import org.openmetadata.schema.type.EntityReference;
import org.openmetadata.schema.type.api.BulkAssets;

// client is an initialized OpenMetadataClient.
var dataProduct = "CustomerInsights";

var tables = new BulkAssets().withAssets(List.of(
    new EntityReference()
        .withId(UUID.fromString("5b3d9f0e-1c7a-4d2e-9a61-3f8e2b7c4d10"))
        .withType("table")));

// Identify the data product by its name, which is also its fully qualified name, and each asset by its UUID.

// Add assets
var added = client.dataProducts().bulkAddAssets(dataProduct, tables);
System.out.println(added.getStatus());

// Remove assets
var removed = client.dataProducts().bulkRemoveAssets(dataProduct, tables);

// Add and remove input ports
client.dataProducts().bulkAddInputPorts(dataProduct, tables);
client.dataProducts().bulkRemoveInputPorts(dataProduct, tables);

// Add and remove output ports
client.dataProducts().bulkAddOutputPorts(dataProduct, tables);
client.dataProducts().bulkRemoveOutputPorts(dataProduct, tables);

# The path identifier is the data product's fully qualified name. URL-encode it if it contains spaces.

# Add assets
curl -X PUT "{base_url}/api/v1/dataProducts/CustomerInsights/assets/add" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '{
    "assets": [
      {"id": "table-uuid-1", "type": "table"},
      {"id": "topic-uuid-1", "type": "topic"}
    ]
  }'

# Remove assets
curl -X PUT "{base_url}/api/v1/dataProducts/CustomerInsights/assets/remove" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '{
    "assets": [
      {"id": "table-uuid-1", "type": "table"}
    ]
  }'

{
  "id": "b1839f98-f046-5f35-bdef-20f9b3395cb2",
  "name": "CustomerInsights",
  "fullyQualifiedName": "CustomerInsights",
  "description": "Curated customer analytics data product",
  "version": 0.3,
  "updatedAt": 1769984330261,
  "updatedBy": "admin",
  "href": "http://localhost:8585/api/v1/dataProducts/b1839f98-f046-5f35-bdef-20f9b3395cb2",
  "domains": [
    {
      "id": "a0729e98-e946-4e25-acde-10e8a2294ba1",
      "type": "domain",
      "name": "Marketing",
      "fullyQualifiedName": "Marketing",
      "deleted": false
    }
  ],
  "deleted": false,
  "owners": [],
  "experts": []
}