locust 2.31.9.dev36__tar.gz → 2.32.0__tar.gz

{locust-2.31.9.dev36 → locust-2.32.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.31.9.dev36
+Version: 2.32.0
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT
@@ -18,6 +18,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Testing
 Classifier: Topic :: Software Development :: Testing :: Traffic Generation
 Classifier: Topic :: System :: Distributed Computing
@@ -26,7 +27,8 @@ Requires-Dist: Flask-Cors (>=3.0.10)
 Requires-Dist: Flask-Login (>=0.6.3)
 Requires-Dist: Werkzeug (>=2.0.0)
 Requires-Dist: flask (>=2.0.0)
-Requires-Dist: gevent (>=22.10.2)
+Requires-Dist: gevent (>=22.10.2) ; python_full_version <= "3.12.0"
+Requires-Dist: gevent (>=24.10.1) ; python_full_version > "3.13.0"
 Requires-Dist: geventhttpclient (>=2.3.1)
 Requires-Dist: msgpack (>=1.0.0)
 Requires-Dist: psutil (>=5.9.1)

{locust-2.31.9.dev36 → locust-2.32.0}/locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.9.dev36"
+__version__ = "2.32.0"
 version = __version__
-__version_tuple__ = (2, 31, 9, "dev36")
+__version_tuple__ = (2, 32, 0)
 version_tuple = __version_tuple__

{locust-2.31.9.dev36 → locust-2.32.0}/locust/main.py

@@ -309,6 +309,9 @@ def main():
 
     greenlet_exception_handler = greenlet_exception_logger(logger)
 
+    if sys.version_info < (3, 10):
+        logger.warning("Python 3.9 support is deprecated and will be removed soon")
+
     if options.stop_timeout:
         try:
             options.stop_timeout = parse_timespan(options.stop_timeout)

{locust-2.31.9.dev36 → locust-2.32.0}/locust/rpc/zmqrpc.py

@@ -1,22 +1,24 @@
 from locust.exception import RPCError, RPCReceiveError, RPCSendError
 from locust.util.exception_handler import retry
 
+import socket as csocket
+from socket import gaierror, has_dualstack_ipv6
+
 import msgpack.exceptions as msgerr
 import zmq.error as zmqerr
 import zmq.green as zmq
-from urllib3.util.connection import HAS_IPV6
 
 from .protocol import Message
 
 
 class BaseSocket:
-    def __init__(self, sock_type):
+    def __init__(self, sock_type, ipv4_only):
         context = zmq.Context()
         self.socket = context.socket(sock_type)
 
         self.socket.setsockopt(zmq.TCP_KEEPALIVE, 1)
         self.socket.setsockopt(zmq.TCP_KEEPALIVE_IDLE, 30)
-        if HAS_IPV6:
+        if has_dualstack_ipv6() and not ipv4_only:
             self.socket.setsockopt(zmq.IPV6, 1)
 
     @retry()
@@ -60,10 +62,19 @@ class BaseSocket:
     def close(self, linger=None):
         self.socket.close(linger=linger)
 
+    def ipv4_only(self, host, port) -> bool:
+        try:
+            if str(csocket.getaddrinfo(host, port, proto=csocket.IPPROTO_TCP)).find("Family.AF_INET6") == -1:
+                return True
+        except gaierror as e:
+            print(f"Error resolving address: {e}")
+            return False
+        return False
+
 
 class Server(BaseSocket):
     def __init__(self, host, port):
-        BaseSocket.__init__(self, zmq.ROUTER)
+        BaseSocket.__init__(self, zmq.ROUTER, self.ipv4_only(host, port))
         if port == 0:
             self.port = self.socket.bind_to_random_port(f"tcp://{host}")
         else:
@@ -76,6 +87,6 @@ class Server(BaseSocket):
 
 class Client(BaseSocket):
     def __init__(self, host, port, identity):
-        BaseSocket.__init__(self, zmq.DEALER)
+        BaseSocket.__init__(self, zmq.DEALER, self.ipv4_only(host, port))
         self.socket.setsockopt(zmq.IDENTITY, identity.encode())
         self.socket.connect("tcp://%s:%i" % (host, port))