mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +00:00
df001db1a9
Black 23 enforces some slightly more specific rules about empty line counts and redundant parenthesis removal, but the result is still compatible with Black 22. (This does not actually upgrade our Python environment to Black 23 yet.) Signed-off-by: Anders Kaseorg <anders@zulip.com>
89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
from typing import Any
|
|
|
|
from django.conf import settings
|
|
from django.db import migrations
|
|
from django.db.backends.postgresql.schema import BaseDatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
|
|
|
|
def fix_messages(apps: StateApps, schema_editor: BaseDatabaseSchemaEditor) -> None:
|
|
"""Conceptually, this migration cleans up the old NEW_USER_BOT and FEEDBACK_BOT
|
|
UserProfile objects (their implementations were removed long ago).
|
|
|
|
We do this by:
|
|
* Changing their sent messages to have been sent by NOTIFICATION_BOT.
|
|
* Changing their 1:1 PMs to be PMs with NOTIFICATION_BOT and deleting their
|
|
PM recipient object.
|
|
* Deleting any Huddles that involve them (zulip.com data suggests there are none,
|
|
so this is mainly out of caution) and Recipient objects (which will cascade to
|
|
associated Subscription, Message, and UserMessage objects if they exist).
|
|
* Deleting their UserProfile objects.
|
|
|
|
The end result if these users are completely removed, with any
|
|
messages that might have been associated with them transferred
|
|
to NOTIFICATION_BOT to preserve history.
|
|
"""
|
|
UserProfile = apps.get_model("zerver", "UserProfile")
|
|
Huddle = apps.get_model("zerver", "Huddle")
|
|
Subscription = apps.get_model("zerver", "Subscription")
|
|
Recipient = apps.get_model("zerver", "Recipient")
|
|
RECIPIENT_HUDDLE = 3
|
|
Message = apps.get_model("zerver", "Message")
|
|
Realm = apps.get_model("zerver", "Realm")
|
|
|
|
try:
|
|
internal_realm = Realm.objects.get(string_id=settings.SYSTEM_BOT_REALM)
|
|
except Realm.DoesNotExist:
|
|
# Server not initialized, or no system bot realm. Either way, we shouldn't do anything.
|
|
return
|
|
|
|
def get_bot_by_delivery_email(email: str) -> Any:
|
|
return UserProfile.objects.select_related().get(
|
|
delivery_email__iexact=email.strip(), realm=internal_realm
|
|
)
|
|
|
|
notification_bot = get_bot_by_delivery_email(settings.NOTIFICATION_BOT)
|
|
|
|
def fix_messages_by_bot(bot_profile: Any) -> None:
|
|
Message.objects.filter(sender=bot_profile).update(sender=notification_bot)
|
|
Message.objects.filter(recipient=bot_profile.recipient).update(
|
|
recipient=notification_bot.recipient
|
|
)
|
|
|
|
def clean_up_bot(bot_profile: Any) -> None:
|
|
huddle_recipient_ids = Subscription.objects.filter(
|
|
user_profile_id=bot_profile.id, recipient__type=RECIPIENT_HUDDLE
|
|
).values_list("recipient_id", flat=True)
|
|
Huddle.objects.filter(recipient_id__in=huddle_recipient_ids).delete()
|
|
Recipient.objects.filter(id__in=huddle_recipient_ids).delete()
|
|
|
|
personal_recipient_id = bot_profile.recipient_id
|
|
bot_profile.delete()
|
|
Recipient.objects.filter(id=personal_recipient_id).delete()
|
|
|
|
new_user_bot_email = getattr(settings, "NEW_USER_BOT", "new-user-bot@zulip.com")
|
|
try:
|
|
new_user_bot = get_bot_by_delivery_email(new_user_bot_email)
|
|
fix_messages_by_bot(new_user_bot)
|
|
clean_up_bot(new_user_bot)
|
|
except UserProfile.DoesNotExist:
|
|
pass
|
|
|
|
feedback_bot_email = getattr(settings, "FEEDBACK_BOT", "feedback@zulip.com")
|
|
try:
|
|
feedback_bot = get_bot_by_delivery_email(feedback_bot_email)
|
|
fix_messages_by_bot(feedback_bot)
|
|
clean_up_bot(feedback_bot)
|
|
except UserProfile.DoesNotExist:
|
|
pass
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
dependencies = [
|
|
("zerver", "0272_realm_default_code_block_language"),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(fix_messages, reverse_code=migrations.RunPython.noop, elidable=True),
|
|
]
|