Ask AI

You are viewing an unreleased or outdated version of the documentation

Internals

Note that APIs imported from Dagster submodules are not considered stable, and are potentially subject to change in the future.

If you find yourself consulting these docs because you are writing custom components and plug-ins, please get in touch with the core team on our Slack. We’re curious what you’re up to, happy to help, excited for new community contributions, and eager to make the system as easy to work with as possible – including for teams who are looking to customize it.

Executors (Experimental)

APIs for constructing custom executors. This is considered advanced experimental usage. Please note that using Dagster-provided executors is considered stable, common usage.

@dagster.executor(name=None, config_schema=None, requirements=None)[source]

Define an executor.

The decorated function should accept an InitExecutorContext and return an instance of Executor.

Parameters:
  • name (Optional[str]) – The name of the executor.

  • config_schema (Optional[ConfigSchema]) – The schema for the config. Configuration data available in init_context.executor_config. If not set, Dagster will accept any config provided for.

  • requirements (Optional[List[ExecutorRequirement]]) – Any requirements that must be met in order for the executor to be usable for a particular job execution.

class dagster.ExecutorDefinition(name, config_schema=None, requirements=None, executor_creation_fn=None, description=None)[source]

An executor is responsible for executing the steps of a job.

Parameters:
  • name (str) – The name of the executor.

  • config_schema (Optional[ConfigSchema]) – The schema for the config. Configuration data available in init_context.executor_config. If not set, Dagster will accept any config provided.

  • requirements (Optional[List[ExecutorRequirement]]) – Any requirements that must be met in order for the executor to be usable for a particular job execution.

  • executor_creation_fn (Optional[Callable]) – Should accept an InitExecutorContext and return an instance of Executor

  • required_resource_keys (Optional[Set[str]]) – Keys for the resources required by the executor.

  • description (Optional[str]) – A description of the executor.

configured(config_or_config_fn, name=None, config_schema=None, description=None)[source]

Wraps this object in an object of the same type that provides configuration to the inner object.

Using configured may result in config values being displayed in the Dagster UI, so it is not recommended to use this API with sensitive values, such as secrets.

Parameters:
  • config_or_config_fn (Union[Any, Callable[[Any], Any]]) – Either (1) Run configuration that fully satisfies this object’s config schema or (2) A function that accepts run configuration and returns run configuration that fully satisfies this object’s config schema. In the latter case, config_schema must be specified. When passing a function, it’s easiest to use configured().

  • name (Optional[str]) – Name of the new definition. If not provided, the emitted definition will inherit the name of the ExecutorDefinition upon which this function is called.

  • config_schema (Optional[ConfigSchema]) – If config_or_config_fn is a function, the config schema that its input must satisfy. If not set, Dagster will accept any config provided.

  • description (Optional[str]) – Description of the new definition. If not specified, inherits the description of the definition being configured.

Returns (ConfigurableDefinition): A configured version of this object.

property description

Description of executor, if provided.

property executor_creation_fn

Callable that takes an InitExecutorContext and returns an instance of Executor.

property name

Name of the executor.

class dagster.InitExecutorContext(job, executor_def, executor_config, instance)[source]

Executor-specific initialization context.

job

The job to be executed.

Type:

IJob

executor_def

The definition of the executor currently being constructed.

Type:

ExecutorDefinition

executor_config

The parsed config passed to the executor.

Type:

dict

instance

The current instance.

Type:

DagsterInstance

class dagster.Executor[source]
abstract execute(plan_context, execution_plan)[source]

For the given context and execution plan, orchestrate a series of sub plan executions in a way that satisfies the whole plan being executed.

Parameters:
  • plan_context (PlanOrchestrationContext) – The plan’s orchestration context.

  • execution_plan (ExecutionPlan) – The plan to execute.

Returns:

A stream of dagster events.

abstract property retries

Whether retries are enabled or disabled for this instance of the executor.

Executors should allow this to be controlled via configuration if possible.

Returns: RetryMode


File Manager (Experimental)

class dagster._core.storage.file_manager.FileManager[source]

Base class for all file managers in dagster.

The file manager is an interface that can be implemented by resources to provide abstract access to a file system such as local disk, S3, or other cloud storage.

For examples of usage, see the documentation of the concrete file manager implementations.

abstract copy_handle_to_local_temp(file_handle)[source]

Copy a file represented by a file handle to a temp file.

In an implementation built around an object store such as S3, this method would be expected to download the file from S3 to local filesystem in a location assigned by the standard library’s python:tempfile module.

Temp files returned by this method are not guaranteed to be reusable across solid boundaries. For files that must be available across solid boundaries, use the read(), read_data(), write(), and write_data() methods.

