Add django+subscriptions example

Author: patrick91

django-subscriptions/Makefile

@@ -0,0 +1,2 @@
+run:
+	poetry run uvicorn demo.asgi:application --reload --debug

django-subscriptions/api/__init__.py

@@ -0,0 +1,5 @@
+from strawberry.asgi import GraphQL
+
+
+class MyGraphQL(GraphQL):
+    ...

django-subscriptions/api/asgi.py

@@ -0,0 +1,22 @@
+import asyncio
+
+import strawberry
+
+
+@strawberry.type
+class Query:
+    @strawberry.field
+    def hello() -> str:
+        return "world"
+
+
+@strawberry.type
+class Subscription:
+    @strawberry.subscription
+    async def count(self, target: int = 100) -> int:
+        for i in range(target):
+            yield i
+            await asyncio.sleep(0.5)
+
+
+schema = strawberry.Schema(query=Query, subscription=Subscription)

django-subscriptions/api/schema.py

@@ -0,0 +1,6 @@
+from django.urls import path
+
+from .schema import schema
+from .views import AsyncGraphQLView
+
+urlpatterns = [path("graphql", AsyncGraphQLView.as_view(schema=schema))]

django-subscriptions/api/urls.py

@@ -0,0 +1,37 @@
+import os
+
+import strawberry
+from django.http import Http404, HttpRequest
+from django.template import RequestContext, Template
+from django.template.exceptions import TemplateDoesNotExist
+from django.template.loader import render_to_string
+from django.template.response import TemplateResponse
+from strawberry.django.views import AsyncGraphQLView as StrawberryAsyncGraphQLView
+
+
+class AsyncGraphQLView(StrawberryAsyncGraphQLView):
+    def _render_graphiql(self, request: HttpRequest, context=None):
+        if not self.graphiql:
+            raise Http404()
+
+        try:
+            template = Template(render_to_string("graphql/graphiql.html"))
+        except TemplateDoesNotExist:
+            template = Template(
+                open(
+                    os.path.join(
+                        os.path.dirname(os.path.abspath(strawberry.__file__)),
+                        "static/graphiql.html",
+                    ),
+                    "r",
+                ).read()
+            )
+
+        context = context or {}
+        # THIS enables subscriptions
+        context.update({"SUBSCRIPTION_ENABLED": "true"})
+
+        response = TemplateResponse(request=request, template=None, context=context)
+        response.content = template.render(RequestContext(request, context))
+
+        return response

django-subscriptions/api/views.py

@@ -0,0 +1,27 @@
+import os
+
+from api.asgi import MyGraphQL
+from django.core.asgi import get_asgi_application
+from starlette.websockets import WebSocketDisconnect
+
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "demo.settings")
+
+django_application = get_asgi_application()
+
+
+async def application(scope, receive, send):
+    # TODO: always use ASGI graphql
+
+    if scope["type"] == "http":
+        await django_application(scope, receive, send)
+    elif scope["type"] == "websocket":
+        try:
+            from api.schema import schema
+
+            graphql_app = MyGraphQL(schema, keep_alive=True, keep_alive_interval=5)
+
+            await graphql_app(scope, receive, send)
+        except WebSocketDisconnect:
+            pass
+    else:
+        raise NotImplementedError(f"Unknown scope type {scope['type']}")

django-subscriptions/demo/__init__.py

@@ -0,0 +1,120 @@
+"""
+Django settings for demo project.
+
+Generated by 'django-admin startproject' using Django 3.1.7.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/3.1/topics/settings/
+
+For the full list of settings and their values, see
+https://docs.djangoproject.com/en/3.1/ref/settings/
+"""
+
+from pathlib import Path
+
+# Build paths inside the project like this: BASE_DIR / 'subdir'.
+BASE_DIR = Path(__file__).resolve().parent.parent
+
+
+# Quick-start development settings - unsuitable for production
+# See https://docs.djangoproject.com/en/3.1/howto/deployment/checklist/
+
+# SECURITY WARNING: keep the secret key used in production secret!
+SECRET_KEY = 'p-h@br&tcqetcsaa%))9%5-80qjv&*-2_@6eddvp!3m)z#kyq%'
+
+# SECURITY WARNING: don't run with debug turned on in production!
+DEBUG = True
+
+ALLOWED_HOSTS = []
+
+
+# Application definition
+
+INSTALLED_APPS = [
+    'django.contrib.admin',
+    'django.contrib.auth',
+    'django.contrib.contenttypes',
+    'django.contrib.sessions',
+    'django.contrib.messages',
+    'django.contrib.staticfiles',
+]
+
+MIDDLEWARE = [
+    'django.middleware.security.SecurityMiddleware',
+    'django.contrib.sessions.middleware.SessionMiddleware',
+    'django.middleware.common.CommonMiddleware',
+    'django.middleware.csrf.CsrfViewMiddleware',
+    'django.contrib.auth.middleware.AuthenticationMiddleware',
+    'django.contrib.messages.middleware.MessageMiddleware',
+    'django.middleware.clickjacking.XFrameOptionsMiddleware',
+]
+
+ROOT_URLCONF = 'demo.urls'
+
+TEMPLATES = [
+    {
+        'BACKEND': 'django.template.backends.django.DjangoTemplates',
+        'DIRS': [],
+        'APP_DIRS': True,
+        'OPTIONS': {
+            'context_processors': [
+                'django.template.context_processors.debug',
+                'django.template.context_processors.request',
+                'django.contrib.auth.context_processors.auth',
+                'django.contrib.messages.context_processors.messages',
+            ],
+        },
+    },
+]
+
+WSGI_APPLICATION = 'demo.wsgi.application'
+
+
+# Database
+# https://docs.djangoproject.com/en/3.1/ref/settings/#databases
+
+DATABASES = {
+    'default': {
+        'ENGINE': 'django.db.backends.sqlite3',
+        'NAME': BASE_DIR / 'db.sqlite3',
+    }
+}
+
+
+# Password validation
+# https://docs.djangoproject.com/en/3.1/ref/settings/#auth-password-validators
+
+AUTH_PASSWORD_VALIDATORS = [
+    {
+        'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
+    },
+    {
+        'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
+    },
+]
+
+
+# Internationalization
+# https://docs.djangoproject.com/en/3.1/topics/i18n/
+
+LANGUAGE_CODE = 'en-us'
+
+TIME_ZONE = 'UTC'
+
+USE_I18N = True
+
+USE_L10N = True
+
+USE_TZ = True
+
+
+# Static files (CSS, JavaScript, Images)
+# https://docs.djangoproject.com/en/3.1/howto/static-files/
+
+STATIC_URL = '/static/'

django-subscriptions/demo/asgi.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+from django.urls import include, path
+
+urlpatterns = [
+    path("admin/", admin.site.urls),
+    path("", include("api.urls")),
+]