avoid fatal exception on invalid collection name (#72296)
* avoid fatal exception on invalid collection name used in ansible-doc filter
This commit is contained in:
parent
48c08f410c
commit
4f0e2fff95
3 changed files with 30 additions and 13 deletions
|
@ -0,0 +1,2 @@
|
||||||
|
bugfixes:
|
||||||
|
- Skip invalid collection names when listing in ansible-doc instead of throwing exception. Issue#72257
|
|
@ -8,6 +8,7 @@ import os
|
||||||
|
|
||||||
from collections import defaultdict
|
from collections import defaultdict
|
||||||
|
|
||||||
|
from ansible.errors import AnsibleError
|
||||||
from ansible.collections import is_collection_path
|
from ansible.collections import is_collection_path
|
||||||
from ansible.module_utils._text import to_bytes
|
from ansible.module_utils._text import to_bytes
|
||||||
from ansible.utils.collection_loader import AnsibleCollectionConfig
|
from ansible.utils.collection_loader import AnsibleCollectionConfig
|
||||||
|
@ -54,6 +55,17 @@ def list_collection_dirs(search_paths=None, coll_filter=None):
|
||||||
:return: list of collection directory paths
|
:return: list of collection directory paths
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
collection = None
|
||||||
|
namespace = None
|
||||||
|
if coll_filter is not None:
|
||||||
|
if '.' in coll_filter:
|
||||||
|
try:
|
||||||
|
(namespace, collection) = coll_filter.split('.')
|
||||||
|
except ValueError:
|
||||||
|
raise AnsibleError("Invalid collection pattern supplied: %s" % coll_filter)
|
||||||
|
else:
|
||||||
|
namespace = coll_filter
|
||||||
|
|
||||||
collections = defaultdict(dict)
|
collections = defaultdict(dict)
|
||||||
for path in list_valid_collection_paths(search_paths):
|
for path in list_valid_collection_paths(search_paths):
|
||||||
|
|
||||||
|
@ -62,32 +74,28 @@ def list_collection_dirs(search_paths=None, coll_filter=None):
|
||||||
b_coll_root = to_bytes(os.path.join(path, 'ansible_collections'))
|
b_coll_root = to_bytes(os.path.join(path, 'ansible_collections'))
|
||||||
|
|
||||||
if os.path.exists(b_coll_root) and os.path.isdir(b_coll_root):
|
if os.path.exists(b_coll_root) and os.path.isdir(b_coll_root):
|
||||||
coll = None
|
|
||||||
if coll_filter is None:
|
if namespace is None:
|
||||||
namespaces = os.listdir(b_coll_root)
|
namespaces = os.listdir(b_coll_root)
|
||||||
else:
|
else:
|
||||||
if '.' in coll_filter:
|
namespaces = [namespace]
|
||||||
(nsp, coll) = coll_filter.split('.')
|
|
||||||
else:
|
|
||||||
nsp = coll_filter
|
|
||||||
namespaces = [nsp]
|
|
||||||
|
|
||||||
for ns in namespaces:
|
for ns in namespaces:
|
||||||
b_namespace_dir = os.path.join(b_coll_root, to_bytes(ns))
|
b_namespace_dir = os.path.join(b_coll_root, to_bytes(ns))
|
||||||
|
|
||||||
if os.path.isdir(b_namespace_dir):
|
if os.path.isdir(b_namespace_dir):
|
||||||
|
|
||||||
if coll is None:
|
if collection is None:
|
||||||
colls = os.listdir(b_namespace_dir)
|
colls = os.listdir(b_namespace_dir)
|
||||||
else:
|
else:
|
||||||
colls = [coll]
|
colls = [collection]
|
||||||
|
|
||||||
for collection in colls:
|
for mycoll in colls:
|
||||||
|
|
||||||
# skip dupe collections as they will be masked in execution
|
# skip dupe collections as they will be masked in execution
|
||||||
if collection not in collections[ns]:
|
if mycoll not in collections[ns]:
|
||||||
b_coll = to_bytes(collection)
|
b_coll = to_bytes(mycoll)
|
||||||
b_coll_dir = os.path.join(b_namespace_dir, b_coll)
|
b_coll_dir = os.path.join(b_namespace_dir, b_coll)
|
||||||
if is_collection_path(b_coll_dir):
|
if is_collection_path(b_coll_dir):
|
||||||
collections[ns][collection] = b_coll_dir
|
collections[ns][mycoll] = b_coll_dir
|
||||||
yield b_coll_dir
|
yield b_coll_dir
|
||||||
|
|
|
@ -18,6 +18,13 @@ current_out="$(ansible-doc --playbook-dir ./ testns.testcol.fakemodule)"
|
||||||
expected_out="$(cat fakemodule.output)"
|
expected_out="$(cat fakemodule.output)"
|
||||||
test "$current_out" == "$expected_out"
|
test "$current_out" == "$expected_out"
|
||||||
|
|
||||||
|
# ensure we do work with valid collection name for list
|
||||||
|
ansible-doc --list testns.testcol --playbook-dir ./ 2>&1 | grep -v "Invalid collection pattern"
|
||||||
|
|
||||||
|
# ensure we dont break on invalid collection name for list
|
||||||
|
ansible-doc --list testns.testcol.fakemodule --playbook-dir ./ 2>&1 | grep "Invalid collection pattern"
|
||||||
|
|
||||||
|
|
||||||
# test listing diff plugin types from collection
|
# test listing diff plugin types from collection
|
||||||
for ptype in cache inventory lookup vars
|
for ptype in cache inventory lookup vars
|
||||||
do
|
do
|
||||||
|
|
Loading…
Reference in a new issue