class DauphinDaemonStatus(dauphin.ObjectType):
class Meta:
name = "DaemonStatus"
daemonType = dauphin.NonNull("DaemonType")
required = dauphin.NonNull(dauphin.Boolean)
healthy = dauphin.Boolean()
lastHeartbeatTime = dauphin.Float()
def __init__(self, daemon_status):
check.inst_param(daemon_status, "daemon_status", DaemonStatus)
if daemon_status.last_heartbeat is None:
last_heartbeat_time = None
else:
last_heartbeat_time = datetime_as_float(
daemon_status.last_heartbeat.timestamp)
super(DauphinDaemonStatus, self).__init__(
daemonType=daemon_status.daemon_type,
required=daemon_status.required,
healthy=daemon_status.healthy,
lastHeartbeatTime=last_heartbeat_time,
)
class DauphinDaemonStatus(dauphin.ObjectType):
class Meta:
name = "DaemonStatus"
daemonType = dauphin.NonNull("DaemonType")
required = dauphin.NonNull(dauphin.Boolean)
healthy = dauphin.Boolean()
lastHeartbeatTime = dauphin.Float()
lastHeartbeatErrors = dauphin.non_null_list("PythonError")
def __init__(self, daemon_status):
check.inst_param(daemon_status, "daemon_status", DaemonStatus)
super(DauphinDaemonStatus, self).__init__(
daemonType=daemon_status.daemon_type,
required=daemon_status.required,
healthy=daemon_status.healthy,
lastHeartbeatTime=daemon_status.last_heartbeat.timestamp
if daemon_status.last_heartbeat else None,
lastHeartbeatErrors=[
DauphinPythonError(error)
for error in daemon_status.last_heartbeat.errors
] if daemon_status.last_heartbeat else [],
)
class DauphinSensorJobData(dauphin.ObjectType):
class Meta:
name = "SensorJobData"
lastTickTimestamp = dauphin.Float()
lastRunKey = dauphin.String()
def __init__(self, job_specific_data):
check.inst_param(job_specific_data, "job_specific_data", SensorJobData)
super(DauphinSensorJobData, self).__init__(
lastTickTimestamp=job_specific_data.last_tick_timestamp,
lastRunKey=job_specific_data.last_run_key,
)
class DauphinScheduleJobData(dauphin.ObjectType):
class Meta:
name = "ScheduleJobData"
cronSchedule = dauphin.NonNull(dauphin.String)
startTimestamp = dauphin.Float()
def __init__(self, job_specific_data):
check.inst_param(job_specific_data, "job_specific_data", ScheduleJobData)
super(DauphinScheduleJobData, self).__init__(
cronSchedule=job_specific_data.cron_schedule,
startTimestamp=job_specific_data.start_timestamp,
)
class DauphinSchedule(dauphin.ObjectType):
class Meta:
name = "Schedule"
id = dauphin.NonNull(dauphin.ID)
name = dauphin.NonNull(dauphin.String)
cron_schedule = dauphin.NonNull(dauphin.String)
pipeline_name = dauphin.NonNull(dauphin.String)
solid_selection = dauphin.List(dauphin.String)
mode = dauphin.NonNull(dauphin.String)
execution_timezone = dauphin.Field(dauphin.String)
scheduleState = dauphin.NonNull("JobState")
partition_set = dauphin.Field("PartitionSet")
futureTicks = dauphin.NonNull("FutureJobTicks",
cursor=dauphin.Float(),
limit=dauphin.Int())
futureTick = dauphin.NonNull("FutureJobTick",
tick_timestamp=dauphin.NonNull(dauphin.Int))
def resolve_id(self, _):
return "%s:%s" % (self.name, self.pipeline_name)
def resolve_partition_set(self, graphene_info):
if self._external_schedule.partition_set_name is None:
return None
repository = graphene_info.context.get_repository_location(
self._external_schedule.handle.location_name).get_repository(
self._external_schedule.handle.repository_name)
external_partition_set = repository.get_external_partition_set(
self._external_schedule.partition_set_name)
return graphene_info.schema.type_named("PartitionSet")(
external_repository_handle=repository.handle,
external_partition_set=external_partition_set,
)
def resolve_futureTicks(self, graphene_info, **kwargs):
cursor = kwargs.get(
"cursor",
get_timestamp_from_utc_datetime(get_current_datetime_in_utc()))
limit = kwargs.get("limit", 10)
tick_times = []
time_iter = self._external_schedule.execution_time_iterator(cursor)
for _ in range(limit):
tick_times.append(next(time_iter).timestamp())
future_ticks = [
graphene_info.schema.type_named("FutureJobTick")(
self._schedule_state, tick_time) for tick_time in tick_times
]
return graphene_info.schema.type_named("FutureJobTicks")(
results=future_ticks, cursor=tick_times[-1] + 1)
def resolve_futureTick(self, graphene_info, tick_timestamp):
return graphene_info.schema.type_named("FutureJobTick")(
self._schedule_state, float(tick_timestamp))
def __init__(self, graphene_info, external_schedule):
self._external_schedule = check.inst_param(external_schedule,
"external_schedule",
ExternalSchedule)
self._schedule_state = graphene_info.context.instance.get_job_state(
self._external_schedule.get_external_origin_id())
if not self._schedule_state:
# Also include a ScheduleState for a stopped schedule that may not
# have a stored database row yet
self._schedule_state = self._external_schedule.get_default_job_state(
graphene_info.context.instance)
super(DauphinSchedule, self).__init__(
name=external_schedule.name,
cron_schedule=external_schedule.cron_schedule,
pipeline_name=external_schedule.pipeline_name,
solid_selection=external_schedule.solid_selection,
mode=external_schedule.mode,
scheduleState=graphene_info.schema.type_named("JobState")(
self._schedule_state),
execution_timezone=(self._external_schedule.execution_timezone
if self._external_schedule.execution_timezone
else pendulum.now().timezone.name),
)
