Handle wrong auth token

recircle/status/consumers.py

@@ -17,8 +17,8 @@ class StatusConsumer(WebsocketConsumer):
     # If login message arrives get the auth token from it
     # If login message arrives get the auth token from it
     # Then subscribe to the channels of the vm's that belongs to the user
     # Then subscribe to the channels of the vm's that belongs to the user
     def login(self, auth_token):
     def login(self, auth_token):
+        try:
             token = Token.objects.get(key=auth_token)
             token = Token.objects.get(key=auth_token)
-        if token:
             self.user = token.user
             self.user = token.user
             instances = Instance.objects.filter(created_by=self.user)
             instances = Instance.objects.filter(created_by=self.user)
             for inst in instances:
             for inst in instances:
@@ -26,7 +26,7 @@ class StatusConsumer(WebsocketConsumer):
                     str(inst.id),
                     str(inst.id),
                     self.channel_name
                     self.channel_name
                 )
                 )
-        else:
+        except Token.DoesNotExist:
             self.disconnect(100)
             self.disconnect(100)
 
 
     # Unsubscribe from registered groups
     # Unsubscribe from registered groups