2019-02-14 13:42:04 +01:00
|
|
|
from typing import Any
|
|
|
|
|
|
|
|
from django.conf import settings
|
2020-01-14 21:59:46 +01:00
|
|
|
from django.core.management.base import BaseCommand, CommandError, \
|
|
|
|
CommandParser
|
2019-02-14 13:42:04 +01:00
|
|
|
|
|
|
|
from zerver.lib.transfer import transfer_uploads_to_s3
|
|
|
|
|
2020-01-14 21:59:46 +01:00
|
|
|
|
2019-02-14 13:42:04 +01:00
|
|
|
class Command(BaseCommand):
|
|
|
|
help = """Transfer uploads to S3 """
|
|
|
|
|
|
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
|
|
parser.add_argument('--processes',
|
|
|
|
dest='processes',
|
|
|
|
action="store",
|
|
|
|
default=6,
|
|
|
|
help='Processes to use for exporting uploads in parallel')
|
|
|
|
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
|
|
num_processes = int(options['processes'])
|
|
|
|
if num_processes < 1:
|
|
|
|
raise CommandError('You must have at least one process.')
|
|
|
|
|
|
|
|
if not settings.LOCAL_UPLOADS_DIR:
|
|
|
|
raise CommandError('Please set the value of LOCAL_UPLOADS_DIR.')
|
|
|
|
|
|
|
|
transfer_uploads_to_s3(num_processes)
|
|
|
|
print("Transfer to S3 completed successfully.")
|