class LibraryDoc:
"""Documentation for a library, a resource file or a suite file."""
def __init__(self, name='', doc='', version='', type='LIBRARY', scope='TEST',
doc_format='ROBOT', source=None, lineno=-1):
self.name = name
self._doc = doc
self.version = version
self.type = type
self.scope = scope
self.doc_format = doc_format
self.source = source
self.lineno = lineno
self.inits = ()
self.keywords = ()
self.type_docs = ()
@property
def doc(self):
if self.doc_format == 'ROBOT' and '%TOC%' in self._doc:
return self._add_toc(self._doc)
return self._doc
def _add_toc(self, doc):
toc = self._create_toc(doc)
return '\n'.join(line if line.strip() != '%TOC%' else toc
for line in doc.splitlines())
def _create_toc(self, doc):
entries = re.findall(r'^\s*=\s+(.+?)\s+=\s*$', doc, flags=re.MULTILINE)
if self.inits:
entries.append('Importing')
if self.keywords:
entries.append('Keywords')
return '\n'.join('- `%s`' % entry for entry in entries)
@setter
def doc_format(self, format):
return format or 'ROBOT'
@setter
def inits(self, inits):
"""Initializer docs as :class:`~KeywordDoc` instances."""
return self._process_keywords(inits)
@setter
def keywords(self, kws):
"""Keyword docs as :class:`~KeywordDoc` instances."""
return self._process_keywords(kws)
@setter
def type_docs(self, type_docs):
return set(type_docs)
def _process_keywords(self, kws):
for keyword in kws:
keyword.parent = self
return sorted(kws)
@property
def all_tags(self):
return Tags(chain.from_iterable(kw.tags for kw in self.keywords))
def save(self, output=None, format='HTML', theme=None):
with LibdocOutput(output, format) as outfile:
LibdocWriter(format, theme).write(self, outfile)
def convert_docs_to_html(self):
formatter = DocFormatter(self.keywords, self.type_docs, self.doc, self.doc_format)
self._doc = formatter.html(self.doc, intro=True)
for item in self.inits + self.keywords:
# If 'short_doc' is not set, it is generated automatically based on 'doc'
# when accessed. Generate and set it to avoid HTML format affecting it.
item.short_doc = item.short_doc
item.doc = formatter.html(item.doc)
for type_doc in self.type_docs:
# Standard docs are always in ROBOT format ...
if type_doc.type == type_doc.STANDARD:
# ... unless they have been converted to HTML already.
if not type_doc.doc.startswith('<p>'):
type_doc.doc = DocToHtml('ROBOT')(type_doc.doc)
else:
type_doc.doc = formatter.html(type_doc.doc)
self.doc_format = 'HTML'
def to_dictionary(self, include_private=False, theme=None):
data = {
'specversion': 3,
'name': self.name,
'doc': self.doc,
'version': self.version,
'generated': get_generation_time(),
'type': self.type,
'scope': self.scope,
'docFormat': self.doc_format,
'source': str(self.source) if self.source else None,
'lineno': self.lineno,
'tags': list(self.all_tags),
'inits': [init.to_dictionary() for init in self.inits],
'keywords': [kw.to_dictionary() for kw in self.keywords
if include_private or not kw.private],
'typedocs': [t.to_dictionary() for t in sorted(self.type_docs)]
}
if theme:
data['theme'] = theme.lower()
return data
def to_json(self, indent=None, include_private=True, theme=None):
data = self.to_dictionary(include_private, theme)
return json.dumps(data, indent=indent)