Parameters:

file_handle (FileHandle) – The handle to the file to make available as a local temp file.

Returns:

Path to the local temp file.

Return type:

str

abstract delete_local_temp()[source]

Delete all local temporary files created by previous calls to copy_handle_to_local_temp().

Should typically only be called by framework implementors.

abstract read(file_handle, mode='rb')[source]

Return a file-like stream for the file handle.

This may incur an expensive network call for file managers backed by object stores such as S3.

Parameters:
  • file_handle (FileHandle) – The file handle to make available as a stream.

  • mode (str) – The mode in which to open the file. Default: "rb".

Returns:

A file-like stream.

Return type:

Union[TextIO, BinaryIO]

abstract read_data(file_handle)[source]

Return the bytes for a given file handle. This may incur an expensive network call for file managers backed by object stores such as s3.

Parameters:

file_handle (FileHandle) – The file handle for which to return bytes.

Returns:

Bytes for a given file handle.

Return type:

bytes

abstract write(file_obj, mode='wb', ext=None)[source]

Write the bytes contained within the given file object into the file manager.

Parameters:
  • file_obj (Union[TextIO, StringIO]) – A file-like object.

  • mode (Optional[str]) – The mode in which to write the file into the file manager. Default: "wb".

  • ext (Optional[str]) – For file managers that support file extensions, the extension with which to write the file. Default: None.

Returns:

A handle to the newly created file.

Return type:

FileHandle

abstract write_data(data, ext=None)[source]

Write raw bytes into the file manager.

Parameters:
  • data (bytes) – The bytes to write into the file manager.

  • ext (Optional[str]) – For file managers that support file extensions, the extension with which to write the file. Default: None.

Returns:

A handle to the newly created file.

Return type:

FileHandle

dagster.local_file_manager ResourceDefinition[source]

FileManager that provides abstract access to a local filesystem.

By default, files will be stored in <local_artifact_storage>/storage/file_manager where <local_artifact_storage> can be configured the dagster.yaml file in $DAGSTER_HOME.

Implements the FileManager API.

Examples

import tempfile

from dagster import job, local_file_manager, op


@op(required_resource_keys={"file_manager"})
def write_files(context):
    fh_1 = context.resources.file_manager.write_data(b"foo")

    with tempfile.NamedTemporaryFile("w+") as fd:
        fd.write("bar")
        fd.seek(0)
        fh_2 = context.resources.file_manager.write(fd, mode="w", ext=".txt")

    return (fh_1, fh_2)


@op(required_resource_keys={"file_manager"})
def read_files(context, file_handles):
    fh_1, fh_2 = file_handles
    assert context.resources.file_manager.read_data(fh_2) == b"bar"
    fd = context.resources.file_manager.read(fh_2, mode="r")
    assert fd.read() == "foo"
    fd.close()


@job(resource_defs={"file_manager": local_file_manager})
def files_pipeline():
    read_files(write_files())

Or to specify the file directory:

@job(
    resource_defs={
        "file_manager": local_file_manager.configured({"base_dir": "/my/base/dir"})
    }
)
def files_pipeline():
    read_files(write_files())
class dagster.FileHandle[source]

A reference to a file as manipulated by a FileManager.

Subclasses may handle files that are resident on the local file system, in an object store, or in any arbitrary place where a file can be stored.

This exists to handle the very common case where you wish to write a computation that reads, transforms, and writes files, but where you also want the same code to work in local development as well as on a cluster where the files will be stored in a globally available object store such as S3.

abstract property path_desc

A representation of the file path for display purposes only.

class dagster.LocalFileHandle(path)[source]

A reference to a file on a local filesystem.

property path

The file’s path.

property path_desc

A representation of the file path for display purposes only.


Instance

class dagster.DagsterInstance(instance_type, local_artifact_storage, run_storage, event_storage, run_coordinator, compute_log_manager, run_launcher, scheduler=None, schedule_storage=None, settings=None, secrets_loader=None, ref=None, **_kwargs)[source]

Core abstraction for managing Dagster’s access to storage and other resources.

Use DagsterInstance.get() to grab the current DagsterInstance which will load based on the values in the dagster.yaml file in $DAGSTER_HOME.

Alternatively, DagsterInstance.ephemeral() can use used which provides a set of transient in-memory components.

Configuration of this class should be done by setting values in $DAGSTER_HOME/dagster.yaml. For example, to use Postgres for dagster storage, you can write a dagster.yaml such as the following:

dagster.yaml
storage:
  postgres:
    postgres_db:
      username: my_username
      password: my_password
      hostname: my_hostname
      db_name: my_database
      port: 5432
