Differential D8684 Diff 40955 examples/docs_snippets/docs_snippets/guides/dagster/run-attribution/custom_run_coordinator_skeleton.py

Changeset View

Standalone View

	"""isort:skip_file"""			"""isort:skip_file"""

	# start_custom_run_coordinator_marker			# start_custom_run_coordinator_marker

	from dagster.core.run_coordinator import QueuedRunCoordinator, SubmitRunContext			from dagster.core.run_coordinator import (
				QueuedRunCoordinator,
				SubmitRunContext,
				)
	from dagster.core.storage.pipeline_run import PipelineRun			from dagster.core.storage.pipeline_run import PipelineRun


	class CustomRunCoordinator(QueuedRunCoordinator):			class CustomRunCoordinator(QueuedRunCoordinator):
	def submit_run(self, context: SubmitRunContext) -> PipelineRun:			def submit_run(self, context: SubmitRunContext) -> PipelineRun:
	pass			pass


	# end_custom_run_coordinator_marker			# end_custom_run_coordinator_marker

	CUSTOM_HEADER_NAME = "X-SOME-HEADER"			CUSTOM_HEADER_NAME = "X-SOME-HEADER"

	# start_flask_header_marker			# start_flask_header_marker

	from flask import has_request_context, request			from flask import has_request_context, request

	desired_header = request.headers.get(CUSTOM_HEADER_NAME) if has_request_context() else None			desired_header = (
				request.headers.get(CUSTOM_HEADER_NAME) if has_request_context() else None
				)

	# end_flask_header_marker			# end_flask_header_marker