diff --git a/wlm_server/wlm_server/asgi.py b/wlm_server/wlm_server/asgi.py index 03d5e6e..033872c 100644 --- a/wlm_server/wlm_server/asgi.py +++ b/wlm_server/wlm_server/asgi.py @@ -10,7 +10,19 @@ import os from django.core.asgi import get_asgi_application +from channels.auth import AuthMiddlewareStack +from channels.routing import ProtocolTypeRouter, URLRouter +from channels.security.websocket import AllowedHostsOriginValidator os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'wlm_server.settings') -application = get_asgi_application() +application = ProtocolTypeRouter( + { + 'http': get_asgi_application(), + 'websocket': AllowedHostsOriginValidator( + AuthMiddlewareStack( + URLRouter([]) + ) + ), + } +)