From 8c0b49d6c4e37bce3f91aea5a5c8a41610b4feb8 Mon Sep 17 00:00:00 2001
From: Jaehun You <jaehun0121@snu.ac.kr>
Date: Thu, 21 Nov 2024 16:14:03 +0900
Subject: [PATCH] Add url pattern for `OperationConsumer`

See also: #17
---
 wlm_server/wlm_server/asgi.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/wlm_server/wlm_server/asgi.py b/wlm_server/wlm_server/asgi.py
index 033872c..27448be 100644
--- a/wlm_server/wlm_server/asgi.py
+++ b/wlm_server/wlm_server/asgi.py
@@ -10,18 +10,24 @@
 import os
 
 from django.core.asgi import get_asgi_application
+from django.urls import path
 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')
 
+# pylint: disable=wrong-import-position
+from operation.consumers import OperationConsumer
+
 application = ProtocolTypeRouter(
     {
         'http': get_asgi_application(),
         'websocket': AllowedHostsOriginValidator(
             AuthMiddlewareStack(
-                URLRouter([])
+                URLRouter([
+                    path('ws/operation/<int:ch>/', OperationConsumer.as_asgi()),
+                ])
             )
         ),
     }