Changeset View
Changeset View
Standalone View
Standalone View
python_modules/dagster/dagster/core/definitions/repository.py
| from dagster import check | from dagster import check | ||||
| from dagster.core.errors import DagsterInvalidDefinitionError, DagsterInvariantViolationError | from dagster.core.errors import DagsterInvalidDefinitionError, DagsterInvariantViolationError | ||||
| from dagster.utils import merge_dicts | from dagster.utils import merge_dicts | ||||
| from .job import JobDefinition | from .job import JobDefinition, JobType | ||||
| from .partition import PartitionScheduleDefinition, PartitionSetDefinition | from .partition import PartitionScheduleDefinition, PartitionSetDefinition | ||||
| from .pipeline import PipelineDefinition | from .pipeline import PipelineDefinition | ||||
| from .schedule import ScheduleDefinition | from .schedule import ScheduleDefinition | ||||
| from .utils import check_valid_name | from .utils import check_valid_name | ||||
| VALID_REPOSITORY_DATA_DICT_KEYS = { | VALID_REPOSITORY_DATA_DICT_KEYS = { | ||||
| "pipelines", | "pipelines", | ||||
| "partition_sets", | "partition_sets", | ||||
| ▲ Show 20 Lines • Show All 399 Lines • ▼ Show 20 Lines | def get_schedule(self, schedule_name): | ||||
| return self._schedules.get_definition(schedule_name) | return self._schedules.get_definition(schedule_name) | ||||
| def has_schedule(self, schedule_name): | def has_schedule(self, schedule_name): | ||||
| check.str_param(schedule_name, "schedule_name") | check.str_param(schedule_name, "schedule_name") | ||||
| return self._schedules.has_definition(schedule_name) | return self._schedules.has_definition(schedule_name) | ||||
| def get_all_sensors(self): | |||||
| return [ | |||||
| definition | |||||
| for definition in self._jobs.get_all_definitions() | |||||
| if definition.job_type == JobType.SENSOR | |||||
| ] | |||||
| def get_sensor(self, name): | |||||
| return self._jobs.get_definition(name) | |||||
| def has_sensor(self, name): | |||||
| return self._jobs.has_definition(name) | |||||
| def get_all_jobs(self): | def get_all_jobs(self): | ||||
| return self._jobs.get_all_definitions() | return self._jobs.get_all_definitions() | ||||
| def get_job(self, name): | def get_job(self, name): | ||||
| check.str_param(name, "name") | check.str_param(name, "name") | ||||
| return self._jobs.get_definition(name) | return self._jobs.get_definition(name) | ||||
| def has_job(self, name): | def has_job(self, name): | ||||
| ▲ Show 20 Lines • Show All 164 Lines • ▼ Show 20 Lines | class RepositoryDefinition(object): | ||||
| def get_schedule_def(self, name): | def get_schedule_def(self, name): | ||||
| return self._repository_data.get_schedule(name) | return self._repository_data.get_schedule(name) | ||||
| def has_schedule_def(self, name): | def has_schedule_def(self, name): | ||||
| return self._repository_data.has_schedule(name) | return self._repository_data.has_schedule(name) | ||||
| @property | @property | ||||
| def sensor_defs(self): | |||||
| return self._repository_data.get_all_sensors() | |||||
| def get_sensor_def(self, name): | |||||
| return self._repository_data.get_sensor(name) | |||||
| def has_sensor_def(self, name): | |||||
| return self._repository_data.has_sensor(name) | |||||
| @property | |||||
| def job_defs(self): | def job_defs(self): | ||||
| return self._repository_data.get_all_jobs() | return self._repository_data.get_all_jobs() | ||||
| def get_job_def(self, name): | def get_job_def(self, name): | ||||
| return self._repository_data.get_job(name) | return self._repository_data.get_job(name) | ||||
| def has_job_def(self, name): | def has_job_def(self, name): | ||||
| return self._repository_data.has_job(name) | return self._repository_data.has_job(name) | ||||