mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +00:00
cf762eaf84
This makes it more clearly the pattern in the Zulip codebase, and thus decreases the risk of accidentally doing database queries.
92 lines
3.8 KiB
Python
92 lines
3.8 KiB
Python
from __future__ import absolute_import
|
|
|
|
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, \
|
|
authenticated_json_post_view, 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 django.core.handlers.base import BaseHandler
|
|
|
|
from typing import Union, Optional, Iterable, Sequence, List, Text
|
|
import time
|
|
import ujson
|
|
|
|
@internal_notify_view
|
|
def notify(request):
|
|
# type: (HttpRequest) -> HttpResponse
|
|
process_notification(ujson.loads(request.POST['data']))
|
|
return json_success()
|
|
|
|
@has_request_variables
|
|
def cleanup_event_queue(request, user_profile, queue_id=REQ()):
|
|
# type: (HttpRequest, UserProfile, Text) -> HttpResponse
|
|
client = get_client_descriptor(str(queue_id))
|
|
if client is None:
|
|
return json_error(_("Bad event queue id: %s") % (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":
|
|
return json_error(result["message"])
|
|
return json_success(result["response"])
|