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

zilencer/management/commands/calculate_first_visible_message_id.py

@@ -0,0 +1,37 @@
+from collections.abc import Iterable
+from typing import Any
+
+from django.core.management.base import CommandParser
+from typing_extensions import override
+
+from zerver.lib.management import ZulipBaseCommand, abort_unless_locked
+from zerver.lib.message import maybe_update_first_visible_message_id
+from zerver.models import Realm
+
+
+class Command(ZulipBaseCommand):
+    help = """Calculate the value of first visible message ID and store it in cache"""
+
+    @override
+    def add_arguments(self, parser: CommandParser) -> None:
+        self.add_realm_args(parser)
+        parser.add_argument(
+            "--lookback-hours",
+            type=int,
+            help="Period a bit larger than that of the cron job that runs "
+            "this command so that the lookback periods are sure to overlap.",
+            required=True,
+        )
+
+    @override
+    @abort_unless_locked
+    def handle(self, *args: Any, **options: Any) -> None:
+        target_realm = self.get_realm(options)
+
+        if target_realm is None:
+            realms: Iterable[Realm] = Realm.objects.all().iterator()
+        else:
+            realms = [target_realm]
+
+        for realm in realms:
+            maybe_update_first_visible_message_id(realm, options["lookback_hours"])