Centurion/system
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update block_wrapper.py
All checks were successful
CI Workflow / test (push) Successful in 1m42s
Details
CI Workflow / Containerize the Block (push) Successful in 1m27s
Details

Browse Source
This commit is contained in:
Admin User 2025-02-25 13:16:47 +00:00
parent 1b10d2968d
commit 0c31359ec5
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
block_wrapper.py
View File

@ -6,7 +6,7 @@ import os
import re import re
import sys import sys
from temporalio import activity from temporalio import activity
from temporalio.exceptions import ApplicationError, ActivityError from temporalio.exceptions import ApplicationError
from jsonschema import validate, ValidationError from jsonschema import validate, ValidationError
from temporalio.client import Client from temporalio.client import Client
from temporalio.worker import Worker from temporalio.worker import Worker
@ -28,7 +28,7 @@ COMMIT_ID = os.getenv('VERSION')
NAMESPACE = os.getenv('NAMESPACE') NAMESPACE = os.getenv('NAMESPACE')
FLOWX_ENGINE_ADDRESS = os.getenv('FLOWX_ENGINE_ADDRESS') FLOWX_ENGINE_ADDRESS = os.getenv('FLOWX_ENGINE_ADDRESS')
if not BRANCH_NAME or not COMMIT_ID or not BRANCH_NAME or not NAMESPACE or not FLOWX_ENGINE_ADDRESS: if not BRANCH_NAME or not COMMIT_ID or not NAMESPACE or not FLOWX_ENGINE_ADDRESS:
raise ValueError("Environment variables BRANCH_NAME, VERSION, BRANCH_NAME, NAMESPACE and FLOWX_ENGINE_ADDRESS must be set.") raise ValueError("Environment variables BRANCH_NAME, VERSION, BRANCH_NAME, NAMESPACE and FLOWX_ENGINE_ADDRESS must be set.")
@ -48,6 +48,7 @@ BLOCK_NAME = REPO_NAME + "_" + BRANCH_NAME
block_name_safe = sanitize_name(BLOCK_NAME) block_name_safe = sanitize_name(BLOCK_NAME)
commit_id_safe = sanitize_name(COMMIT_ID_SHORT) commit_id_safe = sanitize_name(COMMIT_ID_SHORT)
# Construct the task queue name # Construct the task queue name
TASK_QUEUE = f"{block_name_safe}_{commit_id_safe}" TASK_QUEUE = f"{block_name_safe}_{commit_id_safe}"
@ -133,7 +134,7 @@ async def block_main_activity(input_data):
except Exception as e: except Exception as e:
logger.error("Error executing block.py: %s", e) logger.error("Error executing block.py: %s", e)
if not IS_TEST_ENVIRONMENT: if not IS_TEST_ENVIRONMENT:
raise ActivityError("Error during block.py execution") from e raise ApplicationError(f"Error during block execution: {e}") from e
else: else:
raise RuntimeError("Error during block.py execution") from e raise RuntimeError("Error during block.py execution") from e