mirror of
https://github.com/zulip/zulip.git
synced 2025-10-23 16:14:02 +00:00
actions: Split out zerver.actions.bots.
Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg
220
zerver/actions/bots.py
Normal file
220
zerver/actions/bots.py
Normal file
@@ -0,0 +1,220 @@
|
||||
from typing import Optional
|
||||
|
||||
import orjson
|
||||
from django.db import transaction
|
||||
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
|
||||
|
||||
|
||||
@transaction.atomic(durable=True)
|
||||
def do_change_bot_owner(
|
||||
user_profile: UserProfile, bot_owner: UserProfile, acting_user: UserProfile
|
||||
) -> None:
|
||||
previous_owner = user_profile.bot_owner
|
||||
user_profile.bot_owner = bot_owner
|
||||
user_profile.save() # Can't use update_fields because of how the foreign key works.
|
||||
event_time = timezone_now()
|
||||
RealmAuditLog.objects.create(
|
||||
realm=user_profile.realm,
|
||||
acting_user=acting_user,
|
||||
modified_user=user_profile,
|
||||
event_type=RealmAuditLog.USER_BOT_OWNER_CHANGED,
|
||||
event_time=event_time,
|
||||
)
|
||||
|
||||
update_users = bot_owner_user_ids(user_profile)
|
||||
|
||||
# For admins, update event is sent instead of delete/add
|
||||
# event. bot_data of admin contains all the
|
||||
# bots and none of them should be removed/(added again).
|
||||
|
||||
# Delete the bot from previous owner's bot data.
|
||||
if previous_owner and not previous_owner.is_realm_admin:
|
||||
delete_event = dict(
|
||||
type="realm_bot",
|
||||
op="delete",
|
||||
bot=dict(
|
||||
user_id=user_profile.id,
|
||||
),
|
||||
)
|
||||
transaction.on_commit(
|
||||
lambda: send_event(
|
||||
user_profile.realm,
|
||||
delete_event,
|
||||
{previous_owner.id},
|
||||
)
|
||||
)
|
||||
# Do not send update event for previous bot owner.
|
||||
update_users = update_users - {previous_owner.id}
|
||||
|
||||
# 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}))
|
||||
# Do not send update event for bot_owner.
|
||||
update_users = update_users - {bot_owner.id}
|
||||
|
||||
bot_event = dict(
|
||||
type="realm_bot",
|
||||
op="update",
|
||||
bot=dict(
|
||||
user_id=user_profile.id,
|
||||
owner_id=user_profile.bot_owner.id,
|
||||
),
|
||||
)
|
||||
transaction.on_commit(
|
||||
lambda: send_event(
|
||||
user_profile.realm,
|
||||
bot_event,
|
||||
update_users,
|
||||
)
|
||||
)
|
||||
|
||||
# Since `bot_owner_id` is included in the user profile dict we need
|
||||
# to update the users dict with the new bot owner id
|
||||
event = dict(
|
||||
type="realm_user",
|
||||
op="update",
|
||||
person=dict(
|
||||
user_id=user_profile.id,
|
||||
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))
|
||||
)
|
||||
|
||||
|
||||
@transaction.atomic(durable=True)
|
||||
def do_change_default_sending_stream(
|
||||
user_profile: UserProfile, stream: Optional[Stream], *, acting_user: Optional[UserProfile]
|
||||
) -> None:
|
||||
old_value = user_profile.default_sending_stream_id
|
||||
user_profile.default_sending_stream = stream
|
||||
user_profile.save(update_fields=["default_sending_stream"])
|
||||
|
||||
event_time = timezone_now()
|
||||
RealmAuditLog.objects.create(
|
||||
realm=user_profile.realm,
|
||||
event_type=RealmAuditLog.USER_DEFAULT_SENDING_STREAM_CHANGED,
|
||||
event_time=event_time,
|
||||
modified_user=user_profile,
|
||||
acting_user=acting_user,
|
||||
extra_data=orjson.dumps(
|
||||
{
|
||||
RealmAuditLog.OLD_VALUE: old_value,
|
||||
RealmAuditLog.NEW_VALUE: None if stream is None else stream.id,
|
||||
}
|
||||
).decode(),
|
||||
)
|
||||
|
||||
if user_profile.is_bot:
|
||||
if stream:
|
||||
stream_name: Optional[str] = stream.name
|
||||
else:
|
||||
stream_name = None
|
||||
event = dict(
|
||||
type="realm_bot",
|
||||
op="update",
|
||||
bot=dict(
|
||||
user_id=user_profile.id,
|
||||
default_sending_stream=stream_name,
|
||||
),
|
||||
)
|
||||
transaction.on_commit(
|
||||
lambda: send_event(
|
||||
user_profile.realm,
|
||||
event,
|
||||
bot_owner_user_ids(user_profile),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@transaction.atomic(durable=True)
|
||||
def do_change_default_events_register_stream(
|
||||
user_profile: UserProfile, stream: Optional[Stream], *, acting_user: Optional[UserProfile]
|
||||
) -> None:
|
||||
old_value = user_profile.default_events_register_stream_id
|
||||
user_profile.default_events_register_stream = stream
|
||||
user_profile.save(update_fields=["default_events_register_stream"])
|
||||
|
||||
event_time = timezone_now()
|
||||
RealmAuditLog.objects.create(
|
||||
realm=user_profile.realm,
|
||||
event_type=RealmAuditLog.USER_DEFAULT_REGISTER_STREAM_CHANGED,
|
||||
event_time=event_time,
|
||||
modified_user=user_profile,
|
||||
acting_user=acting_user,
|
||||
extra_data=orjson.dumps(
|
||||
{
|
||||
RealmAuditLog.OLD_VALUE: old_value,
|
||||
RealmAuditLog.NEW_VALUE: None if stream is None else stream.id,
|
||||
}
|
||||
).decode(),
|
||||
)
|
||||
|
||||
if user_profile.is_bot:
|
||||
if stream:
|
||||
stream_name: Optional[str] = stream.name
|
||||
else:
|
||||
stream_name = None
|
||||
|
||||
event = dict(
|
||||
type="realm_bot",
|
||||
op="update",
|
||||
bot=dict(
|
||||
user_id=user_profile.id,
|
||||
default_events_register_stream=stream_name,
|
||||
),
|
||||
)
|
||||
transaction.on_commit(
|
||||
lambda: send_event(
|
||||
user_profile.realm,
|
||||
event,
|
||||
bot_owner_user_ids(user_profile),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@transaction.atomic(durable=True)
|
||||
def do_change_default_all_public_streams(
|
||||
user_profile: UserProfile, value: bool, *, acting_user: Optional[UserProfile]
|
||||
) -> None:
|
||||
old_value = user_profile.default_all_public_streams
|
||||
user_profile.default_all_public_streams = value
|
||||
user_profile.save(update_fields=["default_all_public_streams"])
|
||||
|
||||
event_time = timezone_now()
|
||||
RealmAuditLog.objects.create(
|
||||
realm=user_profile.realm,
|
||||
event_type=RealmAuditLog.USER_DEFAULT_ALL_PUBLIC_STREAMS_CHANGED,
|
||||
event_time=event_time,
|
||||
modified_user=user_profile,
|
||||
acting_user=acting_user,
|
||||
extra_data=orjson.dumps(
|
||||
{
|
||||
RealmAuditLog.OLD_VALUE: old_value,
|
||||
RealmAuditLog.NEW_VALUE: value,
|
||||
}
|
||||
).decode(),
|
||||
)
|
||||
|
||||
if user_profile.is_bot:
|
||||
event = dict(
|
||||
type="realm_bot",
|
||||
op="update",
|
||||
bot=dict(
|
||||
user_id=user_profile.id,
|
||||
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),
|
||||
)
|
||||
)
|
||||
Reference in New Issue
Block a user