Source code for scipost.management.commands.populate_db

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


from django.core.management.base import BaseCommand

from comments.factories import (
    CommentaryCommentFactory,
    ThesislinkCommentFactory,
    ReplyCommentFactory,
)


[docs]class Command(BaseCommand):
[docs]    def add_arguments(self, parser):
        parser.add_argument(
            "--comments",
            action="store_true",
            dest="comments",
            default=False,
            help="Add 10 Comments",
        )

[docs]    def handle(self, *args, **kwargs):
        if kwargs["comments"]:
            self.create_comments()

[docs]    def create_comments(self):
        CommentaryCommentFactory.create_batch(3)
        ReplyCommentFactory.create_batch(2)
        ThesislinkCommentFactory.create_batch(3)
        self.stdout.write(self.style.SUCCESS("Successfully created 10 Comments."))