mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
dde832b158
This commit adds the capability to keep track and remove uploaded files. Unclaimed attachments are files that have been uploaded to the server but are not referred in any messages. A management command to remove old unclaimed files after a week is also included. Tests for getting the file referred in messages are also included.
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from zerver.lib.actions import do_delete_old_unclaimed_attachments
|
|
from zerver.models import Attachment, get_old_unclaimed_attachments
|
|
|
|
class Command(BaseCommand):
|
|
help = """Remove unclaimed attachments from storage older than a supplied
|
|
numerical value indicating the limit of how old the attachment can be.
|
|
One week is taken as the default value."""
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('-w', '--weeks',
|
|
dest='delta_weeks',
|
|
default=1,
|
|
help="Limiting value of how old the file can be.")
|
|
|
|
parser.add_argument('-f', '--for-real',
|
|
dest='for_real',
|
|
action='store_true',
|
|
default=False,
|
|
help="Actually remove the files from the storage.")
|
|
|
|
def handle(self, *args, **options):
|
|
|
|
delta_weeks = options['delta_weeks']
|
|
print("Deleting unclaimed attached files older than %s" % (delta_weeks,))
|
|
print("")
|
|
|
|
# print the list of files that are going to be removed
|
|
old_attachments = get_old_unclaimed_attachments(delta_weeks)
|
|
for old_attachment in old_attachments:
|
|
print("%s created at %s" % (old_attachment.file_name, old_attachment.create_time))
|
|
|
|
print("")
|
|
if not options["for_real"]:
|
|
print("This was a dry run. Pass -f to actually delete.")
|
|
exit(1)
|
|
|
|
do_delete_old_unclaimed_attachments(delta_weeks)
|
|
print("")
|
|
print("Unclaimed Files deleted.")
|