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

support uri format otp secrete (#3126)

Browse Source
This commit is contained in:
LawyZheng
2025-08-07 14:59:29 +08:00
committed by GitHub
parent 1a4bf1df1a
commit 44c9a81156
5 changed files with 44 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1964,8 +1964,10 @@ def generate_totp_value(task: Task, parameter: str) -> str:
workflow_run_context = app.WORKFLOW_CONTEXT_MANAGER.get_workflow_run_context(task.workflow_run_id)
totp_secret_key = workflow_run_context.totp_secret_value_key(parameter)
totp_secret = workflow_run_context.get_original_secret_value_or_none(totp_secret_key)
totp_secret_no_whitespace = "".join(totp_secret.split())
return pyotp.TOTP(totp_secret_no_whitespace).now()
if not totp_secret:
LOG.warning("No TOTP secret found, returning the parameter value as is", parameter=parameter)
return parameter
return pyotp.TOTP(totp_secret).now()
async def chain_click(