Split the channels 2 app into different modules

Author: SmileyChris

graphql_ws/django/consumers.py

@@ -0,0 +1,20 @@
+from channels.generic.websocket import AsyncJsonWebsocketConsumer
+from ..constants import WS_PROTOCOL
+from .subscriptions import subscription_server
+
+
+class GraphQLSubscriptionConsumer(AsyncJsonWebsocketConsumer):
+    async def connect(self):
+        self.connection_context = None
+        if WS_PROTOCOL in self.scope["subprotocols"]:
+            self.connection_context = await subscription_server.handle(self, self.scope)
+            await self.accept(subprotocol=WS_PROTOCOL)
+        else:
+            await self.close()
+
+    async def disconnect(self, code):
+        if self.connection_context:
+            await subscription_server.on_close(self.connection_context)
+
+    async def receive_json(self, content):
+        await subscription_server.on_message(self.connection_context, content)

graphql_ws/django/routing.py

@@ -0,0 +1,19 @@
+from channels.routing import ProtocolTypeRouter, URLRouter
+from django.apps import apps
+from django.urls import path
+from .consumers import GraphQLSubscriptionConsumer
+
+if apps.is_installed("django.contrib.auth"):
+    from channels.auth import AuthMiddlewareStack
+else:
+    AuthMiddlewareStack = None
+
+
+websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer)]
+
+application = ProtocolTypeRouter({"websocket": URLRouter(websocket_urlpatterns)})
+
+if AuthMiddlewareStack:
+    auth_application = ProtocolTypeRouter(
+        {"websocket": AuthMiddlewareStack(URLRouter(websocket_urlpatterns))}
+    )

graphql_ws/django/graphql_channels.py renamed to graphql_ws/django/subscriptions.py

@@ -1,11 +1,29 @@
 from asgiref.sync import async_to_sync
-from channels.generic.websocket import AsyncJsonWebsocketConsumer
 from graphene_django.settings import graphene_settings
 from graphql.execution.executors.asyncio import AsyncioExecutor
-from django.urls import path
 from rx import Observer, Observable
 from ..base import BaseConnectionContext, BaseSubscriptionServer
-from ..constants import GQL_CONNECTION_ACK, GQL_CONNECTION_ERROR, WS_PROTOCOL
+from ..constants import GQL_CONNECTION_ACK, GQL_CONNECTION_ERROR
+
+
+class SubscriptionObserver(Observer):
+    def __init__(
+        self, connection_context, op_id, send_execution_result, send_error, on_close
+    ):
+        self.connection_context = connection_context
+        self.op_id = op_id
+        self.send_execution_result = send_execution_result
+        self.send_error = send_error
+        self.on_close = on_close
+
+    def on_next(self, value):
+        self.send_execution_result(self.connection_context, self.op_id, value)
+
+    def on_completed(self):
+        self.on_close(self.connection_context)
+
+    def on_error(self, error):
+        self.send_error(self.connection_context, self.op_id, error)
 
 
 class ChannelsConnectionContext(BaseConnectionContext):
@@ -87,43 +105,3 @@ async def on_stop(self, connection_context, op_id):
 
 
 subscription_server = ChannelsSubscriptionServer(schema=graphene_settings.SCHEMA)
-
-
-class GraphQLSubscriptionConsumer(AsyncJsonWebsocketConsumer):
-    async def connect(self):
-        self.connection_context = None
-        if WS_PROTOCOL in self.scope["subprotocols"]:
-            self.connection_context = await subscription_server.handle(self, self.scope)
-            await self.accept(subprotocol=WS_PROTOCOL)
-        else:
-            await self.close()
-
-    async def disconnect(self, code):
-        if self.connection_context:
-            await subscription_server.on_close(self.connection_context)
-
-    async def receive_json(self, content):
-        await subscription_server.on_message(self.connection_context, content)
-
-
-class SubscriptionObserver(Observer):
-    def __init__(
-        self, connection_context, op_id, send_execution_result, send_error, on_close
-    ):
-        self.connection_context = connection_context
-        self.op_id = op_id
-        self.send_execution_result = send_execution_result
-        self.send_error = send_error
-        self.on_close = on_close
-
-    def on_next(self, value):
-        self.send_execution_result(self.connection_context, self.op_id, value)
-
-    def on_completed(self):
-        self.on_close(self.connection_context)
-
-    def on_error(self, error):
-        self.send_error(self.connection_context, self.op_id, error)
-
-
-websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer)]