mirror of https://github.com/zulip/zulip.git
91 lines
3.7 KiB
Python
91 lines
3.7 KiB
Python
|
|
from django.utils.translation import ugettext as _
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
from zerver.models import get_client, UserProfile, Client
|
|
|
|
from zerver.decorator import asynchronous, \
|
|
internal_notify_view, RespondAsynchronously, \
|
|
has_request_variables, REQ, _RespondAsynchronously
|
|
|
|
from zerver.lib.response import json_success, json_error
|
|
from zerver.lib.validator import check_bool, check_list, check_string
|
|
from zerver.tornado.event_queue import get_client_descriptor, \
|
|
process_notification, fetch_events
|
|
from zerver.tornado.exceptions import BadEventQueueIdError
|
|
from django.core.handlers.base import BaseHandler
|
|
|
|
from typing import Union, Optional, Iterable, Sequence, List, Text
|
|
import time
|
|
import ujson
|
|
|
|
@internal_notify_view(True)
|
|
def notify(request: HttpRequest) -> HttpResponse:
|
|
process_notification(ujson.loads(request.POST['data']))
|
|
return json_success()
|
|
|
|
@has_request_variables
|
|
def cleanup_event_queue(request: HttpRequest, user_profile: UserProfile,
|
|
queue_id: Text=REQ()) -> HttpResponse:
|
|
client = get_client_descriptor(str(queue_id))
|
|
if client is None:
|
|
raise BadEventQueueIdError(queue_id)
|
|
if user_profile.id != client.user_profile_id:
|
|
return json_error(_("You are not authorized to access this queue"))
|
|
request._log_data['extra'] = "[%s]" % (queue_id,)
|
|
client.cleanup()
|
|
return json_success()
|
|
|
|
@asynchronous
|
|
@has_request_variables
|
|
def get_events_backend(request, user_profile, handler,
|
|
user_client = REQ(converter=get_client, default=None),
|
|
last_event_id = REQ(converter=int, default=None),
|
|
queue_id = REQ(default=None),
|
|
apply_markdown = REQ(default=False, validator=check_bool),
|
|
all_public_streams = REQ(default=False, validator=check_bool),
|
|
event_types = REQ(default=None, validator=check_list(check_string)),
|
|
dont_block = REQ(default=False, validator=check_bool),
|
|
narrow = REQ(default=[], validator=check_list(None)),
|
|
lifespan_secs = REQ(default=0, converter=int)):
|
|
# type: (HttpRequest, UserProfile, BaseHandler, Optional[Client], Optional[int], Optional[List[Text]], bool, bool, Optional[Text], bool, Iterable[Sequence[Text]], int) -> Union[HttpResponse, _RespondAsynchronously]
|
|
if user_client is None:
|
|
user_client = request.client
|
|
|
|
events_query = dict(
|
|
user_profile_id = user_profile.id,
|
|
user_profile_email = user_profile.email,
|
|
queue_id = queue_id,
|
|
last_event_id = last_event_id,
|
|
event_types = event_types,
|
|
client_type_name = user_client.name,
|
|
all_public_streams = all_public_streams,
|
|
lifespan_secs = lifespan_secs,
|
|
narrow = narrow,
|
|
dont_block = dont_block,
|
|
handler_id = handler.handler_id)
|
|
|
|
if queue_id is None:
|
|
events_query['new_queue_data'] = dict(
|
|
user_profile_id = user_profile.id,
|
|
realm_id = user_profile.realm_id,
|
|
user_profile_email = user_profile.email,
|
|
event_types = event_types,
|
|
client_type_name = user_client.name,
|
|
apply_markdown = apply_markdown,
|
|
all_public_streams = all_public_streams,
|
|
queue_timeout = lifespan_secs,
|
|
last_connection_time = time.time(),
|
|
narrow = narrow)
|
|
|
|
result = fetch_events(events_query)
|
|
if "extra_log_data" in result:
|
|
request._log_data['extra'] = result["extra_log_data"]
|
|
|
|
if result["type"] == "async":
|
|
handler._request = request
|
|
return RespondAsynchronously
|
|
if result["type"] == "error":
|
|
raise result["exception"]
|
|
return json_success(result["response"])
|