Coverage for python/lsst/daf/butler/direct_butler/_direct_butler_collections.py: 71%
17 statements
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-11 03:16 -0700
« prev ^ index » next coverage.py v7.5.1, created at 2024-05-11 03:16 -0700
1# This file is part of daf_butler.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (http://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This software is dual licensed under the GNU General Public License and also
10# under a 3-clause BSD license. Recipients may choose which of these licenses
11# to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
12# respectively. If you choose the GPL option then the following text applies
13# (but note that there is still no warranty even if you opt for BSD instead):
14#
15# This program is free software: you can redistribute it and/or modify
16# it under the terms of the GNU General Public License as published by
17# the Free Software Foundation, either version 3 of the License, or
18# (at your option) any later version.
19#
20# This program is distributed in the hope that it will be useful,
21# but WITHOUT ANY WARRANTY; without even the implied warranty of
22# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23# GNU General Public License for more details.
24#
25# You should have received a copy of the GNU General Public License
26# along with this program. If not, see <http://www.gnu.org/licenses/>.
28from __future__ import annotations
30__all__ = ("DirectButlerCollections",)
32from collections.abc import Iterable
34from lsst.utils.iteration import ensure_iterable
36from .._butler_collections import ButlerCollections
37from ..registry.sql_registry import SqlRegistry
40class DirectButlerCollections(ButlerCollections):
41 """Implementation of ButlerCollections for DirectButler.
43 Parameters
44 ----------
45 registry : `~lsst.daf.butler.registry.sql_registry.SqlRegistry`
46 Registry object used to work with the collections database.
47 """
49 def __init__(self, registry: SqlRegistry):
50 self._registry = registry
52 def extend_chain(self, parent_collection_name: str, child_collection_names: str | Iterable[str]) -> None:
53 return self._registry._managers.collections.extend_collection_chain(
54 parent_collection_name, list(ensure_iterable(child_collection_names))
55 )
57 def prepend_chain(self, parent_collection_name: str, child_collection_names: str | Iterable[str]) -> None:
58 return self._registry._managers.collections.prepend_collection_chain(
59 parent_collection_name, list(ensure_iterable(child_collection_names))
60 )
62 def redefine_chain(
63 self, parent_collection_name: str, child_collection_names: str | Iterable[str]
64 ) -> None:
65 self._registry._managers.collections.update_chain(
66 parent_collection_name, list(ensure_iterable(child_collection_names))
67 )
69 def remove_from_chain(
70 self, parent_collection_name: str, child_collection_names: str | Iterable[str]
71 ) -> None:
72 return self._registry._managers.collections.remove_from_collection_chain(
73 parent_collection_name, list(ensure_iterable(child_collection_names))
74 )