Bring changes from #74

fabienheureux · fabienheureux · commit a360cf689b92 · 2021-08-20T09:24:41.000+02:00
diff --git a/graphql_ws/django/routing.py b/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)})
 
diff --git a/setup.cfg b/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]
