stepfunction 0.0.1__tar.gz → 0.0.2__tar.gz

{stepfunction-0.0.1/src/StepFunction.egg-info → stepfunction-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: stepfunction
-Version: 0.0.1
+Version: 0.0.2
 Summary: Step Function Workflow Orchestration Library
 Author: Vineeth Penugonda
 Project-URL: Homepage, https://github.com/vinecodes/stepfunction
@@ -8,15 +8,16 @@ Project-URL: Issues, https://github.com/vinecodes/stepfunction/issues
 Project-URL: Blog_Post, https://blog.vineethp.com/posts/introducingstepfunction/
 Keywords: StepFunction,Workflow,Orchestration,Library
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.9
 Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.12
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: graphviz===0.20.3
+Dynamic: license-file
 
 # StepFunction
 

{stepfunction-0.0.1 → stepfunction-0.0.2}/pyproject.toml

@@ -1,15 +1,15 @@
 [project]
 name = "stepfunction"
-version = "0.0.1"
+version = "0.0.2"
 authors = [{ name = "Vineeth Penugonda" }]
 description = "Step Function Workflow Orchestration Library"
 readme = "README.md"
 keywords = ["StepFunction", "Workflow", "Orchestration", "Library"]
-requires-python = ">=3.12"
+requires-python = ">=3.9"
 dependencies = ["graphviz === 0.20.3"]
 classifiers = [
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.9",
     "Operating System :: OS Independent",
     "Development Status :: 3 - Alpha",
     "License :: OSI Approved :: MIT License",

{stepfunction-0.0.1 → stepfunction-0.0.2/src/StepFunction.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: stepfunction
-Version: 0.0.1
+Version: 0.0.2
 Summary: Step Function Workflow Orchestration Library
 Author: Vineeth Penugonda
 Project-URL: Homepage, https://github.com/vinecodes/stepfunction
@@ -8,15 +8,16 @@ Project-URL: Issues, https://github.com/vinecodes/stepfunction/issues
 Project-URL: Blog_Post, https://blog.vineethp.com/posts/introducingstepfunction/
 Keywords: StepFunction,Workflow,Orchestration,Library
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.9
 Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 3 - Alpha
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.12
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: graphviz===0.20.3
+Dynamic: license-file
 
 # StepFunction
 

{stepfunction-0.0.1 → stepfunction-0.0.2}/src/StepFunction.egg-info/SOURCES.txt

@@ -26,4 +26,5 @@ src/stepfunction/types/visualizer_types.py
 src/stepfunction/utils/__init__.py
 src/stepfunction/utils/constants.py
 src/stepfunction/utils/logger.py
-src/stepfunction/utils/utils.py
+src/stepfunction/utils/utils.py
+tests/test_car_purchase_workflow.py

{stepfunction-0.0.1 → stepfunction-0.0.2}/src/stepfunction/core/step_function/step_function.py

@@ -9,7 +9,6 @@ from inspect import iscoroutinefunction
 from typing import Any, Callable, Dict, Optional, Union
 
 from stepfunction.constants.enums import StepFunctionStatus
-from stepfunction.core.visualizer import Visualizer
 from stepfunction.exceptions.step_errors import (ParallelStepExecutionError,
                                                  StepExecutionError)
 from stepfunction.types.step_types import StepParams
@@ -199,8 +198,7 @@ class StepFunction:
                     self.__context[self.__current_step] = results
 
                     self.__logger.info(
-                        f"Parallel step '{
-                            self.__current_step}' succeeded with results: {results}"
+                        f"Parallel step '{self.__current_step}' succeeded with results: {results}"
                     )
                 else:
                     result = await self._execute_step(step["func"], self.__last_result)
@@ -226,8 +224,7 @@ class StepFunction:
 
                 if step["on_failure"]:
                     self.__logger.exception(
-                        f"Executing failure step: {
-                            step['on_failure']} for '{self.__current_step}'"
+                        f"Executing failure step: {step['on_failure']} for '{self.__current_step}'"
                     )
 
                     self.__current_step = step["on_failure"]
@@ -239,8 +236,7 @@ class StepFunction:
                         f"StepFunction - {self.__name} - Status - {self.__status.value}")
                 else:
                     self.__logger.exception(
-                        f"No failure step defined for '{
-                            self.__current_step}'. Raising Exception."
+                        f"No failure step defined for '{self.__current_step}'. Raising Exception."
                     )
 
                     self.__status = StepFunctionStatus.FAILED
@@ -306,6 +302,8 @@ class StepFunction:
 
     def visualize(self):
         """Visualize the workflow."""
+        from stepfunction.core.visualizer import Visualizer
+
         visualizer = Visualizer(self.__name, self.__steps)
 
         self.__logger.debug("Visualizing the step function")
@@ -318,11 +316,13 @@ class StepFunction:
 
         output_file_name = visualizer.output_file_name
 
-        self.__logger.debug(f"Rendered the step function to file: {
-            output_file_name}")
+        self.__logger.debug(
+            f"Rendered the step function to file: {output_file_name}")
 
     def visualize_to_string(self):
         """Visualize the workflow as a string."""
