mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-03 21:43:21 +00:00 
			
		
		
		
	Saw these when grepping for these two decorators; they're actually more numerous than the surviving use sites are. Cut out the noise.
		
			
				
	
	
		
			69 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			69 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
 | 
						|
from django.http import HttpRequest, HttpResponse
 | 
						|
from django.utils.translation import ugettext as _
 | 
						|
from typing import Text
 | 
						|
 | 
						|
from zerver.decorator import \
 | 
						|
    has_request_variables, REQ, to_non_negative_int
 | 
						|
from zerver.lib.actions import do_add_reaction_legacy,\
 | 
						|
    do_remove_reaction_legacy
 | 
						|
from zerver.lib.emoji import check_valid_emoji
 | 
						|
from zerver.lib.message import access_message
 | 
						|
from zerver.lib.request import JsonableError
 | 
						|
from zerver.lib.response import json_success
 | 
						|
from zerver.models import Message, Reaction, UserMessage, UserProfile
 | 
						|
 | 
						|
def create_historical_message(user_profile, message):
 | 
						|
    # type: (UserProfile, Message) -> None
 | 
						|
    # Users can see and react to messages sent to streams they
 | 
						|
    # were not a subscriber to; in order to receive events for
 | 
						|
    # those, we give the user a `historical` UserMessage objects
 | 
						|
    # for the message.  This is the same trick we use for starring
 | 
						|
    # messages.
 | 
						|
    UserMessage.objects.create(user_profile=user_profile,
 | 
						|
                               message=message,
 | 
						|
                               flags=UserMessage.flags.historical | UserMessage.flags.read)
 | 
						|
 | 
						|
@has_request_variables
 | 
						|
def add_reaction_backend(request, user_profile, message_id, emoji_name):
 | 
						|
    # type: (HttpRequest, UserProfile, int, Text) -> HttpResponse
 | 
						|
 | 
						|
    # access_message will throw a JsonableError exception if the user
 | 
						|
    # cannot see the message (e.g. for messages to private streams).
 | 
						|
    message, user_message = access_message(user_profile, message_id)
 | 
						|
 | 
						|
    check_valid_emoji(message.sender.realm, emoji_name)
 | 
						|
 | 
						|
    # We could probably just make this check be a try/except for the
 | 
						|
    # IntegrityError from it already existing, but this is a bit cleaner.
 | 
						|
    if Reaction.objects.filter(user_profile=user_profile,
 | 
						|
                               message=message,
 | 
						|
                               emoji_name=emoji_name).exists():
 | 
						|
        raise JsonableError(_("Reaction already exists"))
 | 
						|
 | 
						|
    if user_message is None:
 | 
						|
        create_historical_message(user_profile, message)
 | 
						|
 | 
						|
    do_add_reaction_legacy(user_profile, message, emoji_name)
 | 
						|
 | 
						|
    return json_success()
 | 
						|
 | 
						|
@has_request_variables
 | 
						|
def remove_reaction_backend(request, user_profile, message_id, emoji_name):
 | 
						|
    # type: (HttpRequest, UserProfile, int, Text) -> HttpResponse
 | 
						|
 | 
						|
    # access_message will throw a JsonableError exception if the user
 | 
						|
    # cannot see the message (e.g. for messages to private streams).
 | 
						|
    message = access_message(user_profile, message_id)[0]
 | 
						|
 | 
						|
    # We could probably just make this check be a try/except for the
 | 
						|
    # IntegrityError from it already existing, but this is a bit cleaner.
 | 
						|
    if not Reaction.objects.filter(user_profile=user_profile,
 | 
						|
                                   message=message,
 | 
						|
                                   emoji_name=emoji_name).exists():
 | 
						|
        raise JsonableError(_("Reaction does not exist"))
 | 
						|
 | 
						|
    do_remove_reaction_legacy(user_profile, message, emoji_name)
 | 
						|
 | 
						|
    return json_success()
 |