syntax: improve factories

2019-01-14 15:35:20 +07:00 · 2019-01-14 15:35:20 +07:00 · de79b7aed3
commit de79b7aed3
parent 666cca49e5
1 changed files with 1 additions and 3 deletions
--- a/djangoldp_notification/management/commands/mock_notification.py
+++ b/djangoldp_notification/management/commands/mock_notification.py
@ -8,7 +8,5 @@ class Command(BaseCommand):
        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()
-
+        NotificationFactory.create_batch(options['size'])
        self.stdout.write(self.style.SUCCESS('Successful data mock install'))