stepfunction 0.0.2__tar.gz → 0.0.3__tar.gz

{stepfunction-0.0.2 → stepfunction-0.0.3}/LICENSE

@@ -18,4 +18,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE.

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

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: stepfunction
-Version: 0.0.2
+Version: 0.0.3
 Summary: Step Function Workflow Orchestration Library
 Author: Vineeth Penugonda
+License-Expression: MIT
 Project-URL: Homepage, https://github.com/vinecodes/stepfunction
 Project-URL: Issues, https://github.com/vinecodes/stepfunction/issues
 Project-URL: Blog_Post, https://blog.vineethp.com/posts/introducingstepfunction/
@@ -11,7 +12,6 @@ Classifier: Programming Language :: Python :: 3
 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.9
 Description-Content-Type: text/markdown
@@ -65,4 +65,3 @@ This project is licensed under the MIT License - see the LICENSE file for detail
 
 ## Author
 Created and maintained by **Vineeth Penugonda**.
-

{stepfunction-0.0.2 → stepfunction-0.0.3}/README.md

@@ -44,4 +44,3 @@ This project is licensed under the MIT License - see the LICENSE file for detail
 
 ## Author
 Created and maintained by **Vineeth Penugonda**.
-

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

@@ -1,18 +1,18 @@
 [project]
 name = "stepfunction"
-version = "0.0.2"
+version = "0.0.3"
 authors = [{ name = "Vineeth Penugonda" }]
 description = "Step Function Workflow Orchestration Library"
 readme = "README.md"
 keywords = ["StepFunction", "Workflow", "Orchestration", "Library"]
 requires-python = ">=3.9"
 dependencies = ["graphviz === 0.20.3"]
+license = "MIT"
 classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.9",
     "Operating System :: OS Independent",
     "Development Status :: 3 - Alpha",
-    "License :: OSI Approved :: MIT License",
     "Topic :: Software Development :: Libraries"
 ]
 
