clerk-sdk 0.3.2__tar.gz → 0.3.4__tar.gz

{clerk_sdk-0.3.2/clerk_sdk.egg-info → clerk_sdk-0.3.4}/PKG-INFO

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

clerk_sdk-0.3.4/clerk/decorator/task_decorator.py

@@ -0,0 +1,92 @@
+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
+                # Write a placeholder output file in case of early failure                
+                with open(output_pkl, "wb") as f:
+                    pickle.dump({"error": "Early failure"}, f)                
+                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.2 → clerk_sdk-0.3.4/clerk_sdk.egg-info}/PKG-INFO

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

{clerk_sdk-0.3.2 → clerk_sdk-0.3.4}/setup.py

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

clerk_sdk-0.3.2/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