dex-python-sdk 0.0.1__py3-none-any.whl

dex/errors.py

@@ -0,0 +1,132 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import json as jsonlib
+from httpx import Response
+
+from dex.dex_api.models import (
+    ErrorResponse,
+    ErrorSubStatus,
+    WorkflowGetResponse,
+    WorkflowStatus,
+)
+
+
+class WorkflowDefinitionError(Exception):
+    pass
+
+
+class InvalidArgumentError(Exception):
+    pass
+
+
+class NotRegisteredError(Exception):
+    pass
+
+
+class HttpError(RuntimeError):
+    def __init__(self, status: int, err_resp: ErrorResponse):
+        super().__init__(err_resp.detail)
+        self.sub_status = err_resp.sub_status
+        self.error_resp = err_resp
+        self.status = status
+
+
+class ClientSideError(HttpError):
+    pass
+
+
+class ServerSideError(HttpError):
+    pass
+
+
+class WorkflowStillRunningError(ClientSideError):
+    pass
+
+
+class WorkflowRPCExecutionError(ClientSideError):
+    pass
+
+
+class WorkflowRPCAcquiringLockFailure(ClientSideError):
+    pass
+
+
+class WorkflowAlreadyStartedError(ClientSideError):
+    pass
+
+
+class WorkflowNotExistsError(ClientSideError):
+    pass
+
+
+def process_http_error(status: int, err_resp: ErrorResponse) -> HttpError:
+    if 400 <= status < 500:
+        if err_resp.sub_status == ErrorSubStatus.WORKFLOW_ALREADY_STARTED_SUB_STATUS:
+            return WorkflowAlreadyStartedError(status, err_resp)
+        elif err_resp.sub_status == ErrorSubStatus.WORKFLOW_NOT_EXISTS_SUB_STATUS:
+            return WorkflowNotExistsError(status, err_resp)
+        else:
+            return ClientSideError(status, err_resp)
+    else:
+        return ServerSideError(status, err_resp)
+
+
+class WorkflowAbnormalExitError(RuntimeError):
+    def __init__(self, get_response: WorkflowGetResponse):
+        self.run_id = get_response.workflow_run_id
+        self.workflow_status = get_response.workflow_status
+        self.error_type = get_response.error_type
+        self.error_message = get_response.error_message
+        # TODO add methods to decode the state results into objects
+        self._state_results = get_response.results
+
+
+class WorkflowFailed(WorkflowAbnormalExitError):
+    pass
+
+
+class WorkflowTimeout(WorkflowAbnormalExitError):
+    pass
+
+
+class WorkflowTerminated(WorkflowAbnormalExitError):
+    pass
+
+
+class WorkflowCanceled(WorkflowAbnormalExitError):
+    pass
+
+
+def process_workflow_abnormal_exit_error(
+    get_response: WorkflowGetResponse,
+) -> WorkflowAbnormalExitError:
+    status = get_response.workflow_status
+    if status == WorkflowStatus.CANCELED:
+        return WorkflowCanceled(get_response)
+    elif status == WorkflowStatus.FAILED:
+        return WorkflowFailed(get_response)
+    elif status == WorkflowStatus.TERMINATED:
+        return WorkflowTerminated(get_response)
+    elif status == WorkflowStatus.TIMEOUT:
+        return WorkflowTimeout(get_response)
+    return WorkflowAbnormalExitError(get_response)
+
+
+def parse_unexpected_error(err) -> ErrorResponse:
+    try:
+        response = Response(err.status_code, content=err.content)
+        return ErrorResponse.from_dict(response.json())
+    except Exception:
+        return ErrorResponse.from_dict(jsonlib.loads(err.content))