Parameters:
  • instance_type (InstanceType) – Indicates whether the instance is ephemeral or persistent. Users should not attempt to set this value directly or in their dagster.yaml files.

  • local_artifact_storage (LocalArtifactStorage) – The local artifact storage is used to configure storage for any artifacts that require a local disk, such as schedules, or when using the filesystem system storage to manage files and intermediates. By default, this will be a dagster._core.storage.root.LocalArtifactStorage. Configurable in dagster.yaml using the ConfigurableClass machinery.

  • run_storage (RunStorage) – The run storage is used to store metadata about ongoing and past pipeline runs. By default, this will be a dagster._core.storage.runs.SqliteRunStorage. Configurable in dagster.yaml using the ConfigurableClass machinery.

  • event_storage (EventLogStorage) – Used to store the structured event logs generated by pipeline runs. By default, this will be a dagster._core.storage.event_log.SqliteEventLogStorage. Configurable in dagster.yaml using the ConfigurableClass machinery.

  • compute_log_manager (Optional[ComputeLogManager]) – The compute log manager handles stdout and stderr logging for op compute functions. By default, this will be a dagster._core.storage.local_compute_log_manager.LocalComputeLogManager. Configurable in dagster.yaml using the ConfigurableClass machinery.

  • run_coordinator (Optional[RunCoordinator]) – A runs coordinator may be used to manage the execution of pipeline runs.

  • run_launcher (Optional[RunLauncher]) – Optionally, a run launcher may be used to enable a Dagster instance to launch pipeline runs, e.g. on a remote Kubernetes cluster, in addition to running them locally.

  • settings (Optional[Dict]) – Specifies certain per-instance settings, such as feature flags. These are set in the dagster.yaml under a set of whitelisted keys.

  • ref (Optional[InstanceRef]) – Used by internal machinery to pass instances across process boundaries.

add_dynamic_partitions(partitions_def_name, partition_keys)[source]

Add partitions to the specified DynamicPartitionsDefinition idempotently. Does not add any partitions that already exist.

Parameters:
  • partitions_def_name (str) – The name of the DynamicPartitionsDefinition.

  • partition_keys (Sequence[str]) – Partition keys to add.

delete_dynamic_partition(partitions_def_name, partition_key)[source]

Delete a partition for the specified DynamicPartitionsDefinition. If the partition does not exist, exits silently.

Parameters:
  • partitions_def_name (str) – The name of the DynamicPartitionsDefinition.

  • partition_key (str) – Partition key to delete.

delete_run(run_id)[source]

Delete a run and all events generated by that from storage.

Parameters:

run_id (str) – The id of the run to delete.

static ephemeral(tempdir=None, preload=None, settings=None)[source]

Create a DagsterInstance suitable for ephemeral execution, useful in test contexts. An ephemeral instance uses mostly in-memory components. Use local_temp to create a test instance that is fully persistent.

Parameters:
  • tempdir (Optional[str]) – The path of a directory to be used for local artifact storage.

  • preload (Optional[Sequence[DebugRunPayload]]) – A sequence of payloads to load into the instance’s run storage. Useful for debugging.

  • settings (Optional[Dict]) – Settings for the instance.

Returns:

An ephemeral DagsterInstance.

Return type:

DagsterInstance

fetch_materializations(records_filter, limit, cursor=None, ascending=False)[source]

Return a list of materialization records stored in the event log storage.

Parameters:
  • records_filter (Union[AssetKey, AssetRecordsFilter]) – the filter by which to filter event records.

  • limit (int) – Number of results to get.

  • cursor (Optional[str]) – Cursor to use for pagination. Defaults to None.

  • ascending (Optional[bool]) – Sort the result in ascending order if True, descending otherwise. Defaults to descending.

Returns:

Object containing a list of event log records and a cursor string

Return type:

EventRecordsResult

fetch_observations(records_filter, limit, cursor=None, ascending=False)[source]

Return a list of observation records stored in the event log storage.

Parameters:
  • records_filter (Optional[Union[AssetKey, AssetRecordsFilter]]) – the filter by which to filter event records.

  • limit (int) – Number of results to get.

  • cursor (Optional[str]) – Cursor to use for pagination. Defaults to None.

  • ascending (Optional[bool]) – Sort the result in ascending order if True, descending otherwise. Defaults to descending.

Returns:

Object containing a list of event log records and a cursor string

Return type:

EventRecordsResult

fetch_run_status_changes(records_filter, limit, cursor=None, ascending=False)[source]

Return a list of run_status_event records stored in the event log storage.

