ddeutil-workflow 0.0.78__py3-none-any.whl → 0.0.80__py3-none-any.whl

ddeutil/workflow/__about__.py

@@ -1 +1 @@
-__version__: str = "0.0.78"
+__version__: str = "0.0.80"

ddeutil/workflow/__init__.py

@@ -52,7 +52,7 @@ from .__types import DictData, DictStr, Matrix, Re, TupleStr
 from .audits import (
     Audit,
     FileAudit,
-    get_audit_model,
+    get_audit,
 )
 from .conf import *
 from .errors import (
@@ -132,11 +132,7 @@ from .stages import (
     VirtualPyStage,
 )
 from .traces import (
-    BaseTrace,
-    FileTrace,
-    Trace,
-    TraceData,
-    TraceMeta,
+    TraceManager,
     get_trace,
 )
 from .utils import *

ddeutil/workflow/api/routes/job.py

@@ -15,7 +15,7 @@ from fastapi.responses import UJSONResponse
 from ...__types import DictData
 from ...errors import JobError
 from ...job import Job
-from ...traces import Trace, get_trace
+from ...traces import TraceManager, get_trace
 from ...utils import gen_id
 
 logger = logging.getLogger("uvicorn.error")
@@ -41,7 +41,7 @@ async def job_execute(
     if extras:
         job.extras = extras
 
-    trace: Trace = get_trace(
+    trace: TraceManager = get_trace(
         run_id, parent_run_id=parent_run_id, extras=job.extras
     )
 

ddeutil/workflow/api/routes/logs.py

@@ -10,7 +10,7 @@ from fastapi import APIRouter, Path, Query
 from fastapi import status as st
 from fastapi.responses import UJSONResponse
 
-from ...audits import get_audit_model
+from ...audits import get_audit
 from ...result import Result
 
 router = APIRouter(
@@ -90,7 +90,7 @@ async def get_audits():
     """
     return {
         "message": "Getting audit logs",
-        "audits": list(get_audit_model().find_audits(name="demo")),
+        "audits": list(get_audit().find_audits(name="demo")),
     }
 
 
@@ -109,7 +109,7 @@ async def get_audit_with_workflow(workflow: str):
     """
     return {
         "message": f"Getting audit logs with workflow name {workflow}",
-        "audits": list(get_audit_model().find_audits(name="demo")),
+        "audits": list(get_audit().find_audits(name="demo")),
     }
 
 
@@ -136,7 +136,7 @@ async def get_audit_with_workflow_release(
             f"Getting audit logs with workflow name {workflow} and release "
             f"{release}"
         ),
-        "audits": list(get_audit_model().find_audits(name="demo")),
+        "audits": list(get_audit().find_audits(name="demo")),
     }
 
 
@@ -167,5 +167,5 @@ async def get_audit_with_workflow_release_run_id(
             f"Getting audit logs with workflow name {workflow}, release "
             f"{release}, and running ID {run_id}"
         ),
-        "audits": list(get_audit_model().find_audits(name="demo")),
+        "audits": list(get_audit().find_audits(name="demo")),
     }

ddeutil/workflow/api/routes/workflows.py

@@ -16,7 +16,7 @@ from fastapi.responses import UJSONResponse
 from pydantic import BaseModel
 
 from ...__types import DictData
-from ...audits import Audit, get_audit_model
+from ...audits import Audit, get_audit
 from ...conf import YamlParser
 from ...result import Result
 from ...workflow import Workflow
@@ -100,7 +100,7 @@ async def get_workflow_audits(name: str):
                     exclude_none=False,
                     exclude_unset=True,
                 )
-                for audit in get_audit_model().find_audits(name=name)
+                for audit in get_audit().find_audits(name=name)
             ],
         }
     except FileNotFoundError:
@@ -114,7 +114,7 @@ async def get_workflow_audits(name: str):
 async def get_workflow_release_audit(name: str, release: str):
     """Get Workflow audit log with an input release value."""
     try:
-        audit: Audit = get_audit_model().find_audit_with_release(
+        audit: Audit = get_audit().find_audit_with_release(
             name=name,
             release=datetime.strptime(release, "%Y%m%d%H%M%S"),
         )