_handle_nglview_custom_msg

nglviewer · Nov 28, 2024 · ac6fbae · ac6fbae
1 parent 8dfd35c
commit ac6fbae
Showing 1 changed file with 15 additions and 18 deletions.
diff --git a/nglview/widget.py b/nglview/widget.py
@@ -971,24 +971,21 @@ def _handle_nglview_custom_msg(self, _, msg, buffers):
         self._ngl_msg = msg
 
         msg_type = self._ngl_msg.get('type')
-        if msg_type == 'request_frame':
-            self._handle_request_frame()
-        elif msg_type == 'updateIDs':
-            self._handle_update_ids()
-        elif msg_type == 'removeComponent':
-            self._handle_remove_component()
-        elif msg_type == 'repr_parameters':
-            self._handle_repr_parameters()
-        elif msg_type == 'request_loaded':
-            self._handle_request_loaded()
-        elif msg_type == 'request_repr_dict':
-            self._handle_request_repr_dict()
-        elif msg_type == 'stage_parameters':
-            self._handle_stage_parameters()
-        elif msg_type == 'async_message':
-            self._handle_async_message()
-        elif msg_type == 'image_data':
-            self._handle_image_data()
+        handlers = {
+            'request_frame': self._handle_request_frame,
+            'updateIDs': self._handle_update_ids,
+            'removeComponent': self._handle_remove_component,
+            'repr_parameters': self._handle_repr_parameters,
+            'request_loaded': self._handle_request_loaded,
+            'request_repr_dict': self._handle_request_repr_dict,
+            'stage_parameters': self._handle_stage_parameters,
+            'async_message': self._handle_async_message,
+            'image_data': self._handle_image_data,
+        }
+
+        handler = handlers.get(msg_type)
+        if handler:
+            handler()
 
     def _request_repr_parameters(self, component=0, repr_index=0):
         if self.n_components > 0: