mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
optimized module docs
This commit is contained in:
parent
65c649aa3e
commit
164092a835
1 changed files with 50 additions and 44 deletions
|
@ -52,57 +52,63 @@ def get_docstring(filename, verbose=False):
|
||||||
M = ast.parse(''.join(open(filename)))
|
M = ast.parse(''.join(open(filename)))
|
||||||
for child in M.body:
|
for child in M.body:
|
||||||
if isinstance(child, ast.Assign):
|
if isinstance(child, ast.Assign):
|
||||||
if 'DOCUMENTATION' in (t.id for t in child.targets):
|
for t in child.targets:
|
||||||
doc = yaml.safe_load(child.value.s)
|
try:
|
||||||
fragments = doc.get('extends_documentation_fragment', [])
|
theid = t.id
|
||||||
|
except AttributeError as e:
|
||||||
|
continue #TODO: should log these to figure out why this happens
|
||||||
|
|
||||||
if isinstance(fragments, basestring):
|
if 'DOCUMENTATION' in theid:
|
||||||
fragments = [ fragments ]
|
doc = yaml.safe_load(child.value.s)
|
||||||
|
fragments = doc.get('extends_documentation_fragment', [])
|
||||||
|
|
||||||
# Allow the module to specify a var other than DOCUMENTATION
|
if isinstance(fragments, basestring):
|
||||||
# to pull the fragment from, using dot notation as a separator
|
fragments = [ fragments ]
|
||||||
for fragment_slug in fragments:
|
|
||||||
fragment_slug = fragment_slug.lower()
|
|
||||||
if '.' in fragment_slug:
|
|
||||||
fragment_name, fragment_var = fragment_slug.split('.', 1)
|
|
||||||
fragment_var = fragment_var.upper()
|
|
||||||
else:
|
|
||||||
fragment_name, fragment_var = fragment_slug, 'DOCUMENTATION'
|
|
||||||
|
|
||||||
fragment_class = fragment_loader.get(fragment_name)
|
# Allow the module to specify a var other than DOCUMENTATION
|
||||||
assert fragment_class is not None
|
# to pull the fragment from, using dot notation as a separator
|
||||||
|
for fragment_slug in fragments:
|
||||||
fragment_yaml = getattr(fragment_class, fragment_var, '{}')
|
fragment_slug = fragment_slug.lower()
|
||||||
fragment = yaml.safe_load(fragment_yaml)
|
if '.' in fragment_slug:
|
||||||
|
fragment_name, fragment_var = fragment_slug.split('.', 1)
|
||||||
if fragment.has_key('notes'):
|
fragment_var = fragment_var.upper()
|
||||||
notes = fragment.pop('notes')
|
|
||||||
if notes:
|
|
||||||
if not doc.has_key('notes'):
|
|
||||||
doc['notes'] = []
|
|
||||||
doc['notes'].extend(notes)
|
|
||||||
|
|
||||||
if 'options' not in fragment.keys():
|
|
||||||
raise Exception("missing options in fragment, possibly misformatted?")
|
|
||||||
|
|
||||||
for key, value in fragment.items():
|
|
||||||
if not doc.has_key(key):
|
|
||||||
doc[key] = value
|
|
||||||
else:
|
else:
|
||||||
if isinstance(doc[key], MutableMapping):
|
fragment_name, fragment_var = fragment_slug, 'DOCUMENTATION'
|
||||||
doc[key].update(value)
|
|
||||||
elif isinstance(doc[key], MutableSet):
|
fragment_class = fragment_loader.get(fragment_name)
|
||||||
doc[key].add(value)
|
assert fragment_class is not None
|
||||||
elif isinstance(doc[key], MutableSequence):
|
|
||||||
doc[key] = sorted(frozenset(doc[key] + value))
|
fragment_yaml = getattr(fragment_class, fragment_var, '{}')
|
||||||
|
fragment = yaml.safe_load(fragment_yaml)
|
||||||
|
|
||||||
|
if fragment.has_key('notes'):
|
||||||
|
notes = fragment.pop('notes')
|
||||||
|
if notes:
|
||||||
|
if not doc.has_key('notes'):
|
||||||
|
doc['notes'] = []
|
||||||
|
doc['notes'].extend(notes)
|
||||||
|
|
||||||
|
if 'options' not in fragment.keys():
|
||||||
|
raise Exception("missing options in fragment, possibly misformatted?")
|
||||||
|
|
||||||
|
for key, value in fragment.items():
|
||||||
|
if not doc.has_key(key):
|
||||||
|
doc[key] = value
|
||||||
else:
|
else:
|
||||||
raise Exception("Attempt to extend a documentation fragement of unknown type")
|
if isinstance(doc[key], MutableMapping):
|
||||||
|
doc[key].update(value)
|
||||||
|
elif isinstance(doc[key], MutableSet):
|
||||||
|
doc[key].add(value)
|
||||||
|
elif isinstance(doc[key], MutableSequence):
|
||||||
|
doc[key] = sorted(frozenset(doc[key] + value))
|
||||||
|
else:
|
||||||
|
raise Exception("Attempt to extend a documentation fragement of unknown type")
|
||||||
|
|
||||||
if 'EXAMPLES' in (t.id for t in child.targets):
|
elif 'EXAMPLES' in theid:
|
||||||
plainexamples = child.value.s[1:] # Skip first empty line
|
plainexamples = child.value.s[1:] # Skip first empty line
|
||||||
|
|
||||||
if 'RETURN' in (t.id for t in child.targets):
|
elif 'RETURN' in theid:
|
||||||
returndocs = child.value.s[1:]
|
returndocs = child.value.s[1:]
|
||||||
except:
|
except:
|
||||||
traceback.print_exc() # temp
|
traceback.print_exc() # temp
|
||||||
if verbose == True:
|
if verbose == True:
|
||||||
|
|
Loading…
Reference in a new issue