SqlRegistry

class lsst.daf.butler.registries.sqlRegistry.SqlRegistry(registryConfig, schemaConfig, dimensionConfig, create=False)

Bases: lsst.daf.butler.Registry

Registry backed by a SQL database.

Parameters:
registryConfig : SqlRegistryConfig or str

Load configuration

schemaConfig : SchemaConfig or str

Definition of the schema to use.

dimensionConfig : DimensionConfig or Config or

DimensionGraph configuration.

create : bool

Assume registry is empty and create a new one.

Attributes Summary

defaultConfigFile Path to configuration defaults.
limited If True, this Registry does not maintain Dimension metadata or relationships (bool).
pixelization Object that interprets SkyPix Dimension values (lsst.sphgeom.Pixelization).

Methods Summary

addDataset(datasetType, dataId, run[, …]) Adds a Dataset entry to the Registry
addDatasetLocation(ref, datastoreName) Add datastore name locating a given dataset.
addDimensionEntry(dimension[, dataId, entry]) Add a new Dimension entry.
addExecution(execution) Add a new Execution to the SqlRegistry.
addQuantum(quantum) Add a new Quantum to the SqlRegistry.
addRun(run) Add a new Run to the SqlRegistry.
associate(collection, refs) Add existing Datasets to a collection, possibly creating the collection in the process.
attachComponent(name, parent, component) Attach a component to a dataset.
disassociate(collection, refs[, remove]) Remove existing Datasets from a collection.
ensureRun(run) Conditionally add a new Run to the SqlRegistry.
find(collection, datasetType[, dataId]) Lookup a dataset.
findDimensionEntry(dimension[, dataId]) Return a Dimension entry corresponding to a DataId.
fromConfig(registryConfig[, schemaConfig, …]) Create Registry subclass instance from config.
getDataset(id) Retrieve a Dataset entry.
getDatasetLocations(ref) Retrieve datastore locations for a given dataset.
getDatasetType(name) Get the DatasetType.
getExecution(id) Retrieve an Execution.
getQuantum(id) Retrieve an Quantum.
getRun([id, collection]) Get a Run corresponding to its collection or id
makeDatabaseDict(table, types, key, value) Construct a DatabaseDict backed by a table in the same database as this Registry.
makeRun(collection) Create a new Run in the SqlRegistry and return it.
markInputUsed(quantum, ref) Record the given DatasetRef as an actual (not just predicted) input of the given Quantum.
query(sql, **params) Execute a SQL SELECT statement directly.
queryDataId([dataId, dimension, metadata, …]) Expand a data ID to include additional information.
registerDatasetType(datasetType) Add a new DatasetType to the SqlRegistry.
removeDatasetLocation(datastoreName, ref) Remove datastore location associated with this dataset.
selectDimensions(originInfo, expression, …) Evaluate a filter expression and lists of DatasetTypes and return a set of data unit values.
setConfigRoot(root, config, full) Set any filesystem-dependent config options for this Registry to be appropriate for a new empty repository with the given root.
setDimensionRegion([dataId, update, region]) Set the region field for a Dimension instance or a combination thereof and update associated spatial join tables.
transaction() Context manager that implements SQL transactions.

Attributes Documentation

defaultConfigFile = None

Path to configuration defaults. Relative to $DAF_BUTLER_DIR/config or absolute path. Can be None if no defaults specified.

limited

If True, this Registry does not maintain Dimension metadata or relationships (bool).

pixelization

Object that interprets SkyPix Dimension values (lsst.sphgeom.Pixelization).

None for limited registries.

Methods Documentation

addDataset(datasetType, dataId, run, producer=None, recursive=False, **kwds)

Adds a Dataset entry to the Registry

This always adds a new Dataset; to associate an existing Dataset with a new collection, use associate.

Parameters:
datasetType : DatasetType or str

A DatasetType or the name of one.

dataId : dict or DataId

A dict-like object containing the Dimension links that identify the dataset within a collection.

run : Run

The Run instance that produced the Dataset. Ignored if producer is passed (producer.run is then used instead). A Run must be provided by one of the two arguments.

producer : Quantum

Unit of work that produced the Dataset. May be None to store no provenance information, but if present the Quantum must already have been added to the Registry.

recursive : bool

If True, recursively add Dataset and attach entries for component Datasets as well.

kwds

Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

Returns:
ref : DatasetRef

A newly-created DatasetRef instance.

Raises:
ValueError

If a Dataset with the given DatasetRef already exists in the given collection.

Exception

If dataId contains unknown or invalid Dimension entries.

addDatasetLocation(ref, datastoreName)

Add datastore name locating a given dataset.

Typically used by Datastore.

Parameters:
ref : DatasetRef

A reference to the dataset for which to add storage information.

datastoreName : str

Name of the datastore holding this dataset.

addDimensionEntry(dimension, dataId=None, entry=None, **kwds)

Add a new Dimension entry.

dimension : str or Dimension
Either a Dimension object or the name of one.
dataId : dict or DataId, optional
A dict-like object containing the Dimension links that form the primary key of the row to insert. If this is a full DataId object, dataId.entries[dimension] will be updated with entry and then inserted into the Registry.
entry : dict
Dictionary that maps column name to column value.
kwds
Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

If values includes a “region” key, setDimensionRegion will automatically be called to set it any associated spatial join tables. Region fields associated with a combination of Dimensions must be explicitly set separately.

Returns:
dataId : DataId

A Data ID for exactly the given dimension that includes the added entry.

Raises:
TypeError

If the given Dimension does not have explicit entries in the registry.

ValueError

If an entry with the primary-key defined in values is already present.

NotImplementedError

Raised if limited is True.

addExecution(execution)

Add a new Execution to the SqlRegistry.

If execution.id is None the SqlRegistry will update it to that of the newly inserted entry.

Parameters:
execution : Execution

Instance to add to the SqlRegistry. The given Execution must not already be present in the SqlRegistry.

Raises:
Exception

If Execution is already present in the SqlRegistry.

addQuantum(quantum)

Add a new Quantum to the SqlRegistry.

Parameters:
quantum : Quantum

Instance to add to the SqlRegistry. The given Quantum must not already be present in the SqlRegistry (or any other), therefore its:

  • run attribute must be set to an existing Run.
  • predictedInputs attribute must be fully populated with DatasetRefs, and its.
  • actualInputs and outputs will be ignored.
addRun(run)

Add a new Run to the SqlRegistry.

Parameters:
run : Run

Instance to add to the SqlRegistry. The given Run must not already be present in the SqlRegistry (or any other). Therefore its id must be None and its collection must not be associated with any existing Run.

Raises:
ValueError

If a run already exists with this collection.

associate(collection, refs)

Add existing Datasets to a collection, possibly creating the collection in the process.

If a DatasetRef with the same exact dataset_id` is already in a collection nothing is changed. If a DatasetRef with the same DatasetType and dimension values but with different dataset_id exists in a collection then exception is raised.

Parameters:
collection : str

Indicates the collection the Datasets should be associated with.

refs : list of DatasetRef

A list of DatasetRef instances that already exist in this SqlRegistry.

Raises:
ValueError

If a Dataset with the given DatasetRef already exists in the given collection.

attachComponent(name, parent, component)

Attach a component to a dataset.

Parameters:
name : str

Name of the component.

parent : DatasetRef

A reference to the parent dataset. Will be updated to reference the component.

component : DatasetRef

A reference to the component dataset.

disassociate(collection, refs, remove=True)

Remove existing Datasets from a collection.

collection and ref combinations that are not currently associated are silently ignored.

Parameters:
collection : str

The collection the Datasets should no longer be associated with.

refs : list of DatasetRef

A list of DatasetRef instances that already exist in this SqlRegistry.

remove : bool

If True, remove Datasets from the SqlRegistry if they are not associated with any collection (including via any composites).

Returns:
removed : list of DatasetRef

If remove is True, the list of DatasetRefs that were removed.

ensureRun(run)

Conditionally add a new Run to the SqlRegistry.

If the run.id is None or a Run with this id doesn’t exist in the Registry yet, add it. Otherwise, ensure the provided run is identical to the one already in the registry.

Parameters:
run : Run

Instance to add to the SqlRegistry.

Raises:
ValueError

If run already exists, but is not identical.

find(collection, datasetType, dataId=None, **kwds)

Lookup a dataset.

This can be used to obtain a DatasetRef that permits the dataset to be read from a Datastore.

Parameters:
collection : str

Identifies the collection to search.

datasetType : DatasetType or str

A DatasetType or the name of one.

dataId : dict or DataId, optional

A dict-like object containing the Dimension links that identify the dataset within a collection.

kwds

Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

Returns:
ref : DatasetRef

A ref to the Dataset, or None if no matching Dataset was found.

Raises:
ValueError

If dataId is invalid.

findDimensionEntry(dimension, dataId=None, **kwds)

Return a Dimension entry corresponding to a DataId.

Parameters:
dimension : str or Dimension

Either a Dimension object or the name of one.

dataId : dict or DataId, optional

A dict-like object containing the Dimension links that form the primary key of the row to retreive. If this is a full DataId object, dataId.entries[dimension] will be updated with the entry obtained from the Registry.

kwds

Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

Returns:
entry : dict

Dictionary with all Dimension values, or None if no matching entry is found. None if there is no entry for the given DataId.

Raises:
NotImplementedError

Raised if limited is True.

static fromConfig(registryConfig, schemaConfig=None, dimensionConfig=None, create=False)

Create Registry subclass instance from config.

Uses registry.cls from config to determine which subclass to instantiate.

Parameters:
registryConfig : ButlerConfig, RegistryConfig, Config or str

Registry configuration

schemaConfig : SchemaConfig, Config or str, optional.

Schema configuration. Can be read from supplied registryConfig if the relevant component is defined and schemaConfig is None.

dimensionConfig : DimensionConfig or Config or

str, optional. DimensionGraph configuration. Can be read from supplied registryConfig if the relevant component is defined and dimensionConfig is None.

create : bool

Assume empty Registry and create a new one.

Returns:
registry : Registry (subclass)

A new Registry subclass instance.

getDataset(id)

Retrieve a Dataset entry.

Parameters:
id : int

The unique identifier for the Dataset.

Returns:
ref : DatasetRef

A ref to the Dataset, or None if no matching Dataset was found.

getDatasetLocations(ref)

Retrieve datastore locations for a given dataset.

Typically used by Datastore.

Parameters:
ref : DatasetRef

A reference to the dataset for which to retrieve storage information.

Returns:
datastores : set of str

All the matching datastores holding this dataset. Empty set if the dataset does not exist anywhere.

getDatasetType(name)

Get the DatasetType.

Parameters:
name : str

Name of the type.

Returns:
type : DatasetType

The DatasetType associated with the given name.

Raises:
KeyError

Requested named DatasetType could not be found in registry.

getExecution(id)

Retrieve an Execution.

Parameters:
id : int

The unique identifier for the Execution.

getQuantum(id)

Retrieve an Quantum.

Parameters:
id : int

The unique identifier for the Quantum.

getRun(id=None, collection=None)

Get a Run corresponding to its collection or id

Parameters:
id : int, optional

Lookup by run id, or:

collection : str

If given, lookup by collection name instead.

Returns:
run : Run

The Run instance.

Raises:
ValueError

Must supply one of collection or id.

makeDatabaseDict(table, types, key, value)

Construct a DatabaseDict backed by a table in the same database as this Registry.

Parameters:
table : table

Name of the table that backs the returned DatabaseDict. If this table already exists, its schema must include at least everything in types.

types : dict

A dictionary mapping str field names to type objects, containing all fields to be held in the database.

key : str

The name of the field to be used as the dictionary key. Must not be present in value._fields.

value : type

The type used for the dictionary’s values, typically a namedtuple. Must have a _fields class attribute that is a tuple of field names (i.e. as defined by namedtuple); these field names must also appear in the types arg, and a _make attribute to construct it from a sequence of values (again, as defined by namedtuple).

makeRun(collection)

Create a new Run in the SqlRegistry and return it.

If a run with this collection already exists, return that instead.

Parameters:
collection : str

The collection used to identify all inputs and outputs of the Run.

Returns:
run : Run

A new Run instance.

markInputUsed(quantum, ref)

Record the given DatasetRef as an actual (not just predicted) input of the given Quantum.

This updates both the SqlRegistry”s Quantum table and the Python Quantum.actualInputs attribute.

Parameters:
quantum : Quantum

Producer to update. Will be updated in this call.

ref : DatasetRef

To set as actually used input.

Raises:
KeyError

If quantum is not a predicted consumer for ref.

query(sql, **params)

Execute a SQL SELECT statement directly.

Named parameters are specified in the SQL query string by preceeding them with a colon. Parameter values are provided as additional keyword arguments. For example:

registry.query(“SELECT * FROM Instrument WHERE instrument=:name”, name=”HSC”)
Parameters:
sql : str

SQL query string. Must be a SELECT statement.

**params

Parameter name-value pairs to insert into the query.

Yields:
row : dict

The next row result from executing the query.

queryDataId(dataId=None, *, dimension=None, metadata=None, region=False, **kwds)

Expand a data ID to include additional information.

Parameters:
dataId : dict or DataId

A dict-like object containing the Dimension links that include the primary keys of the rows to query. If this is a true DataId, the object will be updated in-place.

dimension : Dimension or str

A dimension passed to the DataId constructor to create a true DataId or augment an existing one.

metadata : collections.abc.Mapping, optional

A mapping from Dimension or str name to column name, indicating fields to read into dataId.entries. If dimension is provided, may instead be a sequence of column names for that dimension.

region : bool

If True and the given DataId is uniquely associated with a region on the sky, obtain that region from the Registry and attach it as dataId.region.

kwds

Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

Returns:
dataId : DataId

A Data ID with its region attribute set.

Raises:
NotImplementedError

Raised if limited is True.

registerDatasetType(datasetType)

Add a new DatasetType to the SqlRegistry.

It is not an error to register the same DatasetType twice.

Parameters:
datasetType : DatasetType

The DatasetType to be added.

Returns:
inserted : bool

True if datasetType was inserted, False if an identical existing DatsetType was found.

Raises:
ValueError

DatasetType is not valid for this registry or is already registered but not identical.

removeDatasetLocation(datastoreName, ref)

Remove datastore location associated with this dataset.

Typically used by Datastore when a dataset is removed.

Parameters:
datastoreName : str

Name of this Datastore.

ref : DatasetRef

A reference to the dataset for which information is to be removed.

selectDimensions(originInfo, expression, neededDatasetTypes, futureDatasetTypes)

Evaluate a filter expression and lists of DatasetTypes and return a set of data unit values.

Returned set consists of combinations of units participating in data transformation from neededDatasetTypes to futureDatasetTypes, restricted by existing data and filter expression.

Parameters:
originInfo : DatasetOriginInfo

Object which provides names of the input/output collections.

expression : str

An expression that limits the Dimensions and (indirectly) the Datasets returned.

neededDatasetTypes : list of DatasetType or str

The list of DatasetTypes whose Dimensions will be included in the returned column set. Output is limited to the the Datasets of these DatasetTypes which already exist in the registry.

futureDatasetTypes : list of DatasetType or str

The list of DatasetTypes whose Dimensions will be included in the returned column set. It is expected that Datasets for these DatasetTypes do not exist in the registry, but presently this is not checked.

Yields:
row : PreFlightUnitsRow

Single row is a unique combination of units in a transform.

Raises:
NotImplementedError

Raised if limited is True.

classmethod setConfigRoot(root, config, full)

Set any filesystem-dependent config options for this Registry to be appropriate for a new empty repository with the given root.

Parameters:
root : str

Filesystem path to the root of the data repository.

config : Config

A Config to update. Only the subset understood by this component will be updated. Will not expand defaults.

full : Config

A complete config with all defaults expanded that can be converted to a RegistryConfig. Read-only and will not be modified by this method. Repository-specific options that should not be obtained from defaults when Butler instances are constructed should be copied from full to Config.

setDimensionRegion(dataId=None, *, update=True, region=None, **kwds)

Set the region field for a Dimension instance or a combination thereof and update associated spatial join tables.

Parameters:
dataId : dict or DataId

A dict-like object containing the Dimension links that form the primary key of the row to insert or update. If this is a full DataId, dataId.region will be set to region (if region is not None) and then used to update or insert into the Registry.

update : bool

If True, existing region information for these Dimensions is being replaced. This is usually required because Dimension entries are assumed to be pre-inserted prior to calling this function.

region : lsst.sphgeom.ConvexPolygon, optional

The region to update or insert into the Registry. If not present dataId.region must not be None.

kwds

Additional keyword arguments passed to the DataId constructor to convert dataId to a true DataId or augment an existing one.

Returns:
dataId : DataId

A Data ID with its region attribute set.

Raises:
NotImplementedError

Raised if limited is True.

transaction()

Context manager that implements SQL transactions.

Will roll back any changes to the SqlRegistry database in case an exception is raised in the enclosed block.

This context manager may be nested.