From 7151ee86311c5cda3911a6a6a81c4be6a4f7e794 Mon Sep 17 00:00:00 2001 From: Giuseppe Salvatore Date: Tue, 10 Feb 2026 15:35:29 +0100 Subject: [PATCH] update: update RequestState obj with custom data --- app/src/zcs/core/session/request_state.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/app/src/zcs/core/session/request_state.py b/app/src/zcs/core/session/request_state.py index 8f4d264..1fc7936 100644 --- a/app/src/zcs/core/session/request_state.py +++ b/app/src/zcs/core/session/request_state.py @@ -22,7 +22,8 @@ def __init__( total_pages: Optional[int] = None, job_id: Optional[str] = None, file_id: Optional[str] = None, - attempts: Optional[int] = 0 + attempts: Optional[int] = 0, + custom_data: Optional[dict] = None ): self.__request_id = request_id if request_id else RequestState.generate_op_code(prefix=prefix) @@ -38,6 +39,21 @@ def __init__( self.__job_id = job_id self.__file_id = file_id self.__attempts = attempts + self.__custom_data = custom_data if custom_data else {} + + def getCustomData(self) -> dict: + """ + Get custom data dictionary. + """ + + return self.__custom_data + + def setCustomData(self, custom_data: dict): + """ + Set custom data dictionary. + """ + + self.__custom_data = custom_data def setAttempts(self, attempts: int): """