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
  42         <refsect1>
  43                 <title>System manager directives</title>
  44
  45                 <para>Directives for configuring the behaviour of the
  46                 systemd process.</para>
  47
  48                 <variablelist id='systemd-directives' />
  49         </refsect1>
  50
  51         <refsect1>
  52                 <title>UDEV directives</title>
  53
  54                 <para>Directives for configuring systemd units through the
  55                 udev database.</para>
  56
  57                 <variablelist id='udev-directives' />
  58         </refsect1>
  59 </refentry>
  60 '''
  61
  62 def _extract_directives(directive_groups, page):
  63     t = tree.parse(page)
  64     section = t.find('./refmeta/manvolnum').text
  65     pagename = t.find('./refmeta/refentrytitle').text
  66     for variablelist in t.iterfind('.//variablelist'):
  67         klass = variablelist.attrib.get('class') or 'unit-directives'
  68         stor = directive_groups[klass]
  69         for varname in variablelist.iterfind('./varlistentry/term/varname'):
  70             text = ''.join(varname.text.partition('=')[:2])
  71             stor[text].append((pagename, section))
  72
  73 def _make_section(refentry, name, directives):
  74     varlist = refentry.find(".//*[@id='{}']".format(name))
  75     for varname, manpages in sorted(directives.items()):
  76         entry = tree.SubElement(varlist, 'varlistentry')
  77         a = tree.SubElement(tree.SubElement(entry, 'term'), 'varname')
  78         a.text = varname
  79         para = tree.SubElement(tree.SubElement(entry, 'listitem'), 'para')
  80
  81         b = None
  82         for manpage, manvolume in sorted(manpages):
  83                 if b is not None:
  84                         b.tail = ', '
  85                 b = tree.SubElement(para, 'citerefentry')
  86                 c = tree.SubElement(b, 'refentrytitle')
  87                 c.text = manpage
  88                 d = tree.SubElement(b, 'manvolnum')
  89                 d.text = manvolume
  90         entry.tail = '\n\n'
  91
  92 def _make_page(directive_groups):
  93     """Create an XML tree from directive_groups.
  94
  95     directive_groups = {
  96        'class': {'variable': [('manpage', 'manvolume'), ...],
  97                  'variable2': ...},
  98        ...
  99     }
 100     """
 101     refentry = tree.fromstring(TEMPLATE)
 102
 103     for name, directives in directive_groups.items():
 104             _make_section(refentry, name, directives)
 105
 106     return refentry
 107
 108 def make_page(xml_files):
 109     "Extract directives from xml_files and return XML index tree."
 110     directive_groups = {name:collections.defaultdict(list)
 111                         for name in ['unit-directives',
 112                                      'udev-directives',
 113                                      'systemd-directives',
 114                                      ]}
 115     for page in xml_files:
 116         _extract_directives(directive_groups, page)
 117
 118     return _make_page(directive_groups)
 119
 120 if __name__ == '__main__':
 121     tree.dump(make_page(sys.argv[1:]))