Ruff enable isort (with custom sections)

.git-blame-ignore-revs

@@ -9,3 +9,5 @@
 b5324820b299b1fe7da0608f0cc8ec47f58b1e40
 # upgrade code style to 3.8 using pyupgrade
 60f3bceacc4ab9567433d40ae3ed280750f55ff1
+# sort imports using ruff
+f99e9df700a8020e4c1967eb42dcb37ddd26e296

benchmarks/dispatch.py

@@ -5,16 +5,16 @@
 by people working on Locust's development.
 """
 
+from locust import User
+from locust.dispatch import UsersDispatcher
+from locust.runners import WorkerNode
+
 import itertools
 import statistics
 import time
 
 from prettytable import PrettyTable
 
-from locust import User
-from locust.dispatch import UsersDispatcher
-from locust.runners import WorkerNode
-
 
 class User1(User):
     weight = 5

docs/conf.py

@@ -7,11 +7,12 @@
 # All configuration values have a default value; values that are commented out
 # serve to show the default value.
 
+from locust.argument_parser import get_empty_argument_parser, setup_parser_arguments
+
 import os
 import subprocess
-import sphinx_rtd_theme
 
-from locust.argument_parser import get_empty_argument_parser, setup_parser_arguments
+import sphinx_rtd_theme
 
 
 # Run command `locust --help` and store output in cli-help-output.txt which is included in the docs

examples/add_command_line_argument.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, task, events
+from locust import HttpUser, events, task
 
 
 @events.init_command_line_parser.add_listener

examples/basic.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, TaskSet, task, between
+from locust import HttpUser, TaskSet, between, task
 
 
 def index(l):

examples/browse_docs_sequence_test.py

@@ -1,8 +1,10 @@
 # This locust test script example will simulate a user
 # browsing the Locust documentation on https://docs.locust.io/
 
+from locust import HttpUser, SequentialTaskSet, between, task
+
 import random
-from locust import HttpUser, SequentialTaskSet, task, between
+
 from pyquery import PyQuery
 
 

examples/browse_docs_test.py

@@ -1,8 +1,10 @@
 # This locust test script example will simulate a user
 # browsing the Locust documentation on https://docs.locust.io/
 
+from locust import HttpUser, TaskSet, between, task
+
 import random
-from locust import HttpUser, TaskSet, task, between
+
 from pyquery import PyQuery
 
 

examples/custom_messages.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, task, events, between
+from locust import HttpUser, between, events, task
 from locust.runners import MasterRunner, WorkerRunner
 
 usernames = []

examples/custom_shape/double_wave.py

@@ -1,6 +1,6 @@
+from locust import HttpUser, LoadTestShape, TaskSet, constant, task
+
 import math
-from locust import HttpUser, TaskSet, task, constant
-from locust import LoadTestShape
 
 
 class UserTasks(TaskSet):

examples/custom_shape/stages.py

@@ -1,5 +1,4 @@
-from locust import HttpUser, TaskSet, task, constant
-from locust import LoadTestShape
+from locust import HttpUser, LoadTestShape, TaskSet, constant, task
 
 
 class UserTasks(TaskSet):

examples/custom_shape/staging_user_classes.py

@@ -1,5 +1,4 @@
-from locust import HttpUser, TaskSet, task, constant
-from locust import LoadTestShape
+from locust import HttpUser, LoadTestShape, TaskSet, constant, task
 
 
 class UserTasks(TaskSet):

examples/custom_shape/step_load.py

@@ -1,6 +1,6 @@
+from locust import HttpUser, LoadTestShape, TaskSet, constant, task
+
 import math
-from locust import HttpUser, TaskSet, task, constant
-from locust import LoadTestShape
 
 
 class UserTasks(TaskSet):

examples/custom_shape/wait_user_count.py

@@ -1,10 +1,9 @@
-from collections import namedtuple
+from locust import HttpUser, LoadTestShape, TaskSet, constant, task
+
 import math
-import time
 import random
-
-from locust import HttpUser, TaskSet, task, constant
-from locust import LoadTestShape
+import time
+from collections import namedtuple
 
 
 class UserTasks(TaskSet):

examples/custom_wait_function.py

@@ -1,4 +1,5 @@
 from locust import HttpUser, TaskSet, task
+
 import random
 
 

examples/custom_xmlrpc_client/xmlrpc_locustfile.py

@@ -1,8 +1,8 @@
-import time
-from xmlrpc.client import ServerProxy, Fault
-
 from locust import User, task
 
+import time
+from xmlrpc.client import Fault, ServerProxy
+
 
 class XmlRpcClient(ServerProxy):
     """

