populate_db: Add --extra-bots option to create extra bots.

skapil · Feb 11, 2017 · 7e8d3d9 · 7e8d3d9
1 parent b867ac3
commit 7e8d3d9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/zilencer/management/commands/populate_db.py b/zilencer/management/commands/populate_db.py
@@ -52,6 +52,12 @@ def add_arguments(self, parser):
                             default=0,
                             help='The number of extra users to create')
 
+        parser.add_argument('--extra-bots',
+                            dest='extra_bots',
+                            type=int,
+                            default=0,
+                            help='The number of extra bots to create')
+
         parser.add_argument('--huddles',
                             dest='num_huddles',
                             type=int,
@@ -234,6 +240,8 @@ def handle(self, **options):
                 ("Zulip Error Bot", "[email protected]"),
                 ("Zulip Default Bot", "[email protected]"),
             ]
+            for i in range(options["extra_bots"]):
+                zulip_realm_bots.append(('Extra Bot %d' % (i,), 'extrabot%[email protected]' % (i,)))
             zulip_realm_bots.extend(all_realm_bots)
             create_users(zulip_realm, zulip_realm_bots, bot_type=UserProfile.DEFAULT_BOT)