diff --git a/api/asgi.py b/api/asgi.py index 498a23b..75649d6 100644 --- a/api/asgi.py +++ b/api/asgi.py @@ -11,6 +11,7 @@ import os from channels.routing import ProtocolTypeRouter, URLRouter from channels.auth import AuthMiddlewareStack +from channels.security.websocket import AllowedHostsOriginValidator from django.core.asgi import get_asgi_application from chat.url import websocket_urlpatterns @@ -20,6 +21,8 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", "api.settings") application = ProtocolTypeRouter( { "http": get_asgi_application(), - "websocket": AuthMiddlewareStack(URLRouter(websocket_urlpatterns)), + "websocket": AllowedHostsOriginValidator( + AuthMiddlewareStack(URLRouter(websocket_urlpatterns)) + ), } )