Bring compatibility with Channels v3

graphql_ws/django/routing.py

@@ -1,5 +1,7 @@
+from channels import __version__ as channels_version
 from channels.routing import ProtocolTypeRouter, URLRouter
 from channels.sessions import SessionMiddlewareStack
+from django.utils.version import get_version_tuple
 from django.apps import apps
 from django.urls import path
 from .consumers import GraphQLSubscriptionConsumer
@@ -10,7 +12,13 @@
     AuthMiddlewareStack = None
 
 
-websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer)]
+channels_version_tuple = get_version_tuple(channels_version)
+
+
+if channels_version_tuple > (3, 0 , 0):
+    websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer.as_asgi())]
+else:
+    websocket_urlpatterns = [path("subscriptions", GraphQLSubscriptionConsumer)]
 
 application = ProtocolTypeRouter({"websocket": URLRouter(websocket_urlpatterns)})
 

graphql_ws/django_channels.py

@@ -1,6 +1,10 @@
 import json
 
-from channels.generic.websockets import JsonWebsocketConsumer
+try:
+    # Channels version > 1 renamed the websockets module to websocket.
+    from channels.generic.websockets import JsonWebsocketConsumer
+except ImportError:
+    from channels.generic.websocket import JsonWebsocketConsumer
 from graphene_django.settings import graphene_settings
 
 from .base import BaseConnectionContext

setup.cfg

@@ -57,6 +57,8 @@ test =
     channels==1.*; python_version<"3"
     django==2.*; python_version>="3"
     channels==2.*; python_version>="3"
+    django==3.*; python_version>="3"
+    channels==3.*; python_version>="3"
     aiohttp; python_version>="3.5"
 
 [bdist_wheel]