From c8f53e55db295a2cd4fd7638db7e43e596fd33fb Mon Sep 17 00:00:00 2001 From: Tim Abbott Date: Thu, 24 Aug 2017 22:54:40 -0700 Subject: [PATCH] websocket_client: Use get_system_bot. This messy code is used only for our websocket Nagios tests, so using get_system_bot makes sense. --- zerver/tornado/websocket_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zerver/tornado/websocket_client.py b/zerver/tornado/websocket_client.py index 85bb74e294..23629b05c5 100644 --- a/zerver/tornado/websocket_client.py +++ b/zerver/tornado/websocket_client.py @@ -16,7 +16,7 @@ from tornado.websocket import websocket_connect, WebSocketClientConnection from six.moves.urllib.parse import urlparse, urlunparse, urljoin from six.moves.http_cookies import SimpleCookie -from zerver.models import UserProfile +from zerver.models import get_system_bot from typing import Any, Callable, Dict, Generator, Iterable, Optional @@ -27,7 +27,7 @@ class WebsocketClient(object): # type: (str, str, str, Callable, bool, **Any) -> None self.validate_ssl = validate_ssl self.auth_email = sender_email - self.user_profile = UserProfile.objects.filter(email=self.auth_email).first() + self.user_profile = get_system_bot(sender_email) self.request_id_number = 0 self.parsed_host_url = urlparse(host_url) self.sockjs_url = sockjs_url