Skip to content

Commit 05b0df8

Browse files
authored
Merge pull request #25 from zcsadmin/add-custom-data-to-RequestState-obj
update: update RequestState obj with custom data
2 parents b522578 + 7151ee8 commit 05b0df8

1 file changed

Lines changed: 17 additions & 1 deletion

File tree

app/src/zcs/core/session/request_state.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,8 @@ def __init__(
2222
total_pages: Optional[int] = None,
2323
job_id: Optional[str] = None,
2424
file_id: Optional[str] = None,
25-
attempts: Optional[int] = 0
25+
attempts: Optional[int] = 0,
26+
custom_data: Optional[dict] = None
2627
):
2728

2829
self.__request_id = request_id if request_id else RequestState.generate_op_code(prefix=prefix)
@@ -38,6 +39,21 @@ def __init__(
3839
self.__job_id = job_id
3940
self.__file_id = file_id
4041
self.__attempts = attempts
42+
self.__custom_data = custom_data if custom_data else {}
43+
44+
def getCustomData(self) -> dict:
45+
"""
46+
Get custom data dictionary.
47+
"""
48+
49+
return self.__custom_data
50+
51+
def setCustomData(self, custom_data: dict):
52+
"""
53+
Set custom data dictionary.
54+
"""
55+
56+
self.__custom_data = custom_data
4157

4258
def setAttempts(self, attempts: int):
4359
"""

0 commit comments

Comments
 (0)