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

remove all the whitespace from the totp secret (#1180)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-11-12 15:33:20 -08:00
committed by GitHub
parent f33cbfbb0b
commit 9f3af2555c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1085,7 +1085,8 @@ async def get_actual_value_of_parameter_if_secret(task: Task, parameter: str) ->
if secret_value == BitwardenConstants.TOTP: if secret_value == BitwardenConstants.TOTP:
totp_secret_key = workflow_run_context.totp_secret_value_key(parameter) 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 = workflow_run_context.get_original_secret_value_or_none(totp_secret_key)
secret_value = pyotp.TOTP(totp_secret).now() totp_secret_no_whitespace = "".join(totp_secret.split())
secret_value = pyotp.TOTP(totp_secret_no_whitespace).now()
return secret_value if secret_value is not None else parameter return secret_value if secret_value is not None else parameter