fix(atoss): avoid timeout

DlieBG · May 3, 2024 · f0b1054 · f0b1054
1 parent 0b646ac
commit f0b1054
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/api/src/atoss.py b/api/src/atoss.py
@@ -16,7 +16,7 @@
     'Gleitzeit',
 ]
 
-def __init_session() -> Session:
+def init_session() -> Session:
     init_response = requests.get(
         url=__init_url,
     )
@@ -40,6 +40,7 @@ def __init_session() -> Session:
                 for option in init_response.text.splitlines()
                     if option.startswith('[\'zul.sel.Option\'') and 'value' in option
         },
+        cookies=init_response.cookies.get_dict(),
     )
 
 def __call(session: Session, *arguments: CallItem) -> ResponseDto:
@@ -59,6 +60,7 @@ def __call(session: Session, *arguments: CallItem) -> ResponseDto:
                 ),
             ),
         },
+        cookies=session.cookies,
     )
 
     if 'Kombination Personalnummer / PIN-Code falsch' in call_response.text:
@@ -101,9 +103,9 @@ def __call(session: Session, *arguments: CallItem) -> ResponseDto:
         flexi_balance=response['Gleitzeit'],
     )
 
-def get_balance(personal_number: str, pin: str) -> ResponseDto:
+def get_balance(personal_number: str, pin: str, session: Session = init_session()) -> ResponseDto:
     return __call(
-        __init_session(),
+        session,
         CallItem(
             command=CommandEnum.on_change,
             uuid='badgeno',
@@ -124,9 +126,7 @@ def get_balance(personal_number: str, pin: str) -> ResponseDto:
         ),
     )
 
-def post_booking(personal_number: str, pin: str, action: ActionEnum, type: TypeEnum) -> dict:
-    session = __init_session()
-
+def post_booking(personal_number: str, pin: str, action: ActionEnum, type: TypeEnum, session: Session = init_session()) -> dict:
     return __call(
         session,
         CallItem(

diff --git a/api/src/models.py b/api/src/models.py
@@ -6,6 +6,7 @@ class Session(BaseModel):
     jsession: str
     dt: str
     selection_uuids: dict
+    cookies: dict
 
 class CommandEnum(str, Enum):
     on_change = 'onChange'