use tqdm with publish message

jasonray · Dec 20, 2023 · d3dd9be · d3dd9be
1 parent addd86b
commit d3dd9be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/publish_sample_messages.py b/publish_sample_messages.py
@@ -5,6 +5,7 @@
 import random
 import argparse
 import time
+from tqdm import tqdm
 
 parser = argparse.ArgumentParser(
                     prog='publish_sample_messages',
@@ -45,7 +46,7 @@ def create_random_message(payload: str):
 
 def publish():
     Statman.stopwatch(name='publish-sample-messages.timing', autostart=True)
-    for i in range(0, args.number_of_messages):
+    for i in tqdm(range(0, args.number_of_messages)):
         m = create_random_message(payload=f'HellO WorlD {i}')
         kessel.publish(m)
         Statman.gauge(name='publish-sample-messages.published-messages').increment()

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 statman == 1.4.3
 pulpo-config == 2.0.1
 art == 6.1
-greenstalk==2.0.2
+greenstalk==2.0.2
+tqdm==4.66.0