clerk-sdk 0.3.1__tar.gz → 0.3.3__tar.gz

{clerk_sdk-0.3.1/clerk_sdk.egg-info → clerk_sdk-0.3.3}/PKG-INFO

@@ -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.3/clerk/decorator/task_decorator.py

@@ -0,0 +1,89 @@
+import pickle
+import traceback
+from typing import Callable, Optional
+from functools import wraps
+
+from clerk.exceptions.exceptions import ApplicationException
+from .models import ClerkCodePayload
+
+input_pkl: str = "/app/data/input/input.pkl"
+output_pkl: str = "/app/data/output/output.pkl"
+
+
+def clerk_code():
+    def decorator(func: Callable[[ClerkCodePayload], ClerkCodePayload]):
+        @wraps(func)
+        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:
+                    with open(input_pkl, "rb") as f:
+                        raw_data = pickle.load(f)
+                    payload = (
+                        ClerkCodePayload.model_validate(raw_data)
+                        if not isinstance(raw_data, ClerkCodePayload)
+                        else raw_data
+                    )
+                except Exception as 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
+            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 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:
+                # 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):
+                raise output
+
+            return output
+
+        return wrapper
+
+    return decorator

clerk_sdk-0.3.3/clerk/exceptions/exceptions.py

@@ -0,0 +1,40 @@
+from typing import Optional
+
+
+class AppBaseException(Exception):
+    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)
+
+        # 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 → clerk_sdk-0.3.3/clerk_sdk.egg-info}/PKG-INFO

@@ -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 → clerk_sdk-0.3.3}/setup.py

@@ -13,7 +13,7 @@ gui_requirements = get_requirements("./clerk/gui_automation")
 
 setup(
     name="clerk-sdk",
-    version="0.3.1",
+    version="0.3.3",
     description="Library for interacting with Clerk",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",

clerk_sdk-0.3.1/clerk/decorator/task_decorator.py

@@ -1,68 +0,0 @@
-import os
-import pickle
-import traceback
-from typing import Callable, Optional
-from functools import wraps
-
-from clerk.exceptions.exceptions import ApplicationException
-from .models import ClerkCodePayload
-
-input_pkl: str = "/app/data/input/input.pkl"
-output_pkl: str = "/app/data/output/output.pkl"
-
-
-def clerk_code():
-    def decorator(func: Callable[[ClerkCodePayload], ClerkCodePayload]):
-        @wraps(func)
-        def wrapper(payload: Optional[ClerkCodePayload] = None) -> ClerkCodePayload:
-            # 1. Load payload from file if not provided
-            use_pickle = False
-            if payload is None:
-                use_pickle = True
-                try:
-                    with open(input_pkl, "rb") as f:
-                        raw_data = pickle.load(f)
-                    payload = (
-                        ClerkCodePayload.model_validate(raw_data)
-                        if not isinstance(raw_data, ClerkCodePayload)
-                        else raw_data
-                    )
-                except Exception as e:
-                    raise RuntimeError(
-                        f"Failed to load and parse input pickle: {e}"
-                    ) from e
-
-            # 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()
-                )
-
-            # 3. write to output.pkl
-            try:
-                if use_pickle:
-                    with open(output_pkl, "wb") as f:
-                        if 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)
-
-            # 4. Raise if error or return result
-            if isinstance(output, Exception):
-                raise output
-
-            return output
-
-        return wrapper
-
-    return decorator

clerk_sdk-0.3.1/clerk/exceptions/exceptions.py

@@ -1,13 +0,0 @@
-from typing import Optional
-
-
-class AppBaseException(Exception):
-    def __init__(self, type_: str, message: str, traceback: Optional[str] = None):
-        super().__init__(message)
-        self.type = type_
-        self.message = message
-        self.traceback = traceback
-
-
-class ApplicationException(AppBaseException):
-    pass