前后端第一版提交

hertz_server_django/asgi.py

@@ -0,0 +1,49 @@
+"""
+ASGI config for hertz_server_django project.
+
+It exposes the ASGI callable as a module-level variable named ``application``.
+
+For more information on this file, see
+https://docs.djangoproject.com/en/5.2/howto/deployment/asgi/
+"""
+
+import os
+from django.core.asgi import get_asgi_application
+from django.conf import settings
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.auth import AuthMiddlewareStack
+from channels.security.websocket import AllowedHostsOriginValidator
+
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'hertz_server_django.settings')
+
+# Initialize Django ASGI application early to ensure the AppRegistry
+# is populated before importing code that may import ORM models.
+django_asgi_app = get_asgi_application()
+
+# Import websocket routing AFTER Django setup to avoid AppRegistryNotReady
+from hertz_demo import routing as demo_routing
+from hertz_studio_django_yolo import routing as yolo_routing
+
+# Merge websocket routes
+websocket_urlpatterns = (
+    demo_routing.websocket_urlpatterns +
+    yolo_routing.websocket_urlpatterns
+)
+
+# 在开发环境下放宽Origin校验，便于第三方客户端（如 Apifox、wscat）调试
+websocket_app = AuthMiddlewareStack(
+    URLRouter(
+        websocket_urlpatterns
+    )
+)
+
+if getattr(settings, 'DEBUG', False):
+    application = ProtocolTypeRouter({
+        "http": django_asgi_app,
+        "websocket": websocket_app,
+    })
+else:
+    application = ProtocolTypeRouter({
+        "http": django_asgi_app,
+        "websocket": AllowedHostsOriginValidator(websocket_app),
+    })