Parameters:
  • records_filter (Optional[Union[DagsterEventType, RunStatusChangeRecordsFilter]]) – the filter by which to filter event records.

  • limit (int) – Number of results to get.

  • cursor (Optional[str]) – Cursor to use for pagination. Defaults to None.

  • ascending (Optional[bool]) – Sort the result in ascending order if True, descending otherwise. Defaults to descending.

Returns:

Object containing a list of event log records and a cursor string

Return type:

EventRecordsResult

static get()[source]

Get the current DagsterInstance as specified by the DAGSTER_HOME environment variable.

Returns:

The current DagsterInstance.

Return type:

DagsterInstance

get_asset_keys(prefix=None, limit=None, cursor=None)[source]

Return a filtered subset of asset keys managed by this instance.

Parameters:
  • prefix (Optional[Sequence[str]]) – Return only assets having this key prefix.

  • limit (Optional[int]) – Maximum number of keys to return.

  • cursor (Optional[str]) – Cursor to use for pagination.

Returns:

List of asset keys.

Return type:

Sequence[AssetKey]

get_asset_records(asset_keys=None)[source]

Return an AssetRecord for each of the given asset keys.

Parameters:

asset_keys (Optional[Sequence[AssetKey]]) – List of asset keys to retrieve records for.

Returns:

List of asset records.

Return type:

Sequence[AssetRecord]

get_dynamic_partitions(partitions_def_name)[source]

Get the set of partition keys for the specified DynamicPartitionsDefinition.

Parameters:

partitions_def_name (str) – The name of the DynamicPartitionsDefinition.

get_event_records(event_records_filter, limit=None, ascending=False)[source]

Return a list of event records stored in the event log storage.

Parameters:
  • event_records_filter (Optional[EventRecordsFilter]) – the filter by which to filter event records.

  • limit (Optional[int]) – Number of results to get. Defaults to infinite.

  • ascending (Optional[bool]) – Sort the result in ascending order if True, descending otherwise. Defaults to descending.

Returns:

List of event log records stored in the event log storage.

Return type:

List[EventLogRecord]

get_latest_materialization_code_versions(asset_keys)[source]

Returns the code version used for the latest materialization of each of the provided assets.

Parameters:

asset_keys (Iterable[AssetKey]) – The asset keys to find latest materialization code versions for.

Returns:

A dictionary with a key for each of the provided asset

keys. The values will be None if the asset has no materializations. If an asset does not have a code version explicitly assigned to its definitions, but was materialized, Dagster assigns the run ID as its code version.

Return type:

Mapping[AssetKey, Optional[str]]

get_latest_materialization_event(asset_key)[source]

Fetch the latest materialization event for the given asset key.

Parameters:

asset_key (AssetKey) – Asset key to return materialization for.

Returns:

The latest materialization event for the given asset

key, or None if the asset has not been materialized.

Return type:

Optional[EventLogEntry]

get_run_by_id(run_id)[source]

Get a DagsterRun matching the provided run_id.

Parameters:

run_id (str) – The id of the run to retrieve.

Returns:

The run corresponding to the given id. If no run matching the id

is found, return None.

Return type:

Optional[DagsterRun]

get_run_record_by_id(run_id)[source]

Get a RunRecord matching the provided run_id.

Parameters:

run_id (str) – The id of the run record to retrieve.

Returns:

The run record corresponding to the given id. If no run matching

the id is found, return None.

Return type:

Optional[RunRecord]

get_run_records(filters=None, limit=None, order_by=None, ascending=False, cursor=None, bucket_by=None)[source]

Return a list of run records stored in the run storage, sorted by the given column in given order.

Parameters:
  • filters (Optional[RunsFilter]) – the filter by which to filter runs.

  • limit (Optional[int]) – Number of results to get. Defaults to infinite.

  • order_by (Optional[str]) – Name of the column to sort by. Defaults to id.

  • ascending (Optional[bool]) – Sort the result in ascending order if True, descending otherwise. Defaults to descending.

Returns:

List of run records stored in the run storage.

Return type:

List[RunRecord]

get_status_by_partition(asset_key, partition_keys, partitions_def)[source]

Get the current status of provided partition_keys for the provided asset.

Parameters:
  • asset_key (AssetKey) – The asset to get per-partition status for.

  • partition_keys (Sequence[str]) – The partitions to get status for.

  • partitions_def (PartitionsDefinition) – The PartitionsDefinition of the asset to get per-partition status for.

Returns:

status for each partition key

Return type:

Optional[Mapping[str, AssetPartitionStatus]]

has_asset_key(asset_key)[source]

Return true if this instance manages the given asset key.

Parameters:

asset_key (AssetKey) – Asset key to check.

has_dynamic_partition(partitions_def_name, partition_key)[source]

Check if a partition key exists for the DynamicPartitionsDefinition.

Parameters:
  • partitions_def_name (str) – The name of the DynamicPartitionsDefinition.

  • partition_key (Sequence[str]) – Partition key to check.

static local_temp(tempdir=None, overrides=None)[source]

Create a DagsterInstance that uses a temporary directory for local storage. This is a regular, fully persistent instance. Use ephemeral to get an ephemeral instance with in-memory components.

Parameters:
  • tempdir (Optional[str]) – The path of a directory to be used for local artifact storage.

  • overrides (Optional[DagsterInstanceOverrides]) – Override settings for the instance.

Returns:

DagsterInstance

report_runless_asset_event(asset_event)[source]

experimental This API may break in future versions, even between dot releases.

Record an event log entry related to assets that does not belong to a Dagster run.

wipe_assets(asset_keys)[source]

Wipes asset event history from the event log for the given asset keys.

Parameters:

asset_keys (Sequence[AssetKey]) – Asset keys to wipe.

class dagster._core.instance.InstanceRef(local_artifact_storage_data, compute_logs_data, scheduler_data, run_coordinator_data, run_launcher_data, settings, run_storage_data, event_storage_data, schedule_storage_data, custom_instance_class_data=None, storage_data=None, secrets_loader_data=None)[source]

Serializable representation of a DagsterInstance.

Users should not instantiate this class directly.

class dagster._serdes.ConfigurableClass[source]

Abstract mixin for classes that can be loaded from config.

This supports a powerful plugin pattern which avoids both a) a lengthy, hard-to-synchronize list of conditional imports / optional extras_requires in dagster core and b) a magic directory or file in which third parties can place plugin packages. Instead, the intention is to make, e.g., run storage, pluggable with a config chunk like:

run_storage:
    module: very_cool_package.run_storage
    class: SplendidRunStorage
    config:
        magic_word: "quux"

This same pattern should eventually be viable for other system components, e.g. engines.

The ConfigurableClass mixin provides the necessary hooks for classes to be instantiated from an instance of ConfigurableClassData.

Pieces of the Dagster system which we wish to make pluggable in this way should consume a config type such as:

{'module': str, 'class': str, 'config': Field(Permissive())}
class dagster._serdes.ConfigurableClassData(module_name, class_name, config_yaml)[source]

Serializable tuple describing where to find a class and the config fragment that should be used to instantiate it.

Users should not instantiate this class directly.

Classes intended to be serialized in this way should implement the dagster.serdes.ConfigurableClass mixin.

class dagster._core.storage.root.LocalArtifactStorage(base_dir, inst_data=None)[source]

Storage

class dagster._core.storage.base_storage.DagsterStorage[source]

Abstract base class for Dagster persistent storage, for reading and writing data for runs, events, and schedule/sensor state.

Users should not directly instantiate concrete subclasses of this class; they are instantiated by internal machinery when dagster-webserver and dagster-daemon load, based on the values in the dagster.yaml file in $DAGSTER_HOME. Configuration of concrete subclasses of this class should be done by setting values in that file.


Run storage

class dagster.DagsterRun(job_name, run_id=None, run_config=None, asset_selection=None, asset_check_selection=None, op_selection=None, resolved_op_selection=None, step_keys_to_execute=None, status=None, tags=None, root_run_id=None, parent_run_id=None, job_snapshot_id=None, execution_plan_snapshot_id=None, remote_job_origin=None, job_code_origin=None, has_repository_load_data=None, run_op_concurrency=None)[source]

Serializable internal representation of a dagster run, as stored in a RunStorage.

job_name

The name of the job executed in this run

Type:

str

run_id

The ID of the run

Type:

str

run_config

The config for the run

Type:

Mapping[str, object]

tags

The tags applied to the run

Type:

Mapping[str, str]

property is_cancelable

If this run an be canceled.

Type:

bool

property is_failure

If this run has failed.

Type:

bool

property is_failure_or_canceled

If this run has either failed or was canceled.

Type:

bool

property is_finished

If this run has completely finished execution.

Type:

bool

property is_resume_retry

If this run was created from retrying another run from the point of failure.

Type:

bool

property is_success

If this run has successfully finished executing.

Type:

bool

class dagster.DagsterRunStatus(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)[source]

The status of run execution.

class dagster.RunsFilter(run_ids=None, job_name=None, statuses=None, tags=None, snapshot_id=None, updated_after=None, updated_before=None, created_after=None, created_before=None, exclude_subruns=None)[source]

Defines a filter across job runs, for use when querying storage directly.

Each field of the RunsFilter represents a logical AND with each other. For example, if you specify job_name and tags, then you will receive only runs with the specified job_name AND the specified tags. If left blank, then all values will be permitted for that field.

Parameters:
  • run_ids (Optional[List[str]]) – A list of job run_id values.

  • job_name (Optional[str]) – Name of the job to query for. If blank, all job_names will be accepted.

  • statuses (Optional[List[DagsterRunStatus]]) – A list of run statuses to filter by. If blank, all run statuses will be allowed.

  • tags (Optional[Dict[str, Union[str, List[str]]]]) – A dictionary of run tags to query by. All tags specified here must be present for a given run to pass the filter.

  • snapshot_id (Optional[str]) – The ID of the job snapshot to query for. Intended for internal use.

  • updated_after (Optional[DateTime]) – Filter by runs that were last updated before this datetime.

  • created_before (Optional[DateTime]) – Filter by runs that were created before this datetime.

  • exclude_subruns (Optional[bool]) – If true, runs that were launched to backfill historical data will be excluded from results.

class dagster._core.storage.runs.RunStorage[source]

Abstract base class for storing pipeline run history.

Note that run storages using SQL databases as backing stores should implement SqlRunStorage.

Users should not directly instantiate concrete subclasses of this class; they are instantiated by internal machinery when dagster-webserver and dagster-graphql load, based on the values in the dagster.yaml file in $DAGSTER_HOME. Configuration of concrete subclasses of this class should be done by setting values in that file.

class dagster._core.storage.runs.SqlRunStorage[source]

Base class for SQL based run storages.

class dagster._core.storage.runs.SqliteRunStorage(conn_string, inst_data=None)[source]

SQLite-backed run storage.

Users should not directly instantiate this class; it is instantiated by internal machinery when dagster-webserver and dagster-graphql load, based on the values in the dagster.yaml file in $DAGSTER_HOME. Configuration of this class should be done by setting values in that file.

This is the default run storage when none is specified in the dagster.yaml.

To explicitly specify SQLite for run storage, you can add a block such as the following to your dagster.yaml:

run_storage:
  module: dagster._core.storage.runs
  class: SqliteRunStorage
  config:
    base_dir: /path/to/dir

The base_dir param tells the run storage where on disk to store the database.

class dagster._core.storage.dagster_run.RunRecord(storage_id, dagster_run, create_timestamp, update_timestamp, start_time=None, end_time=None)[source]

Internal representation of a run record, as stored in a RunStorage.

Users should not invoke this class directly.

See also: dagster_postgres.PostgresRunStorage and dagster_mysql.MySQLRunStorage.


Event log storage

class dagster.EventLogEntry(error_info, level, user_message, run_id, timestamp, step_key=None, job_name=None, dagster_event=None)[source]

Entries in the event log.

Users should not instantiate this object directly. These entries may originate from the logging machinery (DagsterLogManager/context.log), from framework events (e.g. EngineEvent), or they may correspond to events yielded by user code (e.g. Output).

Parameters:
  • error_info (Optional[SerializableErrorInfo]) – Error info for an associated exception, if any, as generated by serializable_error_info_from_exc_info and friends.

  • level (Union[str, int]) – The Python log level at which to log this event. Note that framework and user code events are also logged to Python logging. This value may be an integer or a (case-insensitive) string member of PYTHON_LOGGING_LEVELS_NAMES.

  • user_message (str) – For log messages, this is the user-generated message.

  • run_id (str) – The id of the run which generated this event.

  • timestamp (float) – The Unix timestamp of this event.

  • step_key (Optional[str]) – The step key for the step which generated this event. Some events are generated outside of a step context.

  • job_name (Optional[str]) – The job which generated this event. Some events are generated outside of a job context.

  • dagster_event (Optional[DagsterEvent]) – For framework and user events, the associated structured event.

get_dagster_event()[source]

DagsterEvent: Returns the DagsterEvent contained within this entry. If this entry does not contain a DagsterEvent, an error will be raised.

property dagster_event_type

The type of the DagsterEvent contained by this entry, if any.

Type:

Optional[DagsterEventType]

property is_dagster_event

If this entry contains a DagsterEvent.

Type:

bool

property message

Return the message from the structured DagsterEvent if present, fallback to user_message.

class dagster.EventLogRecord(storage_id, event_log_entry)[source]

Internal representation of an event record, as stored in a EventLogStorage.

Users should not instantiate this class directly.

class dagster.EventRecordsFilter(event_type, asset_key=None, asset_partitions=None, after_cursor=None, before_cursor=None, after_timestamp=None, before_timestamp=None, storage_ids=None)[source]

Defines a set of filter fields for fetching a set of event log entries or event log records.

Parameters:
  • event_type (DagsterEventType) – Filter argument for dagster event type

  • asset_key (Optional[AssetKey]) – Asset key for which to get asset materialization event entries / records.

  • asset_partitions (Optional[List[str]]) – Filter parameter such that only asset events with a partition value matching one of the provided values. Only valid when the asset_key parameter is provided.

  • after_cursor (Optional[EventCursor]) – Filter parameter such that only records with storage_id greater than the provided value are returned. Using a run-sharded events cursor will result in a significant performance gain when run against a SqliteEventLogStorage implementation (which is run-sharded)

  • before_cursor (Optional[EventCursor]) – Filter parameter such that records with storage_id less than the provided value are returned. Using a run-sharded events cursor will result in a significant performance gain when run against a SqliteEventLogStorage implementation (which is run-sharded)

  • after_timestamp (Optional[float]) – Filter parameter such that only event records for events with timestamp greater than the provided value are returned.

  • before_timestamp (Optional[float]) – Filter parameter such that only event records for events with timestamp less than the provided value are returned.

class dagster.RunShardedEventsCursor(id, run_updated_after)[source]

Pairs an id-based event log cursor with a timestamp-based run cursor, for improved performance on run-sharded event log storages (e.g. the default SqliteEventLogStorage). For run-sharded storages, the id field is ignored, since they may not be unique across shards.

class dagster._core.storage.event_log.EventLogStorage[source]

Abstract base class for storing structured event logs from pipeline runs.

Note that event log storages using SQL databases as backing stores should implement SqlEventLogStorage.

Users should not directly instantiate concrete subclasses of this class; they are instantiated by internal machinery when dagster-webserver and dagster-graphql load, based on the values in the dagster.yaml file in $DAGSTER_HOME. Configuration of concrete subclasses of this class should be done by setting values in that file.

class dagster._core.storage.event_log.SqlEventLogStorage[source]

Base class for SQL backed event log storages.

Distinguishes between run-based connections and index connections in order to support run-level sharding, while maintaining the ability to do cross-run queries

class dagster._core.storage.event_log.SqliteEventLogStorage(base_dir, inst_data=None)[source]

SQLite-backed event log storage.

Users should not directly instantiate this class; it is instantiated by internal machinery when dagster-webserver and dagster-graphql load, based on the values in the dagster.yaml file insqliteve $DAGSTER_HOME. Configuration of this class should be done by setting values in that file.

This is the default event log storage when none is specified in the dagster.yaml.

To explicitly specify SQLite for event log storage, you can add a block such as the following to your dagster.yaml:

event_log_storage:
  module: dagster._core.storage.event_log
  class: SqliteEventLogStorage
  config:
    base_dir: /path/to/dir

The base_dir param tells the event log storage where on disk to store the databases. To improve concurrent performance, event logs are stored in a separate SQLite database for each run.

class dagster._core.storage.event_log.ConsolidatedSqliteEventLogStorage(base_dir, inst_data=None)[source]

SQLite-backed consolidated event log storage intended for test cases only.

Users should not directly instantiate this class; it is instantiated by internal machinery when dagster-webserver and dagster-graphql load, based on the values in the dagster.yaml file in $DAGSTER_HOME. Configuration of this class should be done by setting values in that file.

To explicitly specify the consolidated SQLite for event log storage, you can add a block such as the following to your dagster.yaml:

run_storage:
  module: dagster._core.storage.event_log
  class: ConsolidatedSqliteEventLogStorage
  config:
    base_dir: /path/to/dir

The base_dir param tells the event log storage where on disk to store the database.

class dagster._core.storage.event_log.AssetRecord(storage_id, asset_entry)[source]

Internal representation of an asset record, as stored in a EventLogStorage.

Users should not invoke this class directly.

See also: dagster_postgres.PostgresEventLogStorage and dagster_mysql.MySQLEventLogStorage.


Compute log manager

class dagster._core.storage.compute_log_manager.ComputeLogManager[source]

Abstract base class for capturing the unstructured logs (stdout/stderr) in the current process, stored / retrieved with a provided log_key.

class dagster._core.storage.local_compute_log_manager.LocalComputeLogManager(base_dir, polling_timeout=None, inst_data=None)[source]

Stores copies of stdout & stderr for each compute step locally on disk.

class dagster._core.storage.noop_compute_log_manager.NoOpComputeLogManager(inst_data=None)[source]

When enabled for a Dagster instance, stdout and stderr will not be available for any step.

See also: dagster_aws.S3ComputeLogManager.


Run launcher

class dagster._core.launcher.RunLauncher[source]
class dagster._core.launcher.DefaultRunLauncher(inst_data=None)[source]

