Source code for submissions.management.commands.create_submissions

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


from django.core.management.base import BaseCommand

from submissions import factories


[docs]class Command(BaseCommand): help = "Create random Submission objects by using the factories."
[docs] def add_arguments(self, parser): parser.add_argument( "number", action="store", default=0, type=int, help="Number of submissions to add", ) parser.add_argument( "-s", "--status", choices=[ "unassigned", "assigned", "resubmitted", "resubmission", "published", ], action="store", dest="status", default="assigned", help="Current status of the Submission", )
[docs] def handle(self, *args, **kwargs): if kwargs["number"]: self.create_submissions(kwargs["number"], status=kwargs["status"])
[docs] def create_submissions(self, n, status="assigned"): if status == "unassigned": factories.UnassignedSubmissionFactory.create_batch(n) elif status == "assigned": factories.EICassignedSubmissionFactory.create_batch(n) elif status == "resubmitted": factories.ResubmittedSubmissionFactory.create_batch(n) elif status == "resubmission": factories.ResubmissionFactory.create_batch(n) elif status == "published": factories.PublishedSubmissionFactory.create_batch(n) self.stdout.write( self.style.SUCCESS("Successfully created {n} Submissions.".format(n=n)) )