Add silent option to %%graph_notebook_config

ChangeLog.md

@@ -4,6 +4,7 @@ Starting with v1.31.6, this file will contain a record of major features and upd
 
 ## Upcoming
 - Added `--connected-table` option to magics with table widget output ([Link to PR](https://github.com/aws/graph-notebook/pull/634))
+- Added `--silent` option to the `%%graph_notebook_config` line and cell magics ([Link to PR](https://github.com/aws/graph-notebook/pull/641))
 - Changed `%%gremlin --store-to` to also store exceptions from non-Neptune queries ([Link to PR](https://github.com/aws/graph-notebook/pull/635))
 - Fixed broken `--help` option for `%%gremlin` ([Link to PR](https://github.com/aws/graph-notebook/pull/630))
 - Fixed openCypher query bug regression in the [`01-About-the-Neptune-Notebook`](https://github.com/aws/graph-notebook/blob/main/src/graph_notebook/notebooks/01-Getting-Started/01-About-the-Neptune-Notebook.ipynb) sample ([Link to PR](https://github.com/aws/graph-notebook/pull/631))

src/graph_notebook/magics/graph_magic.py

@@ -451,30 +451,35 @@ def graph_notebook_config(self, line='', cell='', local_ns: dict = None):
         parser.add_argument('mode', nargs='?', default='show',
                             help='mode (default=show) [show|reset|silent]')
         parser.add_argument('--store-to', type=str, default='', help='store query result to this variable')
+        parser.add_argument('--silent', action='store_true', default=False, help="Display no output.")
         args = parser.parse_args(line.split())
 
         if cell != '':
             data = json.loads(cell)
             config = get_config_from_dict(data, neptune_hosts=self.neptune_cfg_allowlist)
             self.graph_notebook_config = config
             self._generate_client_from_config(config)
-            print('set notebook config to:')
-            print(json.dumps(self.graph_notebook_config.to_dict(), indent=4))
+            if not args.silent:
+                print('set notebook config to:')
+                print(json.dumps(self.graph_notebook_config.to_dict(), indent=4))
         elif args.mode == 'reset':
             self.graph_notebook_config = get_config(self.config_location, neptune_hosts=self.neptune_cfg_allowlist)
-            print('reset notebook config to:')
-            print(json.dumps(self.graph_notebook_config.to_dict(), indent=4))
+            if not args.silent:
+                print('reset notebook config to:')
+                print(json.dumps(self.graph_notebook_config.to_dict(), indent=4))
         elif args.mode == 'silent':
             """
             silent option to that our neptune_menu extension can receive json instead
             of python Configuration object
             """
             config_dict = self.graph_notebook_config.to_dict()
             store_to_ns(args.store_to, json.dumps(config_dict, indent=4), local_ns)
-            return print(json.dumps(config_dict, indent=4))
+            if not args.silent:
+                return print(json.dumps(config_dict, indent=4))
         else:
             config_dict = self.graph_notebook_config.to_dict()
-            print(json.dumps(config_dict, indent=4))
+            if not args.silent:
+                print(json.dumps(config_dict, indent=4))
 
         store_to_ns(args.store_to, json.dumps(self.graph_notebook_config.to_dict(), indent=4), local_ns)