Receive message

recircle/status/consumers.py

@@ -16,7 +16,7 @@ class StatusConsumer(WebsocketConsumer):
 
     # If login message arrives get the auth token from it
     # Then subscribe to the channels of the vm's that belongs to the user
-    def login(self, event):
+    def login(self, auth_token):
         user_token = event["auth_token"]
         token = Token.objects.get(key=user_token)
         if token:
@@ -40,8 +40,11 @@ class StatusConsumer(WebsocketConsumer):
                     self.channel_name
                 )
 
-    def receive(self, text_message):
-        pass
+    def receive(self, text_data):
+        text_data_json = json.loads(text_data)
+        type = text_data_json['type']
+        if type == 'login':
+            self.login(text_data_json['auth_token'])
 
     def status_changed(self, event):
         self.send(text_data=json.dumps({"vm": event["vm"], "status": event["status"]}))