mirror of
https://github.com/th30d4y/ExecuTrace.git
synced 2026-05-26 11:35:51 +00:00
w4nn4d13
27 lines
896 B
Python
27 lines
896 B
Python
from __future__ import annotations
|
|
|
|
from typing import Union
|
|
|
|
from exectrace.core.models import Workflow
|
|
from exectrace.storage.factory import get_storage
|
|
from exectrace.storage.json_storage import JsonStorage
|
|
from exectrace.storage.xml_storage import XmlStorage
|
|
|
|
|
|
class WorkflowEditor:
|
|
"""Load, modify, and save workflows programmatically."""
|
|
|
|
def __init__(self, storage: Union[JsonStorage, XmlStorage] | None = None) -> None:
|
|
self.storage = storage or get_storage("json")
|
|
|
|
def load(self, name: str) -> Workflow:
|
|
return self.storage.load_workflow(name)
|
|
|
|
def save(self, workflow: Workflow) -> None:
|
|
self.storage.save_workflow(workflow)
|
|
|
|
def remove_action(self, workflow: Workflow, index: int) -> None:
|
|
if index < 0 or index >= len(workflow.actions):
|
|
raise IndexError("Action index out of range")
|
|
workflow.actions.pop(index)
|