Add @neptune_graph_only decorator

ChangeLog.md

@@ -4,6 +4,7 @@ Starting with v1.31.6, this file will contain a record of major features and upd
 
 ## Upcoming
 - Updated `create-graph` CLI commands in Neptune Analytics samples ([Link to PR](https://github.com/aws/graph-notebook/pull/565))
+- Added `@neptune_graph_only` magics decorator ([Link to PR](https://github.com/aws/graph-notebook/pull/569))
 
 ## Release 4.1.0 (February 1, 2024)
 - New Neptune Analytics notebook - Vector Similarity Algorithms ([Link to PR](https://github.com/aws/graph-notebook/pull/555))

src/graph_notebook/decorators/decorators.py

@@ -12,7 +12,7 @@
 
 import ipywidgets as widgets
 from graph_notebook.visualization.template_retriever import retrieve_template
-from graph_notebook.neptune.client import NEPTUNE_ANALYTICS_SERVICE_NAME
+from graph_notebook.neptune.client import NEPTUNE_ANALYTICS_SERVICE_NAME, NEPTUNE_DB_SERVICE_NAME
 from gremlin_python.driver.protocol import GremlinServerError
 from requests import HTTPError
 
@@ -151,6 +151,23 @@ def check_neptune_db(*args, **kwargs):
     return check_neptune_db
 
 
+def neptune_graph_only(func):
+    @functools.wraps(func)
+    def check_neptune_graph(*args, **kwargs):
+        self = args[0]
+        if not hasattr(self.graph_notebook_config, 'neptune_service'):
+            return func(*args, **kwargs)
+        else:
+            service_type = self.graph_notebook_config.neptune_service
+            if service_type == NEPTUNE_DB_SERVICE_NAME:
+                print(f'This magic is unavailable for Neptune DB.')
+                return
+            else:
+                return func(*args, **kwargs)
+
+    return check_neptune_graph
+
+
 def http_ex_to_html(http_ex: HTTPError):
     try:
         error = json.loads(http_ex.response.content.decode('utf-8'))