djangoldp-notification/djangoldp_notification/management/commands/mock_notification.py
Matthieu Fesselier aa90880073 feature: factories
2019-01-11 17:46:23 +07:00

15 lines
535 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):
for i in range(0, options['size']):
notif = NotificationFactory.create()
self.stdout.write(self.style.SUCCESS('Successful data mock install'))