13 lines
499 B
Python
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'))
|