Add get_context() to prepare a context object

Author: cpd67

channels_graphql_ws/graphql_ws_consumer.py

@@ -535,9 +535,7 @@ async def _on_gql_start(self, op_id, payload):
             variables = payload.get("variables", {})
 
             # Prepare a context object.
-            context = DictAsObject({})
-            context.channels_scope = self.scope
-            context.channel_name = self.channel_name
+            context = self.get_context()
             context.graphql_operation_name = op_name
             context.graphql_operation_id = op_id
 
@@ -1230,6 +1228,16 @@ def _spawn_background_task(self, awaitable):
         self._background_tasks.add(background_task)
         return background_task
 
+    def get_context(self):
+        """
+        Prepare the context object to use during GraphQL execution.
+        """
+        context = DictAsObject({})
+        context.channels_scope = self.scope
+        context.channel_name = self.channel_name
+        return context
+
+
     @property
     def _channel_layer(self):
         """Channel layer."""