paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-23 04:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

django_api: Extract send_event_on_commit helper.

Browse Source 
django-stubs 4.2.1 gives transaction.on_commit a more accurate type
annotation, but this exposed that mypy can’t handle the lambda default
parameters that we use to recapture loop variables such as

    for stream_id in public_stream_ids:
        peer_user_ids = …
        event = …

        transaction.on_commit(
            lambda event=event, peer_user_ids=peer_user_ids: send_event(
                realm, event, peer_user_ids
            )
        )

https://github.com/python/mypy/issues/15459

A workaround that mypy accepts is

        transaction.on_commit(
            (
                lambda event, peer_user_ids: lambda: send_event(
                    realm, event, peer_user_ids
                )
            )(event, peer_user_ids)
        )

But that’s kind of ugly and potentially error-prone, so let’s make a
helper function for this very common pattern.

        send_event_on_commit(realm, event, peer_user_ids)

Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg
2023-06-17 11:53:07 -07:00
committed by Tim Abbott
parent 77c146b8b0
commit 7657cb4a0f
22 changed files with 114 additions and 180 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
zerver/actions/bots.py
View File

@@ -6,7 +6,7 @@ from django.utils.timezone import now as timezone_now
from zerver.actions.create_user import created_bot_event
from zerver.models import RealmAuditLog, Stream, UserProfile, active_user_ids, bot_owner_user_ids
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
@transaction.atomic(durable=True)
@@ -41,12 +41,10 @@ def do_change_bot_owner(
),
)
previous_owner_id = previous_owner.id
transaction.on_commit(
lambda: send_event(
user_profile.realm,
delete_event,
{previous_owner_id},
)
send_event_on_commit(
user_profile.realm,
delete_event,
{previous_owner_id},
)
# Do not send update event for previous bot owner.
update_users = update_users - {previous_owner.id}
@@ -54,7 +52,7 @@ def do_change_bot_owner(
# Notify the new owner that the bot has been added.
if not bot_owner.is_realm_admin:
add_event = created_bot_event(user_profile)
transaction.on_commit(lambda: send_event(user_profile.realm, add_event, {bot_owner.id}))
send_event_on_commit(user_profile.realm, add_event, {bot_owner.id})
# Do not send update event for bot_owner.
update_users = update_users - {bot_owner.id}
@@ -66,12 +64,10 @@ def do_change_bot_owner(
owner_id=user_profile.bot_owner.id,
),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
bot_event,
update_users,
)
send_event_on_commit(
user_profile.realm,
bot_event,
update_users,
)
# Since `bot_owner_id` is included in the user profile dict we need
@@ -84,9 +80,7 @@ def do_change_bot_owner(
bot_owner_id=user_profile.bot_owner.id,
),
)
transaction.on_commit(
lambda: send_event(user_profile.realm, event, active_user_ids(user_profile.realm_id))
)
send_event_on_commit(user_profile.realm, event, active_user_ids(user_profile.realm_id))
@transaction.atomic(durable=True)
@@ -125,12 +119,10 @@ def do_change_default_sending_stream(
default_sending_stream=stream_name,
),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)
send_event_on_commit(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)
@@ -171,12 +163,10 @@ def do_change_default_events_register_stream(
default_events_register_stream=stream_name,
),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)
send_event_on_commit(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)
@@ -212,10 +202,8 @@ def do_change_default_all_public_streams(
default_all_public_streams=user_profile.default_all_public_streams,
),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)
send_event_on_commit(
user_profile.realm,
event,
bot_owner_user_ids(user_profile),
)

18
zerver/actions/create_user.py
View File

