apply new state handling to API functions

src/delta_barth/api/common.py

@@ -2,32 +2,26 @@ from __future__ import annotations
 
 import re
 from datetime import datetime as Datetime
-from typing import Final, Never
+from typing import TYPE_CHECKING, Final, Never
 
 import requests
 from pydantic import BaseModel, PositiveInt, SkipValidation
 from requests import Response
 
+from delta_barth._management import STATE_HANDLER
 from delta_barth.constants import HTTP_CURRENT_CONNECTION, KnownDelBarApiErrorCodes
 from delta_barth.errors import (
     ApiConnectionError,
-    UnknownApiErrorCode,
     UnspecifiedRequestType,
 )
 from delta_barth.types import DelBarApiError, HttpRequestTypes
 
+if TYPE_CHECKING:
+    from delta_barth.types import Status
+
 LOGIN_ERROR_CODES_KNOWN: Final[frozenset[int]] = frozenset((400, 401, 409, 500))
 
 
-def _raise_for_unknown_error(
-    resp: Response,
-) -> Never:
-    raise UnknownApiErrorCode(
-        f"Unknown response code for request. Status Code: {resp.status_code}, "
-        f"Content: {resp.text}"
-    )
-
-
 def _assert_login_status() -> None:
     if not HTTP_CURRENT_CONNECTION.logged_in:
         raise ApiConnectionError("Curent session is not logged in")
@@ -73,7 +67,6 @@ class LoginRequest(BaseModel):
 
 class LoginResponse(BaseModel):
     token: str
-    error: DelBarApiError | None = None
 
 
 def login(
@@ -82,7 +75,7 @@ def login(
     password: str,
     database: str,
     mandant: str,
-) -> LoginResponse:
+) -> tuple[LoginResponse, Status]:
     ROUTE: Final[str] = "user/login"
     URL: Final = combine_route(base_url, ROUTE)
 
@@ -99,26 +92,23 @@ def login(
     )
 
     response: LoginResponse
+    status: Status
     if resp.status_code == 200:
         response = LoginResponse(**resp.json())
+        status = STATE_HANDLER.pipe_states.SUCCESS
         HTTP_CURRENT_CONNECTION.add_session_token(response.token)
-    elif resp.status_code in KnownDelBarApiErrorCodes.COMMON.value:
+    else:
+        response = LoginResponse(token="")
         err = DelBarApiError(status_code=resp.status_code, **resp.json())
-        response = LoginResponse(token="", error=err)
-    else:  # pragma: no cover
-        _raise_for_unknown_error(resp)
+        status = STATE_HANDLER.api_error(err)
 
-    return response
+    return response, status
 
 
 # ** logout
-class LogoutResponse(BaseModel):
-    error: DelBarApiError | None = None
-
-
 def logout(
     base_url: str,
-) -> LogoutResponse:
+) -> tuple[None, Status]:
     ROUTE: Final[str] = "user/logout"
     URL: Final = combine_route(base_url, ROUTE)
 
@@ -127,17 +117,16 @@ def logout(
         headers=HTTP_CURRENT_CONNECTION.headers,  # type: ignore
     )
 
-    response: LogoutResponse
+    response = None
+    status: Status
     if resp.status_code == 200:
-        response = LogoutResponse()
+        status = STATE_HANDLER.SUCCESS
         HTTP_CURRENT_CONNECTION.remove_session_token()
-    elif resp.status_code in KnownDelBarApiErrorCodes.COMMON.value:
+    else:
         err = DelBarApiError(status_code=resp.status_code, **resp.json())
-        response = LogoutResponse(error=err)
-    else:  # pragma: no cover
-        _raise_for_unknown_error(resp)
+        status = STATE_HANDLER.api_error(err)
 
-    return response
+    return response, status
 
 
 # ** sales data
@@ -156,14 +145,14 @@ class SalesPrognosisResponseEntry(BaseModel):
 
 class SalesPrognosisResponse(BaseModel):
     daten: tuple[SalesPrognosisResponseEntry, ...]
-    error: DelBarApiError | None = None
+    # error: DelBarApiError | None = None
 
 
 def get_sales_prognosis_data(
     base_url: str,
     company_id: int | None = None,
     start_date: Datetime | None = None,
-) -> SalesPrognosisResponse:
+) -> tuple[SalesPrognosisResponse, Status]:
     _assert_login_status()
     ROUTE: Final[str] = "verkauf/umsatzprognosedaten"
     URL: Final = combine_route(base_url, ROUTE)
@@ -179,12 +168,19 @@ def get_sales_prognosis_data(
     )
 
     response: SalesPrognosisResponse
+    status: Status
     if resp.status_code == 200:
         response = SalesPrognosisResponse(**resp.json())
-    elif resp.status_code in KnownDelBarApiErrorCodes.COMMON.value:  # pragma: no cover
+        status = STATE_HANDLER.SUCCESS
+    else:
+        response = SalesPrognosisResponse(daten=tuple())
         err = DelBarApiError(status_code=resp.status_code, **resp.json())
-        response = SalesPrognosisResponse(daten=tuple(), error=err)
-    else:  # pragma: no cover
-        _raise_for_unknown_error(resp)
+        status = STATE_HANDLER.api_error(err)
 
-    return response
+    # elif resp.status_code in KnownDelBarApiErrorCodes.COMMON.value:  # pragma: no cover
+    #     err = DelBarApiError(status_code=resp.status_code, **resp.json())
+    #     response = SalesPrognosisResponse(daten=tuple(), error=err)
+    # else:  # pragma: no cover
+    #     _raise_for_unknown_error(resp)
+
+    return response, status

src/delta_barth/errors.py

@@ -13,10 +13,6 @@ class UnspecifiedRequestType(Exception):
     """exception raised if for a given API endpoint a not defined operation is requested"""
 
 
-class UnknownApiErrorCode(Exception):
-    """exception raised if for a given request a unknown error response code is transmitted"""
-
-
 class ApiConnectionError(Exception):
     """exception raised if an established connection is needed, but the current session is not connected"""