Differential D804 Diff 3678 python_modules/dagster-graphql/dagster_graphql/schema/run_schedule.py

Changeset View

Standalone View

python_modules/dagster-graphql/dagster_graphql/schema/run_schedule.py

This file was added.

				import json

				from dagster import check
				from dagster_graphql import dauphin

				from dagster_graphql.implementation.scheduler import RunSchedule, SystemCronScheduler


				class SchedulerType(dauphin.Enum):
				SystemCronScheduler = 1
				alangenfeldUnsubmitted Done Reply maybe use a string instead of 1? not sure how beneficial it actually is, depends where the value actually shows up alangenfeld: maybe use a string instead of 1? not sure how beneficial it actually is, depends where the…


				def get_scheduler(graphene_info):
				scheduler = graphene_info.context.scheduler

				if isinstance(scheduler, SystemCronScheduler):
				scheduler_type = SchedulerType.SystemCronScheduler
				else:
				raise Exception(
				'Unknown Scheduler type {typ}. Add this Scheduler type to the SchedulerType '
				'GraphQL Enum'.format(typ=type(scheduler))
				)

				schedules = [
				graphene_info.schema.type_named('RunSchedule')(s) for s in scheduler.all_schedules()
				]

				return graphene_info.schema.type_named('Scheduler')(
				scheduler_type=scheduler_type, schedules=schedules
				)


				class DauphinScheduler(dauphin.ObjectType):
				class Meta:
				name = 'Scheduler'

				scheduler_type = dauphin.NonNull(dauphin.String)
				schedules = dauphin.non_null_list('RunSchedule')


				class DauphinRunScheduleInput(dauphin.InputObjectType):
				class Meta:
				name = 'RunScheduleInput'

				name = dauphin.NonNull(dauphin.String)
				cron_schedule = dauphin.NonNull(dauphin.String)
				execution_params = dauphin.NonNull('ExecutionParams')


				class DauphinCreateRunScheduleResult(dauphin.ObjectType):
				class Meta:
				name = 'CreateRunScheduleResult'

				schedule = dauphin.NonNull('RunSchedule')


				class DauphinCreateRunScheduleMutation(dauphin.Mutation):
				class Meta:
				name = 'CreateRunScheduleMutation'

				class Arguments:
				schedule = dauphin.NonNull('RunScheduleInput')

				Output = dauphin.NonNull('CreateRunScheduleResult')

				def mutate(self, graphene_info, **kwargs):
				from dagster_graphql.schema.roots import create_execution_params

				# Check execution_params is valid ExecutionParams
				create_execution_params(kwargs['schedule'].get('execution_params'))

				scheduler = graphene_info.context.scheduler
				schedule = scheduler.create_schedule(
				name=kwargs['schedule'].get('name'),
				cron_schedule=kwargs['schedule'].get('cron_schedule'),
				execution_params=kwargs['schedule'].get('execution_params'),
				)

				return graphene_info.schema.type_named('CreateRunScheduleResult')(
				schedule=graphene_info.schema.type_named('RunSchedule')(schedule)
				)


				class DauphinDeleteRunScheduleResult(dauphin.ObjectType):
				class Meta:
				name = 'DeleteRunScheduleResult'

				deleted_schedule = dauphin.NonNull('RunSchedule')


				class DauphineDeleteRunScheduleMutation(dauphin.Mutation):
				class Meta:
				name = 'DeleteRunScheduleMutation'

				class Arguments:
				schedule_id = dauphin.NonNull(dauphin.String)

				Output = dauphin.NonNull('DeleteRunScheduleResult')

				def mutate(self, graphene_info, **kwargs):
				scheduler = graphene_info.context.scheduler
				schedule = scheduler.remove_schedule(kwargs['schedule_id'])

				return graphene_info.schema.type_named('DeleteRunScheduleResult')(
				deleted_schedule=graphene_info.schema.type_named('RunSchedule')(schedule)
				)


				class DauphinRunSchedule(dauphin.ObjectType):
				class Meta:
				name = 'RunSchedule'

				schedule_id = dauphin.NonNull(dauphin.String)
				name = dauphin.NonNull(dauphin.String)
				cron_schedule = dauphin.NonNull(dauphin.String)
				execution_params = dauphin.NonNull(dauphin.String)
				alangenfeldUnsubmitted Not Done Reply might be worth having a slightly different name since this returns a string that needs to get json decoded by the client alangenfeld: might be worth having a slightly different name since this returns a string that needs to get…

				def __init__(self, schedule):
				self._schedule = check.inst_param(schedule, 'schedule', RunSchedule)
				super(DauphinRunSchedule, self).__init__(
				schedule_id=schedule.schedule_id,
				name=schedule.name,
				cron_schedule=schedule.cron_schedule,
				execution_params=json.dumps(schedule.execution_params),
				)