Enable/disable preview for all the datasets when publishing Kedro-Viz from UI

package/kedro_viz/api/rest/requests.py

@@ -7,5 +7,6 @@ class DeployerConfiguration(BaseModel):
     """Credentials for Deployers."""
 
     platform: str
+    preview: bool
     endpoint: str
     bucket_name: str

package/kedro_viz/api/rest/responses.py

@@ -320,7 +320,7 @@ def get_default_response() -> GraphAPIResponse:
     )
 
 
-def get_node_metadata_response(node_id: str):
+def get_node_metadata_response(node_id: str, preview: bool = True):
     """API response for `/api/nodes/node_id`."""
     node = data_access_manager.nodes.get_node_by_id(node_id)
     if not node:
@@ -333,7 +333,7 @@ def get_node_metadata_response(node_id: str):
         return TaskNodeMetadata(task_node=node)
 
     if isinstance(node, DataNode):
-        return DataNodeMetadata(data_node=node)
+        return DataNodeMetadata(data_node=node, is_preview_enabled_for_all_nodes=preview)
 
     if isinstance(node, TranscodedDataNode):
         return TranscodedDataNodeMetadata(transcoded_data_node=node)
@@ -419,12 +419,12 @@ def save_api_main_response_to_fs(main_path: str, remote_fs: Any):
         raise exc
 
 
