[fix]

mitama-org · Mar 8, 2021 · dee5092 · dee5092
1 parent 8c0edb4
commit dee5092
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/mitama/app/http/request.py b/mitama/app/http/request.py
@@ -3,8 +3,8 @@
 import http.cookies
 import io
 import json
+import importlib
 import wsgiref.util as wsgiutil
-import uwsgi
 from urllib.parse import parse_qs
 
 
@@ -27,11 +27,18 @@ def get(self, key):
 
 class uWSGIWebSocket:
     def __init__(self, env):
-        uwsgi.websocket_handshake(env['HTTP_SEC_WEBSOCKET_KEY'], env.get('HTTP_ORIGIN', ''))
+        self.uwsgi = importlib.import_module("uwsgi")
+        self.uwsgi.websocket_handshake(
+            env['HTTP_SEC_WEBSOCKET_KEY'],
+            env.get('HTTP_ORIGIN', '')
+        )
+
     def receive(self):
-        return uwsgi.websocket_recv()
+        return self.uwsgi.websocket_recv()
+
     def send(self, message):
-        uwsgi.websocket_send(message)
+        self.uwsgi.websocket_send(message)
+
 
 class _RequestPayload:
     def __init__(self, field_storage):