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