def ui(text, file, predefined, variables, remote, output, remap_sigterm, **kwargs): query = None if text is not None and file is None and predefined is None: query = text.strip("'\" \n\t") elif file is not None and text is None and predefined is None: query = file.read() elif predefined is not None and text is None and file is None: query = PREDEFINED_QUERIES[predefined] else: raise click.UsageError( "Must select one and only one of text (-t), file (-f), or predefined (-p) " "to select GraphQL document to execute.") if remap_sigterm: signal.signal(signal.SIGTERM, signal.getsignal(signal.SIGINT)) if remote: res = execute_query_against_remote(remote, query, variables) print(res) # pylint: disable=print-call else: instance = DagsterInstance.get() with get_workspace_from_kwargs(kwargs, instance) as workspace: execute_query_from_cli( workspace, query, instance, variables, output, )
def execute_backfill_command(cli_args, print_fn, instance, using_graph_job_op_apis=False): with get_workspace_from_kwargs(instance, version=dagster_version, kwargs=cli_args) as workspace: repo_location = get_repository_location_from_workspace( workspace, cli_args.get("location")) _execute_backfill_command_at_location(cli_args, print_fn, instance, workspace, repo_location, using_graph_job_op_apis)
def execute_launch_command(instance: DagsterInstance, kwargs: Dict[str, str], using_job_op_graph_apis: bool = False): preset = cast(Optional[str], kwargs.get("preset")) mode = cast(Optional[str], kwargs.get("mode")) check.inst_param(instance, "instance", DagsterInstance) config = get_config_from_args(kwargs) with get_workspace_from_kwargs(instance, version=dagster_version, kwargs=kwargs) as workspace: repo_location = get_repository_location_from_workspace( workspace, kwargs.get("location")) external_repo = get_external_repository_from_repo_location( repo_location, cast(Optional[str], kwargs.get("repository"))) external_pipeline = get_external_pipeline_or_job_from_external_repo( external_repo, cast(Optional[str], kwargs.get("pipeline_or_job")), using_job_op_graph_apis, ) log_external_repo_stats( instance=instance, external_pipeline=external_pipeline, external_repo=external_repo, source="pipeline_launch_command", ) if preset and config: raise click.UsageError( "Can not use --preset with -c / --config / --config-json.") run_tags = get_tags_from_args(kwargs) solid_selection = get_solid_selection_from_args(kwargs) pipeline_run = _create_external_pipeline_run( instance=instance, repo_location=repo_location, external_repo=external_repo, external_pipeline=external_pipeline, run_config=config, mode=mode, preset=preset, tags=run_tags, solid_selection=solid_selection, run_id=cast(Optional[str], kwargs.get("run_id")), ) return instance.submit_run(pipeline_run.run_id, workspace)
def ui(text, file, predefined, variables, remote, output, **kwargs): query = None if text is not None and file is None and predefined is None: query = text.strip('\'" \n\t') elif file is not None and text is None and predefined is None: query = file.read() elif predefined is not None and text is None and file is None: query = PREDEFINED_QUERIES[predefined] else: raise click.UsageError( 'Must select one and only one of text (-t), file (-f), or predefined (-p) ' 'to select GraphQL document to execute.') if remote: res = execute_query_against_remote(remote, query, variables) print(res) else: workspace = get_workspace_from_kwargs(kwargs) execute_query_from_cli(workspace, query, variables, output)
def ui(text, file, predefined, variables, remote, output, remap_sigterm, **kwargs): query = None if text is not None and file is None and predefined is None: query = text.strip("'\" \n\t") elif file is not None and text is None and predefined is None: query = file.read() elif predefined is not None and text is None and file is None: query = PREDEFINED_QUERIES[predefined] else: raise click.UsageError( "Must select one and only one of text (-t), file (-f), or predefined (-p) " "to select GraphQL document to execute.") if remap_sigterm: try: signal.signal(signal.SIGTERM, signal.getsignal(signal.SIGINT)) except ValueError: warnings.warn(( "Unexpected error attempting to manage signal handling on thread {thread_name}. " "You should not invoke this API (ui) from threads " "other than the main thread.").format( thread_name=threading.current_thread().name)) if remote: res = execute_query_against_remote(remote, query, variables) print(res) # pylint: disable=print-call else: instance = DagsterInstance.get() with get_workspace_from_kwargs(kwargs) as workspace: execute_query_from_cli( workspace, query, instance, variables, output, )