-def save_api_node_response_to_fs(nodes_path: str, remote_fs: Any):
+def save_api_node_response_to_fs(nodes_path: str, remote_fs: Any, preview: bool):
     """Saves API /nodes/{node} response to a directory."""
     for nodeId in data_access_manager.nodes.get_node_ids():
         try:
             write_api_response_to_fs(
-                f"{nodes_path}/{nodeId}", get_node_metadata_response(nodeId), remote_fs
+                f"{nodes_path}/{nodeId}", get_node_metadata_response(nodeId, preview), remote_fs
             )
         except Exception as exc:  # pragma: no cover
             logger.exception(
@@ -451,7 +451,7 @@ def save_api_pipeline_response_to_fs(pipelines_path: str, remote_fs: Any):
             raise exc
 
 
-def save_api_responses_to_fs(path: str, remote_fs: Any):
+def save_api_responses_to_fs(path: str, remote_fs: Any, preview: bool):
     """Saves all Kedro Viz API responses to a directory."""
     try:
         logger.debug(
@@ -469,7 +469,7 @@ def save_api_responses_to_fs(path: str, remote_fs: Any):
             remote_fs.makedirs(pipelines_path, exist_ok=True)
 
         save_api_main_response_to_fs(main_path, remote_fs)
-        save_api_node_response_to_fs(nodes_path, remote_fs)
+        save_api_node_response_to_fs(nodes_path, remote_fs, preview)
         save_api_pipeline_response_to_fs(pipelines_path, remote_fs)
 
     except Exception as exc:  # pragma: no cover

package/kedro_viz/api/rest/router.py

@@ -61,7 +61,7 @@ async def get_single_pipeline_data(registered_pipeline_id: str):
 async def deploy_kedro_viz(input_values: DeployerConfiguration):
     try:
         deployer = DeployerFactory.create_deployer(
-            input_values.platform, input_values.endpoint, input_values.bucket_name
+            input_values.platform, input_values.preview,input_values.endpoint, input_values.bucket_name
         )
         deployer.deploy()
         response = {

package/kedro_viz/integrations/deployment/aws_deployer.py

@@ -22,13 +22,14 @@ class AWSDeployer(BaseDeployer):
         _fs (fsspec.filesystem): Filesystem for S3 protocol.
     """
 
-    def __init__(self, endpoint, bucket_name):
+    def __init__(self, preview, endpoint, bucket_name):
         """Initialize S3Deployer with endpoint and bucket name.
         Args:
             endpoint (str): AWS endpoint of the hosted site.
             bucket_name (str): Name of the S3 bucket.
         """
         super().__init__()
+        self.preview = preview
         self._endpoint = endpoint
         self._bucket_name = bucket_name
         self._path = f"{_S3_PROTOCOL}://{bucket_name}"

package/kedro_viz/integrations/deployment/azure_deployer.py

@@ -30,14 +30,15 @@ class AzureDeployer(BaseDeployer):
         _fs (fsspec.filesystem): Filesystem for Azure protocol.
     """
 
-    def __init__(self, endpoint, bucket_name):
+    def __init__(self, preview, endpoint, bucket_name):
         """Initialize AzureBlobStorage with endpoint and bucket name.
 
         Args:
             endpoint (str): Azure endpoint of the hosted site.
             bucket_name (str): Name of the AzureBlobStorage account.
         """
         super().__init__()
+        self.preview = preview
         self._endpoint = endpoint
         self._bucket_name = bucket_name
         self._path = f"{_AZ_PROTOCOL}://$web"

package/kedro_viz/integrations/deployment/base_deployer.py

@@ -30,12 +30,13 @@ class BaseDeployer(abc.ABC):
     """
 
     def __init__(self):
+        self.preview = False
         self._path = None
         self._fs = None
 
     def _upload_api_responses(self):
         """Write API responses to the build."""
-        save_api_responses_to_fs(self._path, self._fs)
+        save_api_responses_to_fs(self._path, self._fs, self.preview)
 
     def _ingest_heap_analytics(self):
         """Ingest heap analytics to index file in the build."""

package/kedro_viz/integrations/deployment/deployer_factory.py

@@ -12,15 +12,15 @@ class DeployerFactory:
     """A class to handle creation of deployer class instances."""
 
     @staticmethod
-    def create_deployer(platform, endpoint=None, bucket_name=None):
+    def create_deployer(platform, preview, endpoint=None, bucket_name=None):
         """Instantiate Kedro-viz deployer classes"""
         platform_name = platform.lower()
         if platform_name == "aws":
-            return AWSDeployer(endpoint, bucket_name)
+            return AWSDeployer(preview, endpoint, bucket_name)
         if platform_name == "azure":
-            return AzureDeployer(endpoint, bucket_name)
+            return AzureDeployer(preview, endpoint, bucket_name)
         if platform_name == "gcp":
-            return GCPDeployer(endpoint, bucket_name)
+            return GCPDeployer(preview, endpoint, bucket_name)
         if platform_name == "local":
             return LocalDeployer()
         raise ValueError(

package/kedro_viz/models/flowchart.py

@@ -769,6 +769,17 @@ class DataNodeMetadata(GraphNodeMetadata):
         description="The statistics for the data node.",
     )
 
+    @model_validator(mode="before")
+    @classmethod
+    def check_is_preview_enabled_for_all_nodes_exists(cls, values):
+        assert "is_preview_enabled_for_all_nodes" in values
+        cls.set_is_preview_enabled_for_all_nodes(values["is_preview_enabled_for_all_nodes"])
+        return values
+
+    @classmethod
+    def set_is_preview_enabled_for_all_nodes(cls, is_preview_enabled_for_all_nodes):
+        cls.is_preview_enabled_for_all_nodes = is_preview_enabled_for_all_nodes    
+
     @model_validator(mode="before")
     @classmethod
     def check_data_node_exists(cls, values):
@@ -806,6 +817,9 @@ def set_run_command(cls, _):
     @field_validator("preview")
     @classmethod
     def set_preview(cls, _):
+        if not cls.is_preview_enabled_for_all_nodes:
+            return None
+
         if cls.data_node.is_preview_disabled() or not hasattr(cls.dataset, "preview"):
             return None
 
@@ -829,6 +843,9 @@ def set_preview(cls, _):
     @field_validator("preview_type")
     @classmethod
     def set_preview_type(cls, _):
+        if not cls.is_preview_enabled_for_all_nodes:
+            return None
+
         if cls.data_node.is_preview_disabled() or not hasattr(cls.dataset, "preview"):
             return None
 

src/components/shareable-url-modal/shareable-url-modal.js

@@ -104,7 +104,7 @@ const ShareableUrlModal = ({ onToggleModal, visible }) => {
           'Content-Type': 'application/json',
         },
         method: 'POST',
-        body: JSON.stringify(inputValues),
+        body: JSON.stringify({ ...inputValues, preview: false }), // preview appended to the body to disable the preview data
       });
       const response = await request.json();