Tidy up django_channels (1) backend and example

Author: SmileyChris

examples/django_subscriptions/django_subscriptions/settings.py

@@ -118,8 +118,7 @@
 ]
 CHANNEL_LAYERS = {
     "default": {
-        "BACKEND": "asgi_redis.RedisChannelLayer",
-        "CONFIG": {"hosts": [("localhost", 6379)]},
+        "BACKEND": "asgiref.inmemory.ChannelLayer",
         "ROUTING": "django_subscriptions.urls.channel_routing",
     },
 }

examples/django_subscriptions/requirements.txt

@@ -0,0 +1,4 @@
+-e ../..
+django<2
+channels<2
+graphene_django<3

graphql_ws/django_channels.py

@@ -8,39 +8,40 @@
 
 
 class DjangoChannelConnectionContext(BaseConnectionContext):
-    def __init__(self, message, request_context=None):
-        self.message = message
-        self.operations = {}
-        self.request_context = request_context
+    def __init__(self, message):
+        super(DjangoChannelConnectionContext, self).__init__(
+            message.reply_channel,
+            request_context={"user": message.user, "session": message.http_session},
+        )
 
     def send(self, data):
-        self.message.reply_channel.send({"text": json.dumps(data)})
+        self.ws.send({"text": json.dumps(data)})
 
     def close(self, reason):
         data = {"close": True, "text": reason}
-        self.message.reply_channel.send(data)
+        self.ws.send(data)
 
 
 class DjangoChannelSubscriptionServer(BaseSyncSubscriptionServer):
     def handle(self, message, connection_context):
         self.on_message(connection_context, message)
 
 
+subscription_server = DjangoChannelSubscriptionServer(graphene_settings.SCHEMA)
+
+
 class GraphQLSubscriptionConsumer(JsonWebsocketConsumer):
     http_user_and_session = True
     strict_ordering = True
 
-    def connect(self, message, **_kwargs):
+    def connect(self, message, **kwargs):
         message.reply_channel.send({"accept": True})
 
-    def receive(self, content, **_kwargs):
+    def receive(self, content, **kwargs):
         """
         Called when a message is received with either text or bytes
         filled out.
         """
-        self.connection_context = DjangoChannelConnectionContext(self.message)
-        self.subscription_server = DjangoChannelSubscriptionServer(
-            graphene_settings.SCHEMA
-        )
-        self.subscription_server.on_open(self.connection_context)
-        self.subscription_server.handle(content, self.connection_context)
+        context = DjangoChannelConnectionContext(self.message)
+        subscription_server.on_open(context)
+        subscription_server.handle(content, context)