atomresponder/management/commands/request_atom_resync.py

from django.conf import settings
from django.core.management.base import BaseCommand
import logging
logger = logging.getLogger(__name__)


class Command(BaseCommand):
    help = "Request a resend of the specific atom ID. This will cause a new version to be created in pluto."

    def add_arguments(self, parser):
        parser.add_argument("id", type=str, help="UUID of the atom to request a resend of")

    def handle(self, *args, **options):
        from atomresponder.media_atom import request_atom_resend

        if "id" not in options or options["id"]=="":
            print("ERROR: You must specify an atom ID as the first argument")
            exit(1)

        #this will raise if it fails, and the user will see the error
        request_atom_resend(options["id"], settings.ATOM_TOOL_HOST, settings.ATOM_TOOL_SECRET)
        print("Resend request to {0} successful".format(settings.ATOM_TOOL_HOST))