# In this example, we're copying the list of dashboard IDs that are in `dashboards_to_copy`
# from the "dev_sdk" and "source_workspace_id" into the "test_sdk" and "test_workspace_id".
# The big assumption is that the LDMs and analytics models are exactly the same between
# "dev" and "test" apart from these dashboards and their insights -- specifically, that the
# two environments contain the same users, metrics, attributes, etc. -- and also that all of the
# IDs are the same between the two environments.
#
# TODO: Publish to the prod connection / workspace instead of the test connection / workspace
# (test is just for, well, testing)
# TODO: Check to see if objects exist in the dest_analytics_model before appending them;
# if they do exist, replace them instead of appending.
# TODO: All sorts of error checking, including trapping errors where the assumption that
# everything else already exists is false.

dest_analytics_model = test_sdk.catalog_workspace_content.get_declarative_analytics_model(
    workspace_id = test_workspace_id
)
source_analytics_model = dev_sdk.catalog_workspace_content.get_declarative_analytics_model(
    workspace_id = source_workspace_id
)

source_visualizations = source_analytics_model.analytics.visualization_objects
source_dashboards = source_analytics_model.analytics.analytical_dashboards
source_filter_contexts = source_analytics_model.analytics.filter_contexts

for dashboard_id in dashboards_to_copy:
    insights_to_copy = []
    
    # Get details about the dashboard
    dashboard = dev_sdk.client.entities_api.get_entity_analytical_dashboards(
        workspace_id=source_workspace_id,
        object_id=dashboard_id
    )

    # Append the dashboard's filter context to the new analytics model
    # TODO: Make sure it doesn't already exist! If it does, update it.
    filter_context_id = dashboard.data.attributes.content['filterContextRef']['identifier']['id']
    for filter_context in source_filter_contexts:
        if filter_context.id == filter_context_id:
            dest_analytics_model.analytics.filter_contexts.append(filter_context)

    # For each section in the dashboard, find the insights and add their IDs to our list
    # of insight IDs to copy
    for section in dashboard.data.attributes.content['layout']['sections']:
        for item in section['items']:
            if item['widget']['type'] == 'insight':
                insights_to_copy.append(item['widget']['insight']['identifier']['id'])
    
    # Now that we have a list of insight IDs to copy, find their full objects in the
    # source visualizations and append to the new analytics model
    # TODO: Make sure they don't already exist! If they do, update them.
    for viz in source_visualizations:
        if viz.id in insights_to_copy:
            dest_analytics_model.analytics.visualization_objects.append(viz)

# Finally, now that we've copied over insights and filter contexts,
# append the dashboards themselves to the new analytics model
# TODO: Make sure they don't already exist! If they do, update them.
for dashboard in source_dashboards:
    if dashboard.id in dashboards_to_copy:
        dest_analytics_model.analytics.analytical_dashboards.append(dashboard)

# And publish the new replacement analytics model
test_sdk.catalog_workspace_content.put_declarative_analytics_model(
    workspace_id = test_workspace_id,
    analytics_model = dest_analytics_model
)