mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.8 KiB
Python
64 lines
2.8 KiB
Python
# (c) 2019 Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
from __future__ import annotations
|
|
|
|
from ansible.errors import AnsibleError
|
|
from ansible.cli.galaxy import with_collection_artifacts_manager
|
|
from ansible.galaxy.collection import find_existing_collections
|
|
from ansible.module_utils.common.text.converters import to_bytes
|
|
from ansible.utils.collection_loader._collection_finder import _get_collection_name_from_path
|
|
from ansible.utils.display import Display
|
|
|
|
display = Display()
|
|
|
|
|
|
@with_collection_artifacts_manager
|
|
def list_collections(coll_filter=None, search_paths=None, dedupe=True, artifacts_manager=None):
|
|
|
|
collections = {}
|
|
for candidate in list_collection_dirs(search_paths=search_paths, coll_filter=coll_filter, artifacts_manager=artifacts_manager, dedupe=dedupe):
|
|
collection = _get_collection_name_from_path(candidate)
|
|
collections[collection] = candidate
|
|
return collections
|
|
|
|
|
|
@with_collection_artifacts_manager
|
|
def list_collection_dirs(search_paths=None, coll_filter=None, artifacts_manager=None, dedupe=True):
|
|
"""
|
|
Return paths for the specific collections found in passed or configured search paths
|
|
:param search_paths: list of text-string paths, if none load default config
|
|
:param coll_filter: limit collections to just the specific namespace or collection, if None all are returned
|
|
:return: list of collection directory paths
|
|
"""
|
|
|
|
namespace_filter = None
|
|
collection_filter = None
|
|
has_pure_namespace_filter = False # whether at least one coll_filter is a namespace-only filter
|
|
if coll_filter is not None:
|
|
if isinstance(coll_filter, str):
|
|
coll_filter = [coll_filter]
|
|
namespace_filter = set()
|
|
for coll_name in coll_filter:
|
|
if '.' in coll_name:
|
|
try:
|
|
namespace, collection = coll_name.split('.')
|
|
except ValueError:
|
|
raise AnsibleError("Invalid collection pattern supplied: %s" % coll_name)
|
|
namespace_filter.add(namespace)
|
|
if not has_pure_namespace_filter:
|
|
if collection_filter is None:
|
|
collection_filter = []
|
|
collection_filter.append(collection)
|
|
else:
|
|
namespace_filter.add(coll_name)
|
|
has_pure_namespace_filter = True
|
|
collection_filter = None
|
|
namespace_filter = sorted(namespace_filter)
|
|
|
|
for req in find_existing_collections(search_paths, artifacts_manager, namespace_filter=namespace_filter,
|
|
collection_filter=collection_filter, dedupe=dedupe):
|
|
|
|
if not has_pure_namespace_filter and coll_filter is not None and req.fqcn not in coll_filter:
|
|
continue
|
|
yield to_bytes(req.src)
|