Sfoglia il codice sorgente

修复了websocket逻辑错误

Shellmiao 4 anni fa
parent
commit
75fb295edc
2 ha cambiato i file con 7 aggiunte e 4 eliminazioni
  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