Various improvements on the django path

Author: syrusakbary

src/graphql_server/channels/handlers/http_handler.py

@@ -244,17 +244,17 @@ class GraphQLHTTPConsumer(
     To use this, place it in your ProtocolTypeRouter for your channels project:
 
     ```
-    from graphql_server.channels import GraphQLHttpRouter
+    from graphql_server.channels import GraphQLHTTPConsumer, GraphQLWSConsumer
     from channels.routing import ProtocolTypeRouter
     from django.core.asgi import get_asgi_application
 
     application = ProtocolTypeRouter({
         "http": URLRouter([
-            re_path("^graphql", GraphQLHTTPRouter(schema=schema)),
+            re_path("^graphql", GraphQLHTTPConsumer.as_asgi(schema=schema)),
             re_path("^", get_asgi_application()),
         ]),
         "websocket": URLRouter([
-            re_path("^ws/graphql", GraphQLWebSocketRouter(schema=schema)),
+            re_path("^ws/graphql", GraphQLWSConsumer.as_asgi(schema=schema)),
         ]),
     })
     ```

src/graphql_server/channels/router.py

@@ -49,8 +49,17 @@ def __init__(
         schema: GraphQLSchema,
         django_application: Optional[str] = None,
         url_pattern: str = "^graphql",
+        http_consumer_class: type[GraphQLHTTPConsumer] = GraphQLHTTPConsumer,
+        ws_consumer_class: type[GraphQLWSConsumer] = GraphQLWSConsumer,
+        *args,
+        **kwargs,
     ) -> None:
-        http_urls = [re_path(url_pattern, GraphQLHTTPConsumer.as_asgi(schema=schema))]
+        http_urls = [
+            re_path(
+                url_pattern,
+                http_consumer_class.as_asgi(*args, **kwargs),
+            )
+        ]
         if django_application is not None:
             http_urls.append(re_path("^", django_application))
 
@@ -59,7 +68,10 @@ def __init__(
                 "http": URLRouter(http_urls),
                 "websocket": URLRouter(
                     [
-                        re_path(url_pattern, GraphQLWSConsumer.as_asgi(schema=schema)),
+                        re_path(
+                            url_pattern,
+                            ws_consumer_class.as_asgi(*args, **kwargs),
+                        ),
                     ]
                 ),
             }

src/graphql_server/django/__init__.py

@@ -0,0 +1,3 @@
+from .views import GraphQLView, AsyncGraphQLView
+
+__all__ = ["GraphQLView", "AsyncGraphQLView"]