opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[autofix.ci] apply automated fixes

This commit is contained in:
autofix-ci[bot] 2025-12-11 07:21:13 +00:00 committed by GitHub
parent d8578dd4ba
commit 66ee07154d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 21 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
api/services/clear_free_plan_expired_workflow_run_logs.py
View File

@ -114,9 +114,7 @@ class WorkflowRunCleanup:
click.echo(click.style(summary_message, fg="white"))
def _load_batch(
self, session: Session, last_seen: tuple[datetime.datetime, str] | None
) -> list[WorkflowRunRow]:
def _load_batch(self, session: Session, last_seen: tuple[datetime.datetime, str] | None) -> list[WorkflowRunRow]:
stmt = (
select(WorkflowRun.id, WorkflowRun.tenant_id, WorkflowRun.created_at)
.where(WorkflowRun.created_at < self.window_end)
@ -166,7 +164,9 @@ class WorkflowRunCleanup:
def _delete_runs(self, session: Session, workflow_run_ids: Sequence[str]) -> dict[str, int]:
node_execution_ids = session.scalars(
select(WorkflowNodeExecutionModel.id).where(WorkflowNodeExecutionModel.workflow_run_id.in_(workflow_run_ids))
select(WorkflowNodeExecutionModel.id).where(
WorkflowNodeExecutionModel.workflow_run_id.in_(workflow_run_ids)
)
).all()
offloads_deleted = 0
@ -199,14 +199,12 @@ class WorkflowRunCleanup:
if pause_ids:
pause_reasons_deleted = (
session.query(WorkflowPauseReason).where(WorkflowPauseReason.pause_id.in_(pause_ids)).delete(
synchronize_session=False
)
session.query(WorkflowPauseReason)
.where(WorkflowPauseReason.pause_id.in_(pause_ids))
.delete(synchronize_session=False)
)
pauses_deleted = (
session.query(WorkflowPause)
.where(WorkflowPause.id.in_(pause_ids))
.delete(synchronize_session=False)
session.query(WorkflowPause).where(WorkflowPause.id.in_(pause_ids)).delete(synchronize_session=False)
)
trigger_logs_deleted = (

34
api/tests/unit_tests/services/test_clear_free_plan_expired_workflow_run_logs.py
View File

@ -85,9 +85,7 @@ def test_run_deletes_only_free_tenants(monkeypatch: pytest.MonkeyPatch) -> None:
batches_returned = 0
def fake_load_batch(
session: DummySession, last_seen: tuple[datetime.datetime, str] | None
) -> list[WorkflowRunRow]:
def fake_load_batch(session: DummySession, last_seen: tuple[datetime.datetime, str] | None) -> list[WorkflowRunRow]:
nonlocal batches_returned
if batches_returned > 0:
return []
@ -130,7 +128,7 @@ def test_run_deletes_only_free_tenants(monkeypatch: pytest.MonkeyPatch) -> None:
cleanup.run()
assert deleted_ids == [["run-free"]]
assert created_sessions
assert created_sessions
assert created_sessions[0].committed is True
@ -142,16 +140,12 @@ def test_run_skips_when_no_free_tenants(monkeypatch: pytest.MonkeyPatch) -> None
monkeypatch.setattr(
cleanup_module.BillingService,
"get_info_bulk",
staticmethod(
lambda tenant_ids: {tenant_id: {"subscription": {"plan": "TEAM"}} for tenant_id in tenant_ids}
),
staticmethod(lambda tenant_ids: {tenant_id: {"subscription": {"plan": "TEAM"}} for tenant_id in tenant_ids}),
)
batches_returned = 0
def fake_load_batch(
session: DummySession, last_seen: tuple[datetime.datetime, str] | None
) -> list[WorkflowRunRow]:
def fake_load_batch(session: DummySession, last_seen: tuple[datetime.datetime, str] | None) -> list[WorkflowRunRow]:
nonlocal batches_returned
if batches_returned > 0:
return []
@ -164,14 +158,14 @@ def test_run_skips_when_no_free_tenants(monkeypatch: pytest.MonkeyPatch) -> None
nonlocal delete_called
delete_called = True
return {
"runs": 0,
"node_executions": 0,
"offloads": 0,
"app_logs": 0,
"trigger_logs": 0,
"pauses": 0,
"pause_reasons": 0
}
"runs": 0,
"node_executions": 0,
"offloads": 0,
"app_logs": 0,
"trigger_logs": 0,
"pauses": 0,
"pause_reasons": 0,
}
def fake_session_factory(engine: object | None = None) -> DummySession: # pragma: no cover - simple factory
return DummySession()
@ -189,9 +183,7 @@ def test_run_skips_when_no_free_tenants(monkeypatch: pytest.MonkeyPatch) -> None
def test_run_exits_on_empty_batch(monkeypatch: pytest.MonkeyPatch) -> None:
cleanup = WorkflowRunCleanup(days=30, batch_size=10)
def fake_load_batch(
session: DummySession, last_seen: tuple[datetime.datetime, str] | None
) -> list[WorkflowRunRow]:
def fake_load_batch(session: DummySession, last_seen: tuple[datetime.datetime, str] | None) -> list[WorkflowRunRow]:
return []
def fake_delete_runs(session: DummySession, workflow_run_ids: list[str]) -> dict[str, int]: