From 1cbb1dd7f3e111c31f9bd2d82775ef7898adfcd8 Mon Sep 17 00:00:00 2001
From: Konstantine Tsafatinos <kosta.tsaf@gmail.com>
Date: Mon, 26 Jun 2023 18:02:21 -0400
Subject: [PATCH] add enqueue spam command

---
 skynet/cli.py | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/skynet/cli.py b/skynet/cli.py
index d7e389d..08a5032 100644
--- a/skynet/cli.py
+++ b/skynet/cli.py
@@ -110,6 +110,7 @@ def download():
 @click.option('--step', '-s', default=26)
 @click.option('--seed', '-S', default=None)
 @click.option('--upscaler', '-U', default='x4')
+@click.option('--jobs', '-j', default=1)
 def enqueue(
     account: str,
     permission: str,
@@ -134,12 +135,19 @@ def enqueue(
         })
         binary = ''
 
-        ec, out = cleos.push_action(
-            'telos.gpu', 'enqueue', [account, req, binary, reward], f'{account}@{permission}'
-        )
-
-        print(collect_stdout(out))
-        assert ec == 0
+        if not kwargs['jobs']:
+            ec, out = cleos.push_action(
+                'telos.gpu', 'enqueue', [account, req, binary, reward], f'{account}@{permission}'
+            )
+            print(collect_stdout(out))
+            assert ec == 0
+        else:
+            for i in kwargs['jobs']:
+                ec, out = cleos.push_action(
+                    'telos.gpu', 'enqueue', [account, req, binary, reward], f'{account}@{permission}'
+                )
+                print(collect_stdout(out))
+                assert ec == 0
 
 @skynet.command()
 @click.option('--loglevel', '-l', default='INFO', help='Logging level')