use tqdm with publish message

beanstalk-stats.py

@@ -0,0 +1,24 @@
+import sys
+from loguru import logger
+from pulpo_messaging.kessel import Message
+from pulpo_messaging.kessel import Pulpo
+from pulpo_messaging.kessel import PulpoConfig
+from statman import Statman
+import random
+import argparse
+import time
+from tqdm import tqdm
+
+parser = argparse.ArgumentParser(
+                    prog='publish_sample_messages',
+                    description='Published a set of test messages to kessel')
+
+parser.add_argument('--config', type=str, help='path to config file')
+args = parser.parse_args()
+
+config = PulpoConfig().fromJsonFile(args.config).fromArgumentParser(args)
+pulpo = Pulpo(config)
+pulpo.initialize_queue_adapter()
+
+
+Statman.report(log_method=logger.info)

publish_sample_messages.py

@@ -7,6 +7,7 @@
 import random
 import argparse
 import time
+from tqdm import tqdm
 
 parser = argparse.ArgumentParser(
                     prog='publish_sample_messages',
@@ -55,7 +56,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()

requirements.txt

@@ -2,4 +2,5 @@ statman == 1.4.5
 pulpo-config == 2.0.1
 art == 6.1
 greenstalk==2.0.2
-loguru==0.7.0
+tqdm==4.66.0
+loguru==0.7.0