examples/debugging.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, task, run_single_user
+from locust import HttpUser, run_single_user, task
 
 
 class QuickstartUser(HttpUser):

examples/debugging_advanced.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, task, run_single_user
+from locust import HttpUser, run_single_user, task
 from locust.exception import StopUser
 
 

examples/dynamic_user_credentials.py

@@ -1,6 +1,6 @@
 # locustfile.py
 
-from locust import HttpUser, TaskSet, task, between
+from locust import HttpUser, TaskSet, between, task
 
 USER_CREDENTIALS = [
     ("user1", "password"),

examples/events.py

@@ -3,8 +3,7 @@
 track the sum of the content-length header in all successful HTTP responses
 """
 
-from locust import HttpUser, TaskSet, task, web, between
-from locust import events
+from locust import HttpUser, TaskSet, between, events, task, web
 
 
 class MyTaskSet(TaskSet):

examples/extend_modern_web_ui.py

@@ -4,12 +4,14 @@
 successful HTTP responses and display them in the web UI.
 """
 
-import os
+from locust import HttpUser, TaskSet, between, events, task, web
+
 import json
-from time import time
+import os
 from html import escape
-from locust import HttpUser, TaskSet, task, web, between, events
-from flask import Blueprint, render_template, jsonify, make_response, request
+from time import time
+
+from flask import Blueprint, jsonify, make_response, render_template, request
 
 
 class MyTaskSet(TaskSet):

examples/extend_web_ui/extend.py

@@ -4,11 +4,13 @@
 successful HTTP responses and display them in the web UI.
 """
 
+from locust import HttpUser, TaskSet, between, events, task, web
+
 import os
-from time import time
 from html import escape
-from locust import HttpUser, TaskSet, task, web, between, events
-from flask import Blueprint, render_template, jsonify, make_response
+from time import time
+
+from flask import Blueprint, jsonify, make_response, render_template
 
 
 class MyTaskSet(TaskSet):

examples/grpc/grpc_user.py

@@ -1,10 +1,12 @@
+from locust import User
+from locust.exception import LocustError
+
 import time
 from typing import Any, Callable
+
 import grpc
 import grpc.experimental.gevent as grpc_gevent
 from grpc_interceptor import ClientInterceptor
-from locust import User
-from locust.exception import LocustError
 
 # patch grpc so that it uses gevent instead of asyncio
 grpc_gevent.init_gevent()

examples/grpc/hello_pb2_grpc.py

@@ -1,7 +1,6 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
-
 import hello_pb2 as hello__pb2
 
 

examples/grpc/hello_server.py

@@ -1,9 +1,10 @@
-import hello_pb2_grpc
-import hello_pb2
-import grpc
-from concurrent import futures
 import logging
 import time
+from concurrent import futures
+
+import grpc
+import hello_pb2
+import hello_pb2_grpc
 
 logger = logging.getLogger(__name__)
 

examples/grpc/locustfile.py

@@ -1,9 +1,10 @@
+from locust import events, task
+
 import gevent
 import grpc_user
 import hello_pb2
 import hello_pb2_grpc
 from hello_server import start_server
-from locust import events, task
 
 
 # Start the dummy server. This is not something you would do in a real test.

examples/locustfile.py

@@ -1,5 +1,6 @@
+from locust import HttpUser, between, task
+
 import time
-from locust import HttpUser, task, between
 
 
 class QuickstartUser(HttpUser):

examples/manual_stats_reporting.py

@@ -17,11 +17,11 @@ def my_task(self):
        # If an Exception is raised, it'll be reported as a failure
 """
 
-import random
-from contextlib import contextmanager, ContextDecorator
-from time import time, sleep
+from locust import User, constant, events, task
 
-from locust import User, task, constant, events
+import random
+from contextlib import ContextDecorator, contextmanager
+from time import sleep, time
 
 
 @contextmanager

examples/multiple_hosts.py

@@ -1,8 +1,8 @@
-import os
-
-from locust import HttpUser, TaskSet, task, between
+from locust import HttpUser, TaskSet, between, task
 from locust.clients import HttpSession
 
+import os
+
 
 class MultipleHostsUser(HttpUser):
     abstract = True

examples/nested_inline_tasksets.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, TaskSet, task, between
+from locust import HttpUser, TaskSet, between, task
 
 
 class WebsiteUser(HttpUser):

examples/rest.py

@@ -1,7 +1,8 @@
-from contextlib import contextmanager
-from locust import task, run_single_user, FastHttpUser
+from locust import FastHttpUser, run_single_user, task
 from locust.contrib.fasthttp import RestResponseContextManager
 from locust.user.wait_time import constant
+
+from contextlib import contextmanager
 from typing import Generator
 
 

examples/sdk_session_patching/session_patch_locustfile.py

@@ -1,5 +1,6 @@
 import locust
 from locust.user import task
+
 from archivist.archivist import Archivist  # Example library under test
 
 

examples/semaphore_wait.py

@@ -1,4 +1,4 @@
-from locust import HttpUser, TaskSet, task, events, between
+from locust import HttpUser, TaskSet, between, events, task
 
 from gevent.lock import Semaphore
 

examples/stop_on_threshold.py

@@ -1,7 +1,9 @@
 # An example of how to stop locust if a threshold (in this case the fail ratio) is exceeded
-from locust import task, HttpUser, events
-from locust.runners import STATE_STOPPING, STATE_STOPPED, STATE_CLEANUP, WorkerRunner
+from locust import HttpUser, events, task
+from locust.runners import STATE_CLEANUP, STATE_STOPPED, STATE_STOPPING, WorkerRunner
+
 import time
+
 import gevent
 
 

examples/terraform/aws/plan/basic.py

@@ -1,5 +1,6 @@
+from locust import HttpUser, between, task
+
 import time
-from locust import HttpUser, task, between
 
 
 class Quickstart(HttpUser):

examples/test_data_management.py

@@ -16,13 +16,14 @@
 #
 # try it out by running:
 #  locust -f test_data_management.py --headless -u 2 -t 5
-from locust.user.wait_time import constant
-from locust import HttpUser, task
-from locust import events
+from locust import HttpUser, events, task
 from locust.runners import MasterRunner
-import requests
+from locust.user.wait_time import constant
+
 import datetime
 
+import requests
+
 
 def timestring():
     now = datetime.datetime.now()

examples/use_as_lib.py

@@ -1,9 +1,10 @@
 #!/usr/bin/env python3
-import gevent
-from locust import HttpUser, task, events
+from locust import HttpUser, events, task
 from locust.env import Environment
-from locust.stats import stats_printer, stats_history
 from locust.log import setup_logging
+from locust.stats import stats_history, stats_printer
+
+import gevent
 
 setup_logging("INFO", None)
 

examples/web_ui_auth.py

@@ -6,10 +6,12 @@
 
 For more information, see https://docs.locust.io/en/stable/extending-locust.html#authentication
 """
-import os
-import json
 from locust import HttpUser, events
-from flask import url_for, Blueprint, request, make_response, redirect, session
+
+import json
+import os
+
+from flask import Blueprint, make_response, redirect, request, session, url_for
 from flask_login import UserMixin, login_user