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 |