snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/migrations/0032_verify_all_medium_avatar_images.py

@@ -0,0 +1,39 @@
+import hashlib
+from unittest.mock import patch
+
+from django.conf import settings
+from django.db import migrations
+from django.db.backends.base.schema import BaseDatabaseSchemaEditor
+from django.db.migrations.state import StateApps
+
+from zerver.lib.upload import ensure_avatar_image
+from zerver.models import UserProfile
+
+
+# We hackishly patch this function in order to revert it to the state
+# it had when this migration was first written.  This is a balance
+# between copying in a historical version of hundreds of lines of code
+# from zerver.lib.upload (which would pretty annoying, but would be a
+# pain) and just using the current version, which doesn't work
+# since we rearranged the avatars in Zulip 1.6.
+def patched_user_avatar_path(user_profile: UserProfile) -> str:
+    email = user_profile.email
+    user_key = email.lower() + settings.AVATAR_SALT
+    return hashlib.sha1(user_key.encode()).hexdigest()
+
+
+@patch("zerver.lib.upload.user_avatar_path", patched_user_avatar_path)
+def verify_medium_avatar_image(apps: StateApps, schema_editor: BaseDatabaseSchemaEditor) -> None:
+    user_profile_model = apps.get_model("zerver", "UserProfile")
+    for user_profile in user_profile_model.objects.filter(avatar_source="U"):
+        ensure_avatar_image(user_profile, medium=True)
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("zerver", "0031_remove_system_avatar_source"),
+    ]
+
+    operations = [
+        migrations.RunPython(verify_medium_avatar_image, elidable=True),
+    ]