Changeset View
Changeset View
Standalone View
Standalone View
python_modules/dagster-graphql/dagster_graphql_tests/graphql/setup.py
| Show First 20 Lines • Show All 41 Lines • ▼ Show 20 Lines | |||||
| from dagster.utils import script_relative_path | from dagster.utils import script_relative_path | ||||
| from dagster_graphql.implementation.context import DagsterGraphQLContext | from dagster_graphql.implementation.context import DagsterGraphQLContext | ||||
| from dagster_graphql.implementation.pipeline_execution_manager import SynchronousExecutionManager | from dagster_graphql.implementation.pipeline_execution_manager import SynchronousExecutionManager | ||||
| from dagster_graphql.implementation.pipeline_run_storage import ( | from dagster_graphql.implementation.pipeline_run_storage import ( | ||||
| FilesystemRunStorage, | FilesystemRunStorage, | ||||
| InMemoryRunStorage, | InMemoryRunStorage, | ||||
| ) | ) | ||||
| from dagster_graphql.implementation.scheduler import SystemCronScheduler | |||||
| class PoorMansDataFrame_(list): | class PoorMansDataFrame_(list): | ||||
| pass | pass | ||||
| @input_hydration_config(Path) | @input_hydration_config(Path) | ||||
| def df_input_schema(_context, path): | def df_input_schema(_context, path): | ||||
| Show All 15 Lines | |||||
| PoorMansDataFrame = as_dagster_type( | PoorMansDataFrame = as_dagster_type( | ||||
| PoorMansDataFrame_, | PoorMansDataFrame_, | ||||
| input_hydration_config=df_input_schema, | input_hydration_config=df_input_schema, | ||||
| output_materialization_config=df_output_schema, | output_materialization_config=df_output_schema, | ||||
| ) | ) | ||||
| def define_context(raise_on_error=True, log_dir=None): | def define_context(raise_on_error=True, log_dir=None, schedule_dir=None): | ||||
| return DagsterGraphQLContext( | return DagsterGraphQLContext( | ||||
| handle=ExecutionTargetHandle.for_repo_fn(define_repository), | handle=ExecutionTargetHandle.for_repo_fn(define_repository), | ||||
| pipeline_runs=FilesystemRunStorage(log_dir) if log_dir else InMemoryRunStorage(), | pipeline_runs=FilesystemRunStorage(log_dir) if log_dir else InMemoryRunStorage(), | ||||
| scheduler=SystemCronScheduler(schedule_dir) if schedule_dir else None, | |||||
| execution_manager=SynchronousExecutionManager(), | execution_manager=SynchronousExecutionManager(), | ||||
| raise_on_error=raise_on_error, | raise_on_error=raise_on_error, | ||||
| ) | ) | ||||
| @lambda_solid( | @lambda_solid( | ||||
| input_defs=[InputDefinition('num', PoorMansDataFrame)], | input_defs=[InputDefinition('num', PoorMansDataFrame)], | ||||
| output_def=OutputDefinition(PoorMansDataFrame), | output_def=OutputDefinition(PoorMansDataFrame), | ||||
| ▲ Show 20 Lines • Show All 415 Lines • Show Last 20 Lines | |||||