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

make totp_identifier parametrized in workflow (#847)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-09-19 15:20:15 -07:00
committed by GitHub
parent 30922a3f8c
commit ffaf3ab454
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -240,6 +240,20 @@ class TaskBlock(Block):
) )
self.url = task_url_parameter_value self.url = task_url_parameter_value
if (
self.totp_identifier
and workflow_run_context.has_parameter(self.totp_identifier)
and workflow_run_context.has_value(self.totp_identifier)
):
totp_identifier_parameter_value = workflow_run_context.get_value(self.totp_identifier)
if totp_identifier_parameter_value:
LOG.info(
"TOTP identifier is parameterized, using parameter value",
totp_identifier_parameter_value=totp_identifier_parameter_value,
totp_identifier_parameter_key=self.totp_identifier,
)
self.totp_identifier = totp_identifier_parameter_value
if self.download_suffix and workflow_run_context.has_parameter(self.download_suffix): if self.download_suffix and workflow_run_context.has_parameter(self.download_suffix):
download_suffix_parameter_value = workflow_run_context.get_value(self.download_suffix) download_suffix_parameter_value = workflow_run_context.get_value(self.download_suffix)
if download_suffix_parameter_value: if download_suffix_parameter_value: