make-directive-index.py

   1 # -*- coding: utf-8 -*-
   2 import sys
   3 import collections
   4 import xml.etree.ElementTree as tree
   5
   6 TEMPLATE = '''\
   7 <refentry id="systemd.directives">
   8
   9         <refentryinfo>
  10                 <title>systemd.directives</title>
  11                 <productname>systemd</productname>
  12
  13                 <authorgroup>
  14                         <author>
  15                                 <contrib>Developer</contrib>
  16                                 <firstname>Zbigniew</firstname>
  17                                 <surname>Jędrzejewski-Szmek</surname>
  18                                 <email>zbyszek@in.waw.pl</email>
  19                         </author>
  20                 </authorgroup>
  21         </refentryinfo>
  22
  23         <refmeta>
  24                 <refentrytitle>systemd.directives</refentrytitle>
  25                 <manvolnum>5</manvolnum>
  26         </refmeta>
  27
  28         <refnamediv>
  29                 <refname>systemd.directives</refname>
  30                 <refpurpose>Index of configuration directives</refpurpose>
  31         </refnamediv>
  32
  33         <refsect1>
  34                 <title>Unit directives</title>
  35
  36                 <para>Directives for configuring units, used in unit
  37                 files.</para>
  38
  39                 <variablelist id='unit-directives' />
  40         </refsect1>
  41 </refentry>
  42 '''
  43
  44 def _extract_directives(directive_groups, page):
  45     t = tree.parse(page)
  46     section = t.find('./refmeta/manvolnum').text
  47     pagename = t.find('./refmeta/refentrytitle').text
  48     for variablelist in t.iterfind('.//variablelist'):
  49         klass = variablelist.attrib.get('class') or 'unit-directives'
  50         stor = directive_groups[klass]
  51         for varname in variablelist.iterfind('./varlistentry/term/varname'):
  52             text = ''.join(varname.text.partition('=')[:2])
  53             stor[text].append((pagename, section))
  54
  55 def _make_section(refentry, name, directives):
  56     varlist = refentry.find(".//*[@id='{}']".format(name))
  57     for varname, manpages in sorted(directives.items()):
  58         entry = tree.SubElement(varlist, 'varlistentry')
  59         a = tree.SubElement(tree.SubElement(entry, 'term'), 'varname')
  60         a.text = varname
  61         para = tree.SubElement(tree.SubElement(entry, 'listitem'), 'para')
  62
  63         b = None
  64         for manpage, manvolume in sorted(manpages):
  65                 if b is not None:
  66                         b.tail = ', '
  67                 b = tree.SubElement(para, 'citerefentry')
  68                 c = tree.SubElement(b, 'refentrytitle')
  69                 c.text = manpage
  70                 d = tree.SubElement(b, 'manvolnum')
  71                 d.text = manvolume
  72         entry.tail = '\n\n'
  73
  74 def _make_page(directive_groups):
  75     """Create an XML tree from directive_groups.
  76
  77     directive_groups = {
  78        'class': {'variable': [('manpage', 'manvolume'), ...],
  79                  'variable2': ...},
  80        ...
  81     }
  82     """
  83     refentry = tree.fromstring(TEMPLATE)
  84
  85     for name, directives in directive_groups.items():
  86             _make_section(refentry, name, directives)
  87
  88     return refentry
  89
  90 def make_page(xml_files):
  91     "Extract directives from xml_files and return XML index tree."
  92     directive_groups = {name:collections.defaultdict(list)
  93                         for name in ['unit-directives',
  94                                      ]}
  95     for page in xml_files:
  96         _extract_directives(directive_groups, page)
  97
  98     return _make_page(directive_groups)
  99
 100 if __name__ == '__main__':
 101     tree.dump(make_page(sys.argv[1:]))