Commit 9c4bae10 authored by harshit aggarwal's avatar harshit aggarwal Committed by Siarhei Khaletski (EPAM)
Browse files

Added support for handling data-partition-id (e.g. CSV parser uses)

parent 59e58330
......@@ -109,7 +109,13 @@ class UpdateStatusOperator(BaseOperator):
if "Payload" in execution_context:
payload_context = Context.populate(execution_context)
else:
payload_context = Context(data_partition_id=execution_context["data-partition-id"],
try:
data_partition_id = execution_context["data-partition-id"]
except KeyError:
# to support some DAGs execution_context interface
data_partition_id = execution_context['dataPartitionId']
payload_context = Context(data_partition_id=data_partition_id,
app_key=execution_context.get("AppKey", ""))
workflow_name = conf["workflow_name"]
run_id = conf["run_id"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment