vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

support workflow_run_id in totp code (#2199)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-21 08:24:29 +08:00
committed by GitHub
parent 8b9c7c17e3
commit 9038e6e047
6 changed files with 46 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
skyvern/webeye/actions/handler.py
View File

@@ -3128,7 +3128,11 @@ async def poll_verification_code(
)
elif totp_identifier:
verification_code = await _get_verification_code_from_db(
task_id, organization_id, totp_identifier, workflow_id=workflow_id
task_id,
organization_id,
totp_identifier,
workflow_id=workflow_id,
workflow_run_id=workflow_run_id,
)
if verification_code:
LOG.info("Got verification code", verification_code=verification_code)
@@ -3169,9 +3173,12 @@ async def _get_verification_code_from_db(
organization_id: str,
totp_identifier: str,
workflow_id: str | None = None,
workflow_run_id: str | None = None,
) -> str | None:
totp_codes = await app.DATABASE.get_totp_codes(organization_id=organization_id, totp_identifier=totp_identifier)
for totp_code in totp_codes:
if totp_code.workflow_run_id and workflow_run_id and totp_code.workflow_run_id != workflow_run_id:
continue
if totp_code.workflow_id and workflow_id and totp_code.workflow_id != workflow_id:
continue
if totp_code.task_id and totp_code.task_id != task_id: