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

introduce pyotp and support generating totp within skyvern (#1176)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-11-12 02:17:22 -08:00
committed by GitHub
parent 79bc51ba54
commit efdf4b5cab
5 changed files with 26 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -7,6 +7,7 @@ import uuid
from datetime import datetime, timedelta, timezone
from typing import Any, Awaitable, Callable, List
import pyotp
import structlog
from playwright.async_api import FileChooser, Frame, Locator, Page, TimeoutError
from pydantic import BaseModel
@@ -1082,8 +1083,9 @@ async def get_actual_value_of_parameter_if_secret(task: Task, parameter: str) ->
secret_value = workflow_run_context.get_original_secret_value_or_none(parameter)
if secret_value == BitwardenConstants.TOTP:
secrets = await workflow_run_context.get_secrets_from_password_manager()
secret_value = secrets[BitwardenConstants.TOTP]
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)
secret_value = pyotp.TOTP(totp_secret).now()
return secret_value if secret_value is not None else parameter