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

do not raise error when parsing totp secret (#3144)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-08-08 11:00:27 -07:00
committed by GitHub
parent e3a3309e9c
commit addf8c5a9a
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,8 +5,6 @@ from urllib.parse import unquote
import pyotp import pyotp
import structlog import structlog
from skyvern.exceptions import NoTOTPSecretFound
LOG = structlog.get_logger() LOG = structlog.get_logger()
@@ -36,7 +34,7 @@ def parse_totp_secret(totp_secret: str) -> str:
LOG.warning("Failed to parse TOTP secret key from URI format, going to extract secret by regex", exc_info=True) LOG.warning("Failed to parse TOTP secret key from URI format, going to extract secret by regex", exc_info=True)
m = re.search(r"(?i)(?:^|[?&])secret=([^&#]+)", unquote(totp_secret_no_whitespace)) m = re.search(r"(?i)(?:^|[?&])secret=([^&#]+)", unquote(totp_secret_no_whitespace))
if m is None: if m is None:
raise NoTOTPSecretFound() return totp_secret_no_whitespace
totp_secret = m.group(1) totp_secret = m.group(1)
totp_secret_no_whitespace = "".join(totp_secret.split()) totp_secret_no_whitespace = "".join(totp_secret.split())
return totp_secret_no_whitespace return totp_secret_no_whitespace