---

import time
# Функция принимает 5 параметров:
# 1. response: locust.FastResponse (https://docs.locust.io/en/stable/_modules/locust/contrib/fasthttp.html#FastResponse) для HTTP-зарпосов
# и тело ответа для всех асинхронных операций
# 2. Объект операции (то, вы создавали здесь в UI). Формат операции и ее ключи можно посмотреть в итоговом JSON
# 3. Словарь, "переменная": "значение" с локальными переменными потока
# 4. Список созданных листенеров Kafka, WS, gRPC
# 5. Объект класса UserClass
#
# И должна возвращать 4 параметра:
# 1. Успешность операции (None, если успешность должна быть определена стандартными средствами
# В http по коду ответа, в других протоколах всегда успешно))
# 2. id следующей операции (None, если на этом нужно завершить итерацию)
# 3. Словарь, "переменная": "значение" (может быть пустым - {}) для локальных переменных потока
# 4. Словарь, "переменная": "значение" (может быть пустым - {}) для глобальных переменных теста


from typing import NamedTuple, Optional
# from locust.contrib.fasthttp import ResponseContextManager
# from app.clients.class_builder import UserClass
# from app.data_classes.config import OperationTypes
# from app.data_classes.user_vars import UserVar
# from app.users.async_response_readers import AsyncResponseReader, WebSocketResponseReader, KafkaResponseReader


class ReturnFromPostProcessor(NamedTuple):
    is_operation_successful: bool
    next_operation_id: Optional[str]
    local_vars: dict
    global_vars: dict


# def post_processor(response: Union[ResponseContextManager, str, bytes], operation: OperationTypes, user_vars: dict[str, UserVar], listeners: list[AsyncResponseReader], user_class: UserClass, *args, **kwargs) -> ReturnFromPostProcessor:
def post_processor(
    response, operation, user_vars, listener, user_class, *args, **kwargs
) -> ReturnFromPostProcessor:
    
    if hasattr(response, "status_code") and (response.status_code == 200 or response.status_code == 204): # if isinstance(response, ResponseContextManager
        if int(user_vars['iteration'].value) < int(user_vars['total_iterations'].value)-1:        

            pacing = (((float(user_vars['numB'].value)+float(user_vars['numA'].value)) / 2) - 15) / int(user_vars['total_iterations'].value) # для равномерной подачи нагрузки задержка между вызовом операций должна быть равна "pacing" / "общее количество итераций"
                        
            time.sleep(pacing)                          

            return ReturnFromPostProcessor(True, "UC01_POST_spending_WEB", {"iteration": str(int(user_vars['iteration'].value) + 1)}, {})
        else:    
            return ReturnFromPostProcessor(True, None, {"iteration": str(int(user_vars['iteration'].value) * 0)}, {})
    else:
        return ReturnFromPostProcessor(False, None, {"iteration": str(int(user_vars['iteration'].value) * 0)}, {})