@@ -20,6 +20,10 @@ classifiers = [
 requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 
+[tool.ruff.lint]
+select = ["E", "F", "I"]
+ignore = ["E501"]
+
 [tool.setuptools.packages.find]
 where = ["src"]
 include = ["stepfunction.*"]

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

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: stepfunction
-Version: 0.0.2
+Version: 0.0.3
 Summary: Step Function Workflow Orchestration Library
 Author: Vineeth Penugonda
+License-Expression: MIT
 Project-URL: Homepage, https://github.com/vinecodes/stepfunction
 Project-URL: Issues, https://github.com/vinecodes/stepfunction/issues
 Project-URL: Blog_Post, https://blog.vineethp.com/posts/introducingstepfunction/
@@ -11,7 +12,6 @@ Classifier: Programming Language :: Python :: 3
 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.9
 Description-Content-Type: text/markdown
@@ -65,4 +65,3 @@ This project is licensed under the MIT License - see the LICENSE file for detail
 
 ## Author
 Created and maintained by **Vineeth Penugonda**.
-

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

@@ -26,5 +26,4 @@ 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
-tests/test_car_purchase_workflow.py
+src/stepfunction/utils/utils.py

{stepfunction-0.0.2 → stepfunction-0.0.3}/src/stepfunction/constants/visualizer.py

@@ -44,4 +44,4 @@ DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_SHAPE = "boxed"
 """str: The default node shape for sub-step functions in the visualizer."""
 
 DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_STYLE = "dotted"
-""" str: The default node style for sub-step functions in the visualizer."""
+""" str: The default node style for sub-step functions in the visualizer."""

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

@@ -1,3 +1,3 @@
 from .step_function import StepFunction
 
-__all__ = ["StepFunction"]
+__all__ = ["StepFunction"]

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

@@ -1,16 +1,18 @@
-""" Module to define the StepFunction class. 
+"""Module to define the StepFunction class.
 
-    Author: Vineeth Penugonda
+Author: Vineeth Penugonda
 """
 
-
+from asyncio import run as asyncio_run
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from inspect import iscoroutinefunction
 from typing import Any, Callable, Dict, Optional, Union
 
 from stepfunction.constants.enums import StepFunctionStatus
-from stepfunction.exceptions.step_errors import (ParallelStepExecutionError,
-                                                 StepExecutionError)
+from stepfunction.exceptions.step_errors import (
+    ParallelStepExecutionError,
+    StepExecutionError,
+)
 from stepfunction.types.step_types import StepParams
 from stepfunction.utils.logger import setup_logger
 
@@ -90,7 +92,7 @@ class StepFunction:
     Parallel Example:
         step_function.add_step("Step1", func1, parallel=True)
         step_function.add_step("ParallelStep", {
-            "task1": func2, 
+            "task1": func2,
             "task2": func3
         }, next_step="Step2", parallel=True)
 
@@ -102,7 +104,7 @@ class StepFunction:
 
     Status Example:
         status = step_function.status  # Will be StepFunctionStatus.INITIALIZED, StepFunctionStatus.RUNNING, StepFunctionStatus.COMPLETED, or StepFunctionStatus.FAILED.
-"""
+    """
 
     def __init__(self, name: str):
         self.__name = name  # Name of the step function
@@ -119,7 +121,8 @@ class StepFunction:
         self.__logger = setup_logger(__name__)
 
         self.__logger.debug(
-            f"StepFunction - {self.__name} - Status - {self.__status.value}")
+            f"StepFunction - {self.__name} - Status - {self.__status.value}"
+        )
 
     def add_step(
         self,
@@ -145,7 +148,13 @@ class StepFunction:
             "stop_on_failure": stop_on_failure,
         }
 
-    def add_sub_step_function(self, name: str, sub_step_function: "StepFunction", next_step: Optional[str] = None, on_failure: Optional[str] = None):
+    def add_sub_step_function(
+        self,
+        name: str,
+        sub_step_function: "StepFunction",
+        next_step: Optional[str] = None,
+        on_failure: Optional[str] = None,
+    ):
         """Add a sub-step function to the workflow."""
 
         if name in self.__steps:
@@ -167,7 +176,7 @@ class StepFunction:
             "branch": None,
             "parallel": False,
             "stop_on_failure": False,
-            "is_sub_step_function": True
+            "is_sub_step_function": True,
         }
 
     def set_start_step(self, name: str):
@@ -183,7 +192,8 @@ class StepFunction:
         self.__status = StepFunctionStatus.RUNNING
 
         self.__logger.debug(
-            f"StepFunction - {self.__name} - Status - {self.__status.value}")
+            f"StepFunction - {self.__name} - Status - {self.__status.value}"
+        )
 
         self.__last_result = initial_input
 
@@ -192,7 +202,8 @@ class StepFunction:
             try:
                 if step["parallel"]:
                     results = self._execute_parallel(
-                        step["func"], step["stop_on_failure"])
+                        step["func"], step["stop_on_failure"]
+                    )
 
                     self.__last_result = results
                     self.__context[self.__current_step] = results
@@ -206,14 +217,17 @@ class StepFunction:
                     self.__last_result = result
                     self.__context[self.__current_step] = result
 
-                    self.__logger.info(
-                        f"Step '{self.__current_step}' succeeded"
-                    )
+                    self.__logger.info(f"Step '{self.__current_step}' succeeded")
 
-                if step['branch'] and result in step['branch']:
-                    self.__current_step = step['branch'][result]
-                else:
-                    self.__current_step = step["next_step"]
+                next_step = None
+
+                if step["branch"]:
+                    if callable(step["branch"]):
+                        next_step = step["branch"](self.__last_result)
+                    else:
+                        next_step = step["branch"].get(self.__last_result)
+
+                self.__current_step = next_step or step["next_step"]
 
             except Exception as exc:
                 self.__logger.exception(
@@ -233,7 +247,8 @@ class StepFunction:
                     self.__status = StepFunctionStatus.FAILED
 
                     self.__logger.debug(
-                        f"StepFunction - {self.__name} - Status - {self.__status.value}")
+                        f"StepFunction - {self.__name} - Status - {self.__status.value}"
+                    )
                 else:
                     self.__logger.exception(
                         f"No failure step defined for '{self.__current_step}'. Raising Exception."
@@ -242,7 +257,8 @@ class StepFunction:
                     self.__status = StepFunctionStatus.FAILED
 
                     self.__logger.debug(
-                        f"StepFunction - {self.__name} - Status - {self.__status.value}")
+                        f"StepFunction - {self.__name} - Status - {self.__status.value}"
+                    )
 
                     raise StepExecutionError(exc)
 
@@ -252,24 +268,34 @@ class StepFunction:
                 self.__status = StepFunctionStatus.COMPLETED
 
                 self.__logger.debug(
-                    f"StepFunction - {self.__name} - Status - {self.__status.value}")
+                    f"StepFunction - {self.__name} - Status - {self.__status.value}"
+                )
 
     async def _execute_step(self, func: Callable, input_value: Any):
-        """ Execute a single step, handling async functions. """
+        """Execute a single step, handling async functions."""
         if iscoroutinefunction(func):
             return await func(input_value)
         else:
             return func(input_value)
 
-    def _execute_parallel(self, func_dict: Dict[str, Callable[[Any], Any]], stop_on_failure: bool = False):
+    def _execute_parallel(
+        self, func_dict: Dict[str, Callable[[Any], Any]], stop_on_failure: bool = False
+    ):
         """Execute the steps in parallel."""
         results = {}
         errors = []
         should_stop_execution = False
 
+        def _run(func, arg):
+            if iscoroutinefunction(func):
+                return asyncio_run(func(arg))
+            return func(arg)
+
         with ThreadPoolExecutor() as executor:
-            futures = {executor.submit(
-                func, self.__last_result): step_name for step_name, func in func_dict.items()}
+            futures = {
+                executor.submit(_run, func, self.__last_result): step_name
+                for step_name, func in func_dict.items()
+            }
 
             for future in as_completed(futures):
                 step_name = futures[future]
@@ -282,7 +308,8 @@ class StepFunction:
                     results[step_name] = result
                 except Exception as exc:
                     self.__logger.exception(
-                        f"Parallel task '{step_name}' failed: {exc}")
+                        f"Parallel task '{step_name}' failed: {exc}"
+                    )
 
                     results[step_name] = exc.args[0]
 
@@ -290,6 +317,8 @@ class StepFunction:
 
                     if stop_on_failure:
                         should_stop_execution = True
+                        for f in futures:
+                            f.cancel()
 
         if errors:
             self.__logger.error(f"Some parallel tasks failed: {errors}")
@@ -316,8 +345,7 @@ 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."""
@@ -333,32 +361,27 @@ class StepFunction:
 
     @property
     def name(self):
-        """ Returns the name of the step function. 
-        """
+        """Returns the name of the step function."""
         return self.__name
 
     @property
     def steps(self):
-        """ Returns the steps of the step function. 
-        """
+        """Returns the steps of the step function."""
         return self.__steps
 
     @property
     def last_result(self):
-        """ Returns the result of the last step.
-        """
+        """Returns the result of the last step."""
         return self.__last_result
 
     @property
     def context(self):
-        """ Returns the context of the step function.
-        """
+        """Returns the context of the step function."""
         return self.__context
 
     @property
     def status(self):
-        """ Returns the status of the step function.
-        """
+        """Returns the status of the step function."""
         return self.__status
 
     def __str__(self):

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

@@ -1,3 +1,3 @@
 from .visualizer import Visualizer
 
-__all__ = ["Visualizer"]
+__all__ = ["Visualizer"]

stepfunction-0.0.3/src/stepfunction/core/visualizer/visualizer.py

@@ -0,0 +1,148 @@
+"""This module contains the visualizer class for the graph model.
+
+Author: Vineeth Penugonda
+"""
+
+from os import getcwd
+
+from graphviz import Digraph
+
+from stepfunction.constants.visualizer import (
+    DEFAULT_VISUALIZER_EXTENSION,
+    DEFAULT_VISUALIZER_FAILURE_EDGE_COLOR,
+    DEFAULT_VISUALIZER_FAILURE_EDGE_LABEL,
+    DEFAULT_VISUALIZER_FOLDER,
+    DEFAULT_VISUALIZER_FORMAT,
+    DEFAULT_VISUALIZER_PARALLEL_STEP_EDGE_STYLE,
+    DEFAULT_VISUALIZER_RENDERER,
+    DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_COLOR,
+    DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_LABEL,
+    DEFAULT_VISUALIZER_STRING_ENCODING,
+    DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_SHAPE,
+    DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_STYLE,
+    DEFAULT_VISUALIZER_SUCCESS_EDGE_LABEL,
+)
+from stepfunction.types.step_types import StepParams
+from stepfunction.types.visualizer_types import RenderStepFunctionParams
+
+
+class Visualizer:
+    """This class is responsible for visualizing the graph model."""
+
+    def __init__(self, graph_name: str, steps: StepParams = None):
+        """Initializes the visualizer."""
+
+        self.graph_name = graph_name
+        self.__steps = steps
+
+        self.__output_file_name = None
+        self.__output_file_path = None
+
+        self.__dot = Digraph(comment=self.graph_name)
+
+    def visualize_step_function(self):
+        """Visualizes the graph model."""
+
+        if not self.__steps:
+            raise ValueError("No steps found to visualize.")
+
+        for step_name, step_info in self.__steps.items():
+            if (
+                "is_sub_step_function" in step_info
+                and step_info["is_sub_step_function"]
+            ):
+                self.__dot.node(
+                    step_name,
+                    step_name,
+                    shape=DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_SHAPE,
+                    style=DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_STYLE,
+                )
+            else:
+                self.__dot.node(step_name, step_name)
+
+            if step_info["next_step"]:
+                self.__dot.edge(
+                    step_name,
+                    step_info["next_step"],
+                    label=DEFAULT_VISUALIZER_SUCCESS_EDGE_LABEL,
+                )
+
+            if step_info["on_failure"]:
+                failure_edge_label = DEFAULT_VISUALIZER_FAILURE_EDGE_LABEL
+                failure_edge_color = DEFAULT_VISUALIZER_FAILURE_EDGE_COLOR
+
+                if step_info.get("stop_on_failure"):
+                    failure_edge_label = DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_LABEL
+                    failure_edge_color = DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_COLOR
+
+                self.__dot.edge(
+                    step_name,
+                    step_info["on_failure"],
+                    label=failure_edge_label,
+                    color=failure_edge_color,
+                )
+
+            if step_info["parallel"]:
+                # func is a dictionary for parallel steps
+                parallel_function_names = step_info["func"]
+
+                with self.__dot.subgraph() as s:
+                    s.attr(rank="same")
+
+                    for parallel_step_name, func in parallel_function_names.items():
+                        self.__dot.node(parallel_step_name, parallel_step_name)
+                        self.__dot.edge(
+                            step_name,
+                            parallel_step_name,
+                            style=DEFAULT_VISUALIZER_PARALLEL_STEP_EDGE_STYLE,
+                        )
+
+                        if step_info["next_step"]:
+                            self.__dot.edge(parallel_step_name, step_info["next_step"])
+
+            if step_info.get("branch"):
+                for result, next_step in step_info["branch"].items():
+                    self.__dot.edge(step_name, next_step, label=f"Branch: {result}")
+
+    def render_step_function(self, **kwargs: RenderStepFunctionParams):
+        """Renders the graph model."""
+        current_dir = getcwd()
+
+        format = kwargs.get("format", DEFAULT_VISUALIZER_FORMAT)
+        renderer = kwargs.get("renderer", DEFAULT_VISUALIZER_RENDERER)
+
+        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}"
+        )
+
+        self.__output_file_path = file_path
+        self.__output_file_name = file_name
+
+        self.__dot.render(
+            filename=f"{self.__output_file_path}/{self.__output_file_name}",
+            format=format,
+            renderer=renderer,
+        )
+
+    def render_step_function_to_string(self, **kwargs: RenderStepFunctionParams):
+        """Renders the graph model as a string."""
+
+        format = kwargs.get("format", DEFAULT_VISUALIZER_FORMAT)
+        renderer = kwargs.get("renderer", DEFAULT_VISUALIZER_RENDERER)
+
+        return self.__dot.pipe(format=format, renderer=renderer).decode(
+            DEFAULT_VISUALIZER_STRING_ENCODING
+        )
+
+    @property
+    def output_file_name(self):
+        """Returns the output file name."""
+        return self.__output_file_name
+
+    @property
+    def output_file_path(self):
+        """Returns the output file path."""
+        return self.__output_file_path

