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

fix valid opt secret detect (#3127)

Browse Source
This commit is contained in:
LawyZheng
2025-08-07 15:31:09 +08:00
committed by GitHub
parent 44c9a81156
commit d0abe13e14
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/forge/sdk/services/credentials.py
View File

@@ -21,10 +21,13 @@ def parse_totp_secret(totp_secret: str) -> str:
return "" return ""
totp_secret_no_whitespace = "".join(totp_secret.split()) totp_secret_no_whitespace = "".join(totp_secret.split())
if len(totp_secret_no_whitespace) == 32: try:
# to verify if it's a valid TOTP secret
pyotp.TOTP(totp_secret_no_whitespace).byte_secret()
return totp_secret_no_whitespace return totp_secret_no_whitespace
except Exception:
LOG.warning("It's not a valid TOTP secret, going to parse it from URI format", exc_info=True)
LOG.info("TOTP secret key is not 32 characters, try to parse it from URI format")
try: try:
totp_secret = pyotp.parse_uri(totp_secret_no_whitespace).secret totp_secret = pyotp.parse_uri(totp_secret_no_whitespace).secret
totp_secret_no_whitespace = "".join(totp_secret.split()) totp_secret_no_whitespace = "".join(totp_secret.split())