@@ -48,7 +48,7 @@ from zerver.models import (
bot_owner_user_ids,
get_system_bot,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
if settings.BILLING_ENABLED:
from corporate.lib.stripe import update_license_ledger_if_needed
@@ -311,20 +311,12 @@ def notify_created_user(user_profile: UserProfile) -> None:
event: Dict[str, Any] = dict(
type="realm_user", op="add", person=person_for_real_email_access_users
)
transaction.on_commit(
lambda event=event: send_event(
user_profile.realm, event, user_ids_with_real_email_access
)
)
send_event_on_commit(user_profile.realm, event, user_ids_with_real_email_access)
if user_ids_without_real_email_access:
assert person_for_without_real_email_access_users is not None
event = dict(type="realm_user", op="add", person=person_for_without_real_email_access_users)
transaction.on_commit(
lambda event=event: send_event(
user_profile.realm, event, user_ids_without_real_email_access
)
)
send_event_on_commit(user_profile.realm, event, user_ids_without_real_email_access)
def created_bot_event(user_profile: UserProfile) -> Dict[str, Any]:
@@ -361,9 +353,7 @@ def created_bot_event(user_profile: UserProfile) -> Dict[str, Any]:
def notify_created_bot(user_profile: UserProfile) -> None:
event = created_bot_event(user_profile)
transaction.on_commit(
lambda: send_event(user_profile.realm, event, bot_owner_user_ids(user_profile))
)
send_event_on_commit(user_profile.realm, event, bot_owner_user_ids(user_profile))
def do_create_user(

6
zerver/actions/default_streams.py
View File

@@ -13,7 +13,7 @@ from zerver.models import (
active_non_guest_user_ids,
get_default_stream_groups,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def check_default_stream_group_name(group_name: str) -> None:
@@ -52,7 +52,7 @@ def notify_default_streams(realm: Realm) -> None:
type="default_streams",
default_streams=streams_to_dicts_sorted(get_default_streams_for_realm(realm.id)),
)
transaction.on_commit(lambda: send_event(realm, event, active_non_guest_user_ids(realm.id)))
send_event_on_commit(realm, event, active_non_guest_user_ids(realm.id))
def notify_default_stream_groups(realm: Realm) -> None:
@@ -62,7 +62,7 @@ def notify_default_stream_groups(realm: Realm) -> None:
get_default_stream_groups(realm)
),
)
transaction.on_commit(lambda: send_event(realm, event, active_non_guest_user_ids(realm.id)))
send_event_on_commit(realm, event, active_non_guest_user_ids(realm.id))
def do_add_default_stream(stream: Stream) -> None:

6
zerver/actions/message_delete.py
View File

@@ -1,12 +1,10 @@
from typing import Iterable, List, TypedDict
from django.db import transaction
from zerver.lib import retention
from zerver.lib.retention import move_messages_to_archive
from zerver.lib.stream_subscription import get_active_subscriptions_for_stream_id
from zerver.models import Message, Realm, UserMessage, UserProfile
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
class DeleteMessagesEvent(TypedDict, total=False):
@@ -56,7 +54,7 @@ def do_delete_messages(realm: Realm, messages: Iterable[Message]) -> None:
move_messages_to_archive(message_ids, realm=realm, chunk_size=archiving_chunk_size)
event["message_type"] = message_type
transaction.on_commit(lambda: send_event(realm, event, users_to_notify))
send_event_on_commit(realm, event, users_to_notify)
def do_delete_messages_by_sender(user: UserProfile) -> None:

5
zerver/actions/reactions.py
View File