+        from stepfunction.core.visualizer import Visualizer
+
         visualizer = Visualizer(self.__name, self.__steps)
 
         self.__logger.debug("Visualizing the step function")

{stepfunction-0.0.1 → stepfunction-0.0.2}/src/stepfunction/core/visualizer/visualizer.py

@@ -98,8 +98,8 @@ class Visualizer:
 
         file_path = kwargs.get(
             'file_path', f"{current_dir}/{DEFAULT_VISUALIZER_FOLDER}")
-        file_name = kwargs.get('file_name', f"{self.graph_name}.{
-                               DEFAULT_VISUALIZER_EXTENSION}")
+        file_name = kwargs.get(
+            'file_name', f"{self.graph_name}.{DEFAULT_VISUALIZER_EXTENSION}")
 
         self.__output_file_path = file_path
         self.__output_file_name = file_name

{stepfunction-0.0.1 → stepfunction-0.0.2}/src/stepfunction/utils/logger.py

@@ -3,13 +3,15 @@
 import logging
 import logging.config
 
+from typing import Optional
+
 from stepfunction.utils.constants import (DEFAULT_LOG_LEVEL,
                                           DEFAULT_LOGGING_FORMAT,
                                           ENVIRONMENT_VARIABLE_LOG_LEVEL)
 from stepfunction.utils.utils import get_environment_variable
 
 
-def setup_logger(name: str | None = None, log_format: str = DEFAULT_LOGGING_FORMAT) -> logging.Logger:
+def setup_logger(name: Optional[str] = None, log_format: str = DEFAULT_LOGGING_FORMAT) -> logging.Logger:
     """
     Set up and return a logger with the given name.
     If no name is provided, return the root logger.

stepfunction-0.0.2/tests/test_car_purchase_workflow.py

@@ -0,0 +1,103 @@
+from pathlib import Path
+import sys
+import types
+import unittest
+
+
+ROOT = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT / "src"))
+sys.path.insert(0, str(ROOT))
+
+
+class _FakeSubgraph:
+    def __init__(self):
+        self.attrs = {}
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc, tb):
+        return False
+
+    def attr(self, **kwargs):
+        self.attrs.update(kwargs)
+
+
+class _FakeDigraph:
+    def __init__(self, comment=None):
+        self.comment = comment
+        self.nodes = []
+        self.edges = []
+        self.render_calls = []
+
+    def node(self, *args, **kwargs):
+        self.nodes.append((args, kwargs))
+
+    def edge(self, *args, **kwargs):
+        self.edges.append((args, kwargs))
+
+    def subgraph(self):
+        return _FakeSubgraph()
+
+    def render(self, *args, **kwargs):
+        self.render_calls.append((args, kwargs))
+
+    def pipe(self, format=None, renderer=None):
+        return f"digraph {self.comment} ({format}, {renderer})".encode("utf-8")
+
+
+fake_graphviz = types.ModuleType("graphviz")
+fake_graphviz.Digraph = _FakeDigraph
+sys.modules.setdefault("graphviz", fake_graphviz)
+
+
+from examples.car_purchase_workflow import validate_car_purchase_transaction_workflow
+from stepfunction.constants.enums import StepFunctionStatus
+
+
+class CarPurchaseWorkflowTests(unittest.IsolatedAsyncioTestCase):
+    async def test_valid_car_purchase_workflow_runs_to_completion(self):
+        workflow = await validate_car_purchase_transaction_workflow(
+            transaction={
+                "transaction_id": "CAR-1001",
+                "customer_email": "buyer@example.com",
+                "is_valid": True,
+            },
+            visualize=True,
+        )
+
+        self.assertEqual(workflow.status, StepFunctionStatus.COMPLETED)
+        self.assertEqual(workflow.last_result["workflow"], "done")
+        self.assertEqual(
+            workflow.context["Update_Company_DB_Records"]["update_shipping_db_record"],
+            "shipping_updated",
+        )
+        self.assertEqual(
+            workflow.context["Update_Car_Purchase_Transaction_Status"]["status"],
+            "completed",
+        )
+        self.assertTrue(
+            workflow.context["Send_Notification_To_User_Email"]["notification_sent"])
+
+    async def test_invalid_car_purchase_skips_parallel_update_steps(self):
+        workflow = await validate_car_purchase_transaction_workflow(
+            transaction={
+                "transaction_id": "CAR-1002",
+                "customer_email": "buyer@example.com",
+                "is_valid": False,
+            },
+            visualize=False,
+        )
+
+        self.assertEqual(workflow.status, StepFunctionStatus.COMPLETED)
+        self.assertEqual(
+            workflow.context["Check_If_Car_Purchase_Transaction_Is_Valid"], "invalid")
+        self.assertNotIn("Update_Company_DB_Records", workflow.context)
+        self.assertEqual(
+            workflow.context["Send_Notification_To_User_Email"]["status"],
+            "validation_failed",
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()