djangoldp-notification/djangoldp_notification/management/commands/mock_notification.py
2019-01-14 15:35:20 +07:00

13 lines
499 B
Python

from django.core.management.base import BaseCommand, CommandError
from djangoldp_notification.factories import NotificationFactory
class Command(BaseCommand):
help = 'Mock data'
def add_arguments(self, parser):
parser.add_argument('--size', type=int, default=0, help='Number of notifications to create')
def handle(self, *args, **options):
NotificationFactory.create_batch(options['size'])
self.stdout.write(self.style.SUCCESS('Successful data mock install'))