@@ -1,6 +1,5 @@
from typing import Any, Dict, Optional
from django.db import transaction
from django.utils.translation import gettext as _
from zerver.actions.create_user import create_historical_user_messages
@@ -9,7 +8,7 @@ from zerver.lib.exceptions import JsonableError
from zerver.lib.message import access_message, update_to_dict_cache
from zerver.lib.stream_subscription import subscriber_ids_with_stream_history_access
from zerver.models import Message, Reaction, Recipient, Stream, UserMessage, UserProfile
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def notify_reaction_update(
@@ -60,7 +59,7 @@ def notify_reaction_update(
stream = Stream.objects.get(id=stream_id)
user_ids |= subscriber_ids_with_stream_history_access(stream)
transaction.on_commit(lambda: send_event(user_profile.realm, event, list(user_ids)))
send_event_on_commit(user_profile.realm, event, list(user_ids))
def do_add_reaction(

10
zerver/actions/realm_domains.py
View File

@@ -14,7 +14,7 @@ from zerver.models import (
active_user_ids,
get_realm_domains,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
@transaction.atomic(durable=True)
@@ -46,7 +46,7 @@ def do_add_realm_domain(
domain=realm_domain.domain, allow_subdomains=realm_domain.allow_subdomains
),
)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
return realm_domain
@@ -82,9 +82,7 @@ def do_change_realm_domain(
domain=realm_domain.domain, allow_subdomains=realm_domain.allow_subdomains
),
)
transaction.on_commit(
lambda: send_event(realm_domain.realm, event, active_user_ids(realm_domain.realm_id))
)
send_event_on_commit(realm_domain.realm, event, active_user_ids(realm_domain.realm_id))
@transaction.atomic(durable=True)
@@ -119,4 +117,4 @@ def do_remove_realm_domain(
# confusing than the alternative.
do_set_realm_property(realm, "emails_restricted_to_domains", False, acting_user=acting_user)
event = dict(type="realm_domains", op="remove", domain=domain)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))

4
zerver/actions/realm_emoji.py
View File

@@ -12,12 +12,12 @@ from zerver.lib.exceptions import JsonableError
from zerver.lib.pysa import mark_sanitized
from zerver.lib.upload import upload_emoji_image
from zerver.models import EmojiInfo, Realm, RealmAuditLog, RealmEmoji, UserProfile, active_user_ids
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def notify_realm_emoji(realm: Realm, realm_emoji: Dict[str, EmojiInfo]) -> None:
event = dict(type="realm_emoji", op="update", realm_emoji=realm_emoji)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
def check_add_realm_emoji(

5
zerver/actions/realm_export.py
View File

@@ -1,17 +1,16 @@
import orjson
from django.db import transaction
from django.utils.timezone import now as timezone_now
from zerver.lib.export import get_realm_exports_serialized
from zerver.lib.upload import delete_export_tarball
from zerver.models import RealmAuditLog, UserProfile
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def notify_realm_export(user_profile: UserProfile) -> None:
# In the future, we may want to send this event to all realm admins.
event = dict(type="realm_export", exports=get_realm_exports_serialized(user_profile))
transaction.on_commit(lambda: send_event(user_profile.realm, event, [user_profile.id]))
send_event_on_commit(user_profile.realm, event, [user_profile.id])
def do_delete_realm_export(user_profile: UserProfile, export: RealmAuditLog) -> None:

12
zerver/actions/realm_icon.py
View File

@@ -5,7 +5,7 @@ from django.utils.timezone import now as timezone_now
from zerver.lib.realm_icon import realm_icon_url
from zerver.models import Realm, RealmAuditLog, UserProfile, active_user_ids
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
@transaction.atomic(durable=True)
@@ -31,10 +31,8 @@ def do_change_icon_source(
property="icon",
data=dict(icon_source=realm.icon_source, icon_url=realm_icon_url(realm)),
)
transaction.on_commit(
lambda: send_event(
realm,
event,
active_user_ids(realm.id),
)
send_event_on_commit(
realm,
event,
active_user_ids(realm.id),
)

4
zerver/actions/realm_linkifiers.py
View File

@@ -13,12 +13,12 @@ from zerver.models import (
active_user_ids,
linkifiers_for_realm,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def notify_linkifiers(realm: Realm, realm_linkifiers: List[LinkifierDict]) -> None:
event: Dict[str, object] = dict(type="realm_linkifiers", realm_linkifiers=realm_linkifiers)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
# NOTE: Regexes must be simple enough that they can be easily translated to JavaScript

4
zerver/actions/realm_logo.py
View File

@@ -5,7 +5,7 @@ from django.utils.timezone import now as timezone_now
from zerver.lib.realm_logo import get_realm_logo_data
from zerver.models import Realm, RealmAuditLog, UserProfile, active_user_ids
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
@transaction.atomic(durable=True)
@@ -35,4 +35,4 @@ def do_change_logo_source(
property="night_logo" if night else "logo",
data=get_realm_logo_data(realm, night),
)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))

4
zerver/actions/realm_playgrounds.py
View File

@@ -13,12 +13,12 @@ from zerver.models import (
active_user_ids,
get_realm_playgrounds,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def notify_realm_playgrounds(realm: Realm, realm_playgrounds: List[RealmPlaygroundDict]) -> None:
event = dict(type="realm_playgrounds", realm_playgrounds=realm_playgrounds)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
@transaction.atomic(durable=True)

8
zerver/actions/realm_settings.py
View File

@@ -36,7 +36,7 @@ from zerver.models import (
active_user_ids,
get_realm,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event, send_event_on_commit
if settings.BILLING_ENABLED:
from corporate.lib.stripe import downgrade_now_without_creating_additional_invoices
@@ -83,7 +83,7 @@ def do_set_realm_property(
data={name: value},
)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
event_time = timezone_now()
RealmAuditLog.objects.create(
@@ -435,7 +435,7 @@ def do_change_realm_org_type(
)
event = dict(type="realm", op="update", property="org_type", value=org_type)
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
@transaction.atomic(savepoint=False)
@@ -499,7 +499,7 @@ def do_change_realm_plan_type(
"value": plan_type,
"extra_data": {"upload_quota": realm.upload_quota_bytes()},
}
transaction.on_commit(lambda: send_event(realm, event, active_user_ids(realm.id)))
send_event_on_commit(realm, event, active_user_ids(realm.id))
def do_send_realm_reactivation_email(realm: Realm, *, acting_user: Optional[UserProfile]) -> None:

26
zerver/actions/streams.py
View File

@@ -64,7 +64,7 @@ from zerver.models import (
active_non_guest_user_ids,
get_system_bot,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event, send_event_on_commit
@transaction.atomic(savepoint=False)
@@ -127,7 +127,7 @@ def do_deactivate_stream(
stream_dict = stream.to_dict()
stream_dict.update(dict(name=old_name, invite_only=was_invite_only))
event = dict(type="stream", op="delete", streams=[stream_dict])
transaction.on_commit(lambda: send_event(stream.realm, event, affected_user_ids))
send_event_on_commit(stream.realm, event, affected_user_ids)
event_time = timezone_now()
RealmAuditLog.objects.create(
@@ -403,11 +403,7 @@ def send_peer_subscriber_events(
stream_ids=[stream_id],
user_ids=sorted(altered_user_ids),
)
transaction.on_commit(
lambda event=event, peer_user_ids=peer_user_ids: send_event(
realm, event, peer_user_ids
)
)
send_event_on_commit(realm, event, peer_user_ids)
public_stream_ids = [
stream_id
@@ -445,11 +441,7 @@ def send_peer_subscriber_events(
stream_ids=[stream_id],
user_ids=sorted(altered_user_ids),
)
transaction.on_commit(
lambda event=event, peer_user_ids=peer_user_ids: send_event(
realm, event, peer_user_ids
)
)
send_event_on_commit(realm, event, peer_user_ids)
for user_id, stream_ids in user_streams.items():
peer_user_ids = public_peer_ids - {user_id}
@@ -459,11 +451,7 @@ def send_peer_subscriber_events(
stream_ids=sorted(stream_ids),
user_ids=[user_id],
)
transaction.on_commit(
lambda event=event, peer_user_ids=peer_user_ids: send_event(
realm, event, peer_user_ids
)
)
send_event_on_commit(realm, event, peer_user_ids)
SubT = Tuple[List[SubInfo], List[SubInfo]]
@@ -1359,6 +1347,4 @@ def do_change_stream_group_based_setting(
stream_id=stream.id,
name=stream.name,
)
transaction.on_commit(
lambda: send_event(stream.realm, event, can_access_stream_user_ids(stream))
)
send_event_on_commit(stream.realm, event, can_access_stream_user_ids(stream))

5
zerver/actions/submessage.py
View File

@@ -1,9 +1,8 @@
from django.db import transaction
from django.utils.translation import gettext as _
from zerver.lib.exceptions import JsonableError
from zerver.models import Realm, SubMessage, UserMessage
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event_on_commit
def verify_submessage_sender(
@@ -60,4 +59,4 @@ def do_add_submessage(
ums = UserMessage.objects.filter(message_id=message_id)
target_user_ids = [um.user_profile_id for um in ums]
transaction.on_commit(lambda: send_event(realm, event, target_user_ids))
send_event_on_commit(realm, event, target_user_ids)

10
zerver/actions/user_groups.py
View File

@@ -16,7 +16,7 @@ from zerver.models import (
UserProfile,
active_user_ids,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event, send_event_on_commit
class MemberGroupUserDict(TypedDict):
@@ -181,9 +181,7 @@ def do_send_user_group_members_update_event(
event_name: str, user_group: UserGroup, user_ids: List[int]
) -> None:
event = dict(type="user_group", op=event_name, group_id=user_group.id, user_ids=user_ids)
transaction.on_commit(
lambda: send_event(user_group.realm, event, active_user_ids(user_group.realm_id))
)
send_event_on_commit(user_group.realm, event, active_user_ids(user_group.realm_id))
@transaction.atomic(savepoint=False)
@@ -216,9 +214,7 @@ def do_send_subgroups_update_event(
event = dict(
type="user_group", op=event_name, group_id=user_group.id, direct_subgroup_ids=subgroup_ids
)
transaction.on_commit(
lambda: send_event(user_group.realm, event, active_user_ids(user_group.realm_id))
)
send_event_on_commit(user_group.realm, event, active_user_ids(user_group.realm_id))
@transaction.atomic

76
zerver/actions/user_settings.py
View File

@@ -42,18 +42,16 @@ from zerver.models import (
get_client,
get_user_profile_by_id,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event, send_event_on_commit
def send_user_email_update_event(user_profile: UserProfile) -> None:
payload = dict(user_id=user_profile.id, new_email=user_profile.email)
event = dict(type="realm_user", op="update", person=payload)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
event,
active_user_ids(user_profile.realm_id),
)
send_event_on_commit(
user_profile.realm,
event,
active_user_ids(user_profile.realm_id),
)
@@ -81,22 +79,18 @@ def send_delivery_email_update_events(
if delivery_email_now_visible_user_ids:
person = dict(user_id=user_profile.id, delivery_email=user_profile.delivery_email)
event = dict(type="realm_user", op="update", person=person)
transaction.on_commit(
lambda event=event: send_event(
user_profile.realm,
event,
delivery_email_now_visible_user_ids,
)
send_event_on_commit(
user_profile.realm,
event,
delivery_email_now_visible_user_ids,
)
if delivery_email_now_invisible_user_ids:
person = dict(user_id=user_profile.id, delivery_email=None)
event = dict(type="realm_user", op="update", person=person)
transaction.on_commit(
lambda event=event: send_event(
user_profile.realm,
event,
delivery_email_now_invisible_user_ids,
)
send_event_on_commit(
user_profile.realm,
event,
delivery_email_now_invisible_user_ids,
)
@@ -116,9 +110,7 @@ def do_change_user_delivery_email(user_profile: UserProfile, new_email: str) ->
event = dict(type="realm_user", op="update", person=payload)
delivery_email_visible_user_ids = get_users_with_access_to_real_email(user_profile)
transaction.on_commit(
lambda: send_event(user_profile.realm, event, delivery_email_visible_user_ids)
)
send_event_on_commit(user_profile.realm, event, delivery_email_visible_user_ids)
if user_profile.avatar_source == UserProfile.AVATAR_FROM_GRAVATAR:
# If the user is using Gravatar to manage their email address,
@@ -320,12 +312,10 @@ def notify_avatar_url_change(user_profile: UserProfile) -> None:
avatar_url=avatar_url(user_profile),
),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
bot_event,
bot_owner_user_ids(user_profile),
)
send_event_on_commit(
user_profile.realm,
bot_event,
bot_owner_user_ids(user_profile),
)
payload = dict(
@@ -339,12 +329,10 @@ def notify_avatar_url_change(user_profile: UserProfile) -> None:
)
event = dict(type="realm_user", op="update", person=payload)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
event,
active_user_ids(user_profile.realm_id),
)
send_event_on_commit(
user_profile.realm,
event,
active_user_ids(user_profile.realm_id),
)
@@ -450,7 +438,7 @@ def do_change_user_setting(
assert isinstance(setting_value, str)
event["language_name"] = get_language_name(setting_value)
transaction.on_commit(lambda: send_event(user_profile.realm, event, [user_profile.id]))
send_event_on_commit(user_profile.realm, event, [user_profile.id])
if setting_name in UserProfile.notification_settings_legacy:
# This legacy event format is for backwards-compatibility with
@@ -462,9 +450,7 @@ def do_change_user_setting(
"notification_name": setting_name,
"setting": setting_value,
}
transaction.on_commit(
lambda: send_event(user_profile.realm, legacy_event, [user_profile.id])
)
send_event_on_commit(user_profile.realm, legacy_event, [user_profile.id])
if setting_name in UserProfile.display_settings_legacy or setting_name == "timezone":
# This legacy event format is for backwards-compatibility with
@@ -480,9 +466,7 @@ def do_change_user_setting(
assert isinstance(setting_value, str)
legacy_event["language_name"] = get_language_name(setting_value)
transaction.on_commit(
lambda: send_event(user_profile.realm, legacy_event, [user_profile.id])
)
send_event_on_commit(user_profile.realm, legacy_event, [user_profile.id])
# Updates to the time zone display setting are sent to all users
if setting_name == "timezone":
@@ -492,12 +476,10 @@ def do_change_user_setting(
timezone=canonicalize_timezone(user_profile.timezone),
)
timezone_event = dict(type="realm_user", op="update", person=payload)
transaction.on_commit(
lambda: send_event(
user_profile.realm,
timezone_event,
active_user_ids(user_profile.realm_id),
)
send_event_on_commit(
user_profile.realm,
timezone_event,
active_user_ids(user_profile.realm_id),
)
if setting_name == "email_address_visibility":

18
zerver/actions/users.py
View File

@@ -39,7 +39,7 @@ from zerver.models import (
get_fake_email_domain,
get_user_profile_by_id,
)
from zerver.tornado.django_api import send_event
from zerver.tornado.django_api import send_event, send_event_on_commit
if settings.BILLING_ENABLED:
from corporate.lib.stripe import update_license_ledger_if_needed
@@ -297,10 +297,8 @@ def do_deactivate_user(
op="remove",
person=dict(user_id=user_profile.id, full_name=user_profile.full_name),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm, event_remove_user, active_user_ids(user_profile.realm_id)
)
send_event_on_commit(
user_profile.realm, event_remove_user, active_user_ids(user_profile.realm_id)
)
if user_profile.is_bot:
@@ -309,10 +307,8 @@ def do_deactivate_user(
op="remove",
bot=dict(user_id=user_profile.id, full_name=user_profile.full_name),
)
transaction.on_commit(
lambda: send_event(
user_profile.realm, event_remove_bot, bot_owner_user_ids(user_profile)
)
send_event_on_commit(
user_profile.realm, event_remove_bot, bot_owner_user_ids(user_profile)
)
@@ -342,9 +338,7 @@ def do_change_user_role(
event = dict(
type="realm_user", op="update", person=dict(user_id=user_profile.id, role=user_profile.role)
)
transaction.on_commit(
lambda: send_event(user_profile.realm, event, active_user_ids(user_profile.realm_id))
)
send_event_on_commit(user_profile.realm, event, active_user_ids(user_profile.realm_id))
UserGroupMembership.objects.filter(
user_profile=user_profile, user_group=old_system_group

2
zerver/tests/test_message_edit.py
View File

@@ -4476,7 +4476,7 @@ class DeleteMessageTest(ZulipTestCase):
message = self.get_last_message()
with self.capture_send_event_calls(expected_num_events=1):
with mock.patch("zerver.actions.message_edit.send_event") as m:
with mock.patch("zerver.tornado.django_api.queue_json_publish") as m:
m.side_effect = AssertionError(
"Events should be sent only after the transaction commits."
)

4
zerver/tests/test_reactions.py
View File

@@ -1060,7 +1060,7 @@ class ReactionAPIEventTest(EmojiReactionBase):
"reaction_type": "unicode_emoji",
}
with self.capture_send_event_calls(expected_num_events=1) as events:
with mock.patch("zerver.actions.reactions.send_event") as m:
with mock.patch("zerver.tornado.django_api.queue_json_publish") as m:
m.side_effect = AssertionError(
"Events should be sent only after the transaction commits!"
)
@@ -1143,7 +1143,7 @@ class ReactionAPIEventTest(EmojiReactionBase):
)
with self.capture_send_event_calls(expected_num_events=1):
with mock.patch("zerver.actions.reactions.send_event") as m:
with mock.patch("zerver.tornado.django_api.queue_json_publish") as m:
m.side_effect = AssertionError(
"Events should be sent only after the transaction commits."
)

2
zerver/tests/test_submessage.py
View File

@@ -195,7 +195,7 @@ class TestBasics(ZulipTestCase):
message_id = self.send_stream_message(hamlet, "Denmark")
with self.capture_send_event_calls(expected_num_events=1):
with mock.patch("zerver.actions.submessage.send_event") as m:
with mock.patch("zerver.tornado.django_api.queue_json_publish") as m:
m.side_effect = AssertionError(
"Events should be sent only after the transaction commits."
)

7
zerver/tornado/django_api.py
View File

@@ -6,6 +6,7 @@ from urllib.parse import urlparse
import orjson
import requests
from django.conf import settings
from django.db import transaction
from requests.adapters import ConnectionError, HTTPAdapter
from requests.models import PreparedRequest, Response
from urllib3.util import Retry
@@ -186,3 +187,9 @@ def send_event(
dict(event=event, users=port_users),
lambda *args, **kwargs: send_notification_http(port, *args, **kwargs),
)
def send_event_on_commit(
realm: Realm, event: Mapping[str, Any], users: Union[Iterable[int], Iterable[Mapping[str, Any]]]
) -> None:
transaction.on_commit(lambda: send_event(realm, event, users))