소스 검색

修复了websocket逻辑错误

Shellmiao 4 년 전
부모
커밋
75fb295edc
2개의 변경된 파일7개의 추가작업 그리고 4개의 파일을 삭제
  1. BIN
      db.sqlite3
  2. 7 4
      infrastructure/consumers.py

BIN
db.sqlite3


+ 7 - 4
infrastructure/consumers.py

@@ -16,7 +16,10 @@ class KnockConsumer(WebsocketConsumer):
 
 def knock_knock(username, message):
     ws = user_dict.get(username)
-    message = json.dumps({'message': message})
-    ws.send(message)
-    print(message)
-    return True
+    if ws:
+        message = json.dumps({'message': message})
+        ws.send(message)
+        print(message)
+        return True
+    else:
+        return False