Support of cloudwatch metrics - fix setup and usage of missing_extra

SvenskaSpel · Oct 4, 2023 · 7ce7902 · 7ce7902
1 parent c2ab076
commit 7ce7902
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/examples/cloudwatch_listener_ex.py b/examples/cloudwatch_listener_ex.py
@@ -1,16 +1,11 @@
-import traceback
-import logging
-import sys
 from locust import main, events
 from locust_plugins.listeners.cloudwatch import CloudwatchAdapter, ServiceContext
-
+from locust_plugins import missing_extra
 
 try:
     import boto3
 except ModuleNotFoundError:
-    traceback.print_exc()
-    logging.error("boto3 is not installed by default, you need to install it using 'pip install boto3'")
-    sys.exit(1)
+    missing_extra("boto3", "boto3")
 
 
 @events.init.add_listener

diff --git a/setup.py b/setup.py
@@ -13,6 +13,7 @@
     "mqtt": ["paho-mqtt>=1.5.0"],
     "appinsights": ["opencensus-ext-azure"],
     "resource": ["lxml"],
+    "boto3":["boto3"],
 }
 
 setup(