LoadTestShape about invokust HOT 3 OPEN

def lambda_handler(event, context=None):
    try:
        if event:
            settings = create_settings(**event)
        else:
            settings = create_settings(from_environment=True)

        loadtest = LocustLoadTest(settings)
        loadtest.run()
    except Exception as e:
        logging.error("Locust exception {0}".format(repr(e)))

    else:
        locust_stats = loadtest.stats()
        lambda_runtime_info = get_lambda_runtime_info(context)
        loadtest_results = locust_stats.copy()
        loadtest_results.update(lambda_runtime_info)
        json_results = json.dumps(loadtest_results)

        logging.info(json_results)
        return json_results

class LoadTest(HttpUser):
    wait_time = between(0, 1)
    @task
    def query(self):
        self.client.get("/api")

class StagesShapeWithCustomUsers(LoadTestShape):
    stages = [
        {"duration": 10, "users": 100, "spawn_rate": 10, "user_classes": [LoadTest]},
        {"duration": 20, "users": 150, "spawn_rate": 10, "user_classes": [LoadTest]},
        {"duration": 30, "users": 200, "spawn_rate": 10, "user_classes": [LoadTest]}
    ]

    def tick(self):
        run_time = self.get_run_time()
        for stage in self.stages:
            if run_time < stage["duration"]:
                try:
                    tick_data = (stage["users"], stage["spawn_rate"], stage["user_classes"])
                except KeyError:
                    tick_data = (stage["users"], stage["spawn_rate"])
                return tick_data
        return None