{stepfunction-0.0.2 → stepfunction-0.0.3}/src/stepfunction/types/step_types.py

@@ -1,12 +1,11 @@
-from typing import Any, Callable, Dict, Optional, TypedDict
+from typing import Any, Callable, Dict, Optional, TypedDict, Union
 
 
 class StepParams(TypedDict, total=False):
-
     func: Callable[[Any], Any]
     next_step: Optional[str]
     on_failure: Optional[str]
-    branch: Optional[Dict[Any, str]]
+    branch: Optional[Union[Dict[Any, str], Callable[[Any], Optional[str]]]]
     parallel: bool
     stop_on_failure: bool
     is_sub_step_function: bool

{stepfunction-0.0.2 → stepfunction-0.0.3}/src/stepfunction/types/visualizer_types.py

@@ -1,7 +1,8 @@
-from typing import TypedDict, Optional
+from typing import Optional, TypedDict
+
 
 class RenderStepFunctionParams(TypedDict, total=False):
     file_path: Optional[str]
     file_name: Optional[str]
     format: str
-    renderer: str
+    renderer: str

{stepfunction-0.0.2 → stepfunction-0.0.3}/src/stepfunction/utils/constants.py

@@ -1,4 +1,4 @@
-""" Constants for the project. """
+"""Constants for the project."""
 
 # Logging
 DEFAULT_LOG_LEVEL = "INFO"
