def archive_stream_messages(realm: Realm,
chunk_size: int = MESSAGE_BATCH_SIZE) -> None:
logger.info("Archiving stream messages for realm " + realm.string_id)
# We don't archive, if the stream has message_retention_days set to -1,
# or if neither the stream nor the realm have a retention policy.
streams = Stream.objects.filter(realm_id=realm.id).exclude(
message_retention_days=-1)
if not realm.message_retention_days:
streams = streams.exclude(message_retention_days__isnull=True)
retention_policy_dict = {} # type: Dict[int, int]
for stream in streams:
# if stream.message_retention_days is null, use the realm's policy
if stream.message_retention_days:
retention_policy_dict[stream.id] = stream.message_retention_days
else:
assert realm.message_retention_days is not None
retention_policy_dict[stream.id] = realm.message_retention_days
recipients = get_stream_recipients([stream.id for stream in streams])
message_count = 0
for recipient in recipients:
message_count += archive_messages_by_recipient(
recipient, retention_policy_dict[recipient.type_id], realm,
chunk_size)
logger.info("Done. Archived {} messages.".format(message_count))
def archive_stream_messages() -> None:
# We don't archive, if the stream has message_retention_days set to -1,
# or if neither the stream nor the realm have a retention policy.
streams = Stream.objects.exclude(message_retention_days=-1).filter(
Q(message_retention_days__isnull=False) | Q(realm__message_retention_days__isnull=False)
)
retention_policy_dict = {} # type: Dict[int, int]
for stream in streams:
# if stream.message_retention_days is null, use the realm's policy
if stream.message_retention_days:
retention_policy_dict[stream.id] = stream.message_retention_days
else:
retention_policy_dict[stream.id] = stream.realm.message_retention_days
recipients = get_stream_recipients([stream.id for stream in streams])
for recipient in recipients:
archive_messages_by_recipient(recipient, retention_policy_dict[recipient.type_id])