Launches runs against running GRPC servers.


Run coordinator

class dagster._core.run_coordinator.DefaultRunCoordinator(inst_data=None)[source]

Immediately send runs to the run launcher.

dagster._core.run_coordinator.QueuedRunCoordinator RunCoordinator[source]

Config Schema:
max_concurrent_runs (dagster.IntSource, optional):

The maximum number of runs that are allowed to be in progress at once. Defaults to 10. Set to -1 to disable the limit. Set to 0 to stop any runs from launching. Any other negative values are disallowed.

tag_concurrency_limits (Union[List[strict dict], None], optional):

A set of limits that are applied to runs with particular tags. If a value is set, the limit is applied to only that key-value pair. If no value is set, the limit is applied across all values of that key. If the value is set to a dict with applyLimitPerUniqueValue: true, the limit will apply to the number of unique values for that key.

dequeue_interval_seconds (dagster.IntSource, optional):

The interval in seconds at which the Dagster Daemon should periodically check the run queue for new runs to launch.

dequeue_use_threads (Bool, optional):

Whether or not to use threads for concurrency when launching dequeued runs.

dequeue_num_workers (dagster.IntSource, optional):

If dequeue_use_threads is true, limit the number of concurrent worker threads.

max_user_code_failure_retries (dagster.IntSource, optional):

If there is an error reaching a Dagster gRPC server while dequeuing the run, how many times to retry the dequeue before failing it. The only run launcher that requires the gRPC server to be running is the DefaultRunLauncher, so setting this will have no effect unless that run launcher is being used.

Default Value: 0

user_code_failure_retry_delay (dagster.IntSource, optional):

If there is an error reaching a Dagster gRPC server while dequeuing the run, how long to wait before retrying any runs from that same code location. The only run launcher that requires the gRPC server to be running is the DefaultRunLauncher, so setting this will have no effect unless that run launcher is being used.

Default Value: 60

block_op_concurrency_limited_runs (strict dict, optional):
Default Value:
{}
Config Schema:
enabled (Bool, optional):

op_concurrency_slot_buffer (Int, optional):

Determines whether or not a run will be dequeued if it consists of ops that will all be initially blocked waiting for global op concurrency slots to be free.

Enqueues runs via the run storage, to be deqeueued by the Dagster Daemon process. Requires the Dagster Daemon process to be alive in order for runs to be launched.


Scheduling

class dagster._core.scheduler.Scheduler[source]

Abstract base class for a scheduler. This component is responsible for interfacing with an external system such as cron to ensure scheduled repeated execution according.

class dagster._core.storage.schedules.ScheduleStorage[source]

Abstract class for managing persistance of scheduler artifacts.

class dagster._core.storage.schedules.SqlScheduleStorage[source]

Base class for SQL backed schedule storage.

class dagster._core.storage.schedules.SqliteScheduleStorage(conn_string, inst_data=None)[source]

Local SQLite backed schedule storage.

see also: dagster_postgres.PostgresScheduleStorage and dagster_mysql.MySQLScheduleStorage.


Exception handling

dagster._core.errors.user_code_error_boundary(error_cls, msg_fn, log_manager=None, **kwargs)[source]

Wraps the execution of user-space code in an error boundary. This places a uniform policy around any user code invoked by the framework. This ensures that all user errors are wrapped in an exception derived from DagsterUserCodeExecutionError, and that the original stack trace of the user error is preserved, so that it can be reported without confusing framework code in the stack trace, if a tool author wishes to do so.

Examples: .. code-block:: python

with user_code_error_boundary(

# Pass a class that inherits from DagsterUserCodeExecutionError DagsterExecutionStepExecutionError, # Pass a function that produces a message “Error occurred during step execution”

):

call_user_provided_function()


Step Launchers (Superseded)

Learn how to migrate from Step Launchers to Dagster Pipes in the migration guide.

class dagster.StepLauncher(*args, **kwargs)[source]

superseded This API has been superseded and its usage is discouraged.

Consider using Dagster Pipes instead. Learn more here: https://docs.dagster.io/concepts/dagster-pipes.

A StepLauncher is responsible for executing steps, either in-process or in an external process.

class dagster.StepRunRef(run_config, dagster_run, run_id, retry_mode, step_key, recon_job, known_state)[source]

A serializable object that specifies what’s needed to hydrate a step so that it can be executed in a process outside the plan process.

Users should not instantiate this class directly.

class dagster.StepExecutionContext(plan_data, execution_data, log_manager, step, output_capture, known_state)[source]

Context for the execution of a step. Users should not instantiate this class directly.

This context assumes that user code can be run directly, and thus includes resource and information.