clerk-sdk 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

clerk/decorator/task_decorator.py

@@ -1,4 +1,3 @@
-import os
 import pickle
 import traceback
 from typing import Callable, Optional
@@ -17,6 +16,9 @@ def clerk_code():
         def wrapper(payload: Optional[ClerkCodePayload] = None) -> ClerkCodePayload:
             # 1. Load payload from file if not provided
             use_pickle = False
+            output = None
+            error_occurred = False
+            error_info = None
             if payload is None:
                 use_pickle = True
                 try:
@@ -28,34 +30,53 @@ def clerk_code():
                         else raw_data
                     )
                 except Exception as e:
-                    raise RuntimeError(
-                        f"Failed to load and parse input pickle: {e}"
-                    ) from e
+                    error_occurred = True
+                    error_info = ApplicationException(
+                        type_=str(type(e)),
+                        message=f"Failed to load and parse input pickle: {e}",
+                        traceback=traceback.format_exc(),
+                    )
 
             # 2. Execute function
-            try:
-                output = func(payload)
-                if not isinstance(output, ClerkCodePayload):
-                    raise TypeError("Function must return a ClerkCodePayload instance.")
-            except Exception as e:
-                # parse no standard errors into the standard Application Error
-                output = ApplicationException(
-                    type_=str(type(e)), message=str(e), traceback=traceback.format_exc()
-                )
+            if not error_occurred:
+                try:
+                    output = func(payload)
+                    if not isinstance(output, ClerkCodePayload):
+                        raise TypeError(
+                            "Function must return a ClerkCodePayload instance."
+                        )
+                except Exception as e:
+                    error_occurred = True
+                    error_info = ApplicationException(
+                        type_=str(type(e)),
+                        message=str(e),
+                        traceback=traceback.format_exc(),
+                    )
 
             # 3. write to output.pkl
             try:
                 if use_pickle:
                     with open(output_pkl, "wb") as f:
-                        if isinstance(output, Exception):
+                        if error_occurred:
+                            pickle.dump(error_info, f)
+                        elif isinstance(output, Exception):
                             pickle.dump(output, f)
                         else:
                             pickle.dump(output.model_dump(mode="json"), f)
-
             except Exception as e:
-                output = RuntimeError(f"Failed to write output pickle: {str(e)}")
-                with open(output_pkl, "wb") as f:
-                    pickle.dump(output, f)
+                # If writing output.pkl fails, try to write a minimal error
+                try:
+                    with open(output_pkl, "wb") as f:
+                        pickle.dump(
+                            ApplicationException(
+                                type_=str(type(e)),
+                                message=f"Failed to write output pickle: {str(e)}",
+                                traceback=traceback.format_exc(),
+                            ),
+                            f,
+                        )
+                except Exception:
+                    pass  # Last resort: do nothing if even this fails
 
             # 4. Raise if error or return result
             if isinstance(output, Exception):

clerk/exceptions/exceptions.py

@@ -2,12 +2,39 @@ from typing import Optional
 
 
 class AppBaseException(Exception):
-    def __init__(self, type_: str, message: str, traceback: Optional[str] = None):
+    def __init__(
+        self,
+        *args,
+        type_: Optional[str] = None,
+        message: Optional[str] = None,
+        traceback: Optional[str] = None,
+    ):
+        # If called with positional args (e.g., during unpickling or raise("msg")),
+        # treat args[0] as the message.
+        if args and message is None:
+            message = args[0]
+
+        # Always call base Exception with just the message so .args == (message,)
         super().__init__(message)
-        self.type = type_
-        self.message = message
+
+        # Store structured fields
+        self.type = type_ or self.__class__.__name__
+        self.message = message or ""
         self.traceback = traceback
 
+    # (Optional) make pickling round-trip the extra fields explicitly
+    def __reduce__(self):
+        # Reconstruct with message-only (what Exception expects) and restore extras via state
+        return (
+            self.__class__,
+            (self.message,),
+            {"type": self.type, "traceback": self.traceback},
+        )
+
+    def __setstate__(self, state):
+        for k, v in state.items():
+            setattr(self, k, v)
+
 
 class ApplicationException(AppBaseException):
     pass

{clerk_sdk-0.3.1.dist-info → clerk_sdk-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.3.1
+Version: 0.3.3
 Summary: Library for interacting with Clerk
 Home-page: https://github.com/F-ONE-Group/clerk_pypi
 Author: F-ONE Group

{clerk_sdk-0.3.1.dist-info → clerk_sdk-0.3.3.dist-info}/RECORD

@@ -3,9 +3,9 @@ clerk/base.py,sha256=Ayb2zCQr8fqKZM_Qh3nVxVJOQkkYt-eD5VtpPXd1TLs,2669
 clerk/client.py,sha256=6nWWWjGB7a6FNUWsZRM-IoBb_mshRi6g5KX8meMiJ2E,576
 clerk/decorator/__init__.py,sha256=4VCGOFNq7pA7iJS410V_R9eWfjxP7mwxwa4thwaUTf8,39
 clerk/decorator/models.py,sha256=JWFOJuUh3F40iOL0aoNgUC9GRHU3Fcq3GOjSXA7Gwng,394
-clerk/decorator/task_decorator.py,sha256=ZHBNaG5zlRZ6_OaVFgVhBaowCc0Q8YgdjUPSnmOL7r4,2454
+clerk/decorator/task_decorator.py,sha256=6frvOIHTtDjP2SpBMx-KLZgd1EdXHahzPm9DLSxJfjQ,3390
 clerk/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-clerk/exceptions/exceptions.py,sha256=pCG_KwJ8nwtfnljeqWXchd-qNgTeEfD8KQPXeUkN81A,331
+clerk/exceptions/exceptions.py,sha256=aIrmID62HoQo0pLErfX9NR2nopTyf53BpOu4gtqAy-A,1201
 clerk/exceptions/remote_device.py,sha256=R0Vfmk8ejibEFeV29A8Vqst2YA6yxdqEX9lP5wWubgM,134
 clerk/gui_automation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/gui_automation/client.py,sha256=LqrUQtC-4pjUJYDDn_s3t073K066j_p0vf4kXLChBc0,4891
@@ -46,8 +46,8 @@ clerk/models/ui_operator.py,sha256=mKTJUFZgv7PeEt5oys28HVZxHOJsofmRQOcRpqj0dbU,2
 clerk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/utils/logger.py,sha256=vHbp-lUK3W3c5fhyPsp05p9LGyDHj95v8XM_XQ_FlLc,3691
 clerk/utils/save_artifact.py,sha256=94aYkYNVGcSUaSWZmdjiY6Oc-3yCKb2XWCZ56IAXQqk,1158
-clerk_sdk-0.3.1.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
-clerk_sdk-0.3.1.dist-info/METADATA,sha256=trPJodCcT2rWlxBIe9EBEbgoHcAbg7MvnPsHOpjm4Y0,3508
-clerk_sdk-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-clerk_sdk-0.3.1.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
-clerk_sdk-0.3.1.dist-info/RECORD,,
+clerk_sdk-0.3.3.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.3.3.dist-info/METADATA,sha256=nGsbiRaB7DVlftGElyUsH24-U_KRZKlj7GX0X6dghTw,3508
+clerk_sdk-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clerk_sdk-0.3.3.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
+clerk_sdk-0.3.3.dist-info/RECORD,,