@@ -9,4 +9,4 @@ DEFAULT_LOGGING_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
 
 # Environment variables
 ENVIRONMENT_VARIABLE_LOG_LEVEL = "LOG_LEVEL"
-""" str: The environment variable for the logging level."""
+""" str: The environment variable for the logging level."""

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

@@ -1,24 +1,28 @@
-""" Module for setting up loggers. """
+"""Module for setting up loggers."""
 
 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.constants import (
+    DEFAULT_LOG_LEVEL,
+    DEFAULT_LOGGING_FORMAT,
+    ENVIRONMENT_VARIABLE_LOG_LEVEL,
+)
 from stepfunction.utils.utils import get_environment_variable
 
 
-def setup_logger(name: Optional[str] = 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.
     """
 
     LOG_LEVEL = get_environment_variable(
-        ENVIRONMENT_VARIABLE_LOG_LEVEL, DEFAULT_LOG_LEVEL)
+        ENVIRONMENT_VARIABLE_LOG_LEVEL, DEFAULT_LOG_LEVEL
+    )
 
     logging_config = {
         "version": 1,
@@ -44,7 +48,6 @@ def setup_logger(name: Optional[str] = None, log_format: str = DEFAULT_LOGGING_F
                 "handlers": ["console"],
                 "propagate": False,
             },
-
             "botocore": {
                 "level": "WARNING",  # Set level to WARNING to ignore DEBUG logs
                 "handlers": ["console"],

stepfunction-0.0.3/src/stepfunction/utils/utils.py

@@ -0,0 +1,18 @@
+"""This module contains utility functions for the stepfunction package."""
+
+from os import getenv
+from typing import Optional
+
+
+def get_environment_variable(name: str, default: Optional[str] = None) -> Optional[str]:
+    """
+    Returns the value of the environment variable with the given name.
+
+    Args:
+        name (str): The name of the environment variable.
+        default (str): The default value to return if the environment variable is not set.
+
+    Returns:
+        str: The value of the environment variable.
+    """
+    return getenv(name, default)

stepfunction-0.0.2/src/stepfunction/core/visualizer/visualizer.py

@@ -1,129 +0,0 @@
-""" This module contains the visualizer class for the graph model.
-
-    Author: Vineeth Penugonda
-"""
-
-from os import getcwd
-
-from graphviz import Digraph
-
-from stepfunction.constants.visualizer import (
-    DEFAULT_VISUALIZER_EXTENSION, DEFAULT_VISUALIZER_FAILURE_EDGE_COLOR,
-    DEFAULT_VISUALIZER_FAILURE_EDGE_LABEL, DEFAULT_VISUALIZER_FOLDER,
-    DEFAULT_VISUALIZER_FORMAT, DEFAULT_VISUALIZER_PARALLEL_STEP_EDGE_STYLE,
-    DEFAULT_VISUALIZER_RENDERER, DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_COLOR,
-    DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_LABEL,
-    DEFAULT_VISUALIZER_STRING_ENCODING,
-    DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_SHAPE,
-    DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_STYLE,
-    DEFAULT_VISUALIZER_SUCCESS_EDGE_LABEL)
-from stepfunction.types.step_types import StepParams
-from stepfunction.types.visualizer_types import RenderStepFunctionParams
-
-
-class Visualizer:
-    """ This class is responsible for visualizing the graph model.
-    """
-
-    def __init__(self, graph_name: str, steps: StepParams = None):
-        """ Initializes the visualizer."""
-
-        self.graph_name = graph_name
-        self.__steps = steps
-
-        self.__output_file_name = None
-        self.__output_file_path = None
-
-        self.__dot = Digraph(comment=self.graph_name)
-
-    def visualize_step_function(self):
-        """ Visualizes the graph model.
-        """
-
-        if not self.__steps:
-            raise ValueError("No steps found to visualize.")
-
-        for step_name, step_info in self.__steps.items():
-
-            if 'is_sub_step_function' in step_info and step_info['is_sub_step_function']:
-                self.__dot.node(step_name, step_name, shape=DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_SHAPE,
-                                style=DEFAULT_VISUALIZER_SUB_STEP_FUNCTION_NODE_STYLE)
-            else:
-                self.__dot.node(step_name, step_name)
-
-            if step_info['next_step']:
-                self.__dot.edge(
-                    step_name, step_info['next_step'], label=DEFAULT_VISUALIZER_SUCCESS_EDGE_LABEL)
-
-            if step_info['on_failure']:
-                failure_edge_label = DEFAULT_VISUALIZER_FAILURE_EDGE_LABEL
-                failure_edge_color = DEFAULT_VISUALIZER_FAILURE_EDGE_COLOR
-
-                if step_info.get('stop_on_failure'):
-                    failure_edge_label = DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_LABEL
-                    failure_edge_color = DEFAULT_VISUALIZER_STOP_ON_FAILURE_EDGE_COLOR
-
-                self.__dot.edge(
-                    step_name, step_info['on_failure'], label=failure_edge_label, color=failure_edge_color)
-
-            if step_info['parallel']:
-
-                # func is a dictionary for parallel steps
-                parallel_function_names = step_info['func']
-
-                with self.__dot.subgraph() as s:
-                    s.attr(rank='same')
-
-                    for parallel_step_name, func in parallel_function_names.items():
-                        self.__dot.node(parallel_step_name, parallel_step_name)
-                        self.__dot.edge(
-                            step_name, parallel_step_name, style=DEFAULT_VISUALIZER_PARALLEL_STEP_EDGE_STYLE)
-
-                        if step_info['next_step']:
-                            self.__dot.edge(parallel_step_name,
-                                            step_info['next_step'])
-
-            if step_info.get('branch'):
-                for result, next_step in step_info['branch'].items():
-                    self.__dot.edge(step_name, next_step,
-                                    label=f"Branch: {result}")
-
-    def render_step_function(self, **kwargs: RenderStepFunctionParams):
-        """ Renders the graph model.
-        """
-        current_dir = getcwd()
-
-        format = kwargs.get('format', DEFAULT_VISUALIZER_FORMAT)
-        renderer = kwargs.get('renderer', DEFAULT_VISUALIZER_RENDERER)
-
-        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}")
-
-        self.__output_file_path = file_path
-        self.__output_file_name = file_name
-
-        self.__dot.render(
-            filename=f"{self.__output_file_path}/{self.__output_file_name}", format=format, renderer=renderer)
-
-    def render_step_function_to_string(self, **kwargs: RenderStepFunctionParams):
-        """ Renders the graph model as a string.
-        """
-
-        format = kwargs.get('format', DEFAULT_VISUALIZER_FORMAT)
-        renderer = kwargs.get('renderer', DEFAULT_VISUALIZER_RENDERER)
-
-        return self.__dot.pipe(format=format, renderer=renderer).decode(DEFAULT_VISUALIZER_STRING_ENCODING)
-
-    @property
-    def output_file_name(self):
-        """ Returns the output file name.
-        """
-        return self.__output_file_name
-
-    @property
-    def output_file_path(self):
-        """ Returns the output file path.
-        """
-        return self.__output_file_path

stepfunction-0.0.2/src/stepfunction/utils/utils.py

@@ -1,17 +0,0 @@
-""" This module contains utility functions for the stepfunction package. """
-
-from os import getenv
-
-
-def get_environment_variable(name: str, default: str = None) -> str:
-    """
-        Returns the value of the environment variable with the given name.
-
-        Args:
-            name (str): The name of the environment variable.
-            default (str): The default value to return if the environment variable is not set.
-
-        Returns:
-            str: The value of the environment variable.
-    """
-    return getenv(name, default)

stepfunction-0.0.2/tests/test_car_purchase_workflow.py

@@ -1,103 +0,0 @@
-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()