docutils/docutils/writers/odf_odt/prepstyles.py

   1 #!/usr/bin/env python3
   2
   3 # $Id$
   4 # Author: Dave Kuhlman <dkuhlman@rexx.com>
   5 # Copyright: This module has been placed in the public domain.
   6
   7 """
   8 Adapt a word-processor-generated styles.odt for odtwriter use:
   9
  10 Drop page size specifications from styles.xml in STYLE_FILE.odt.
  11 See https://docutils.sourceforge.io/docs/user/odt.html#page-size
  12 """
  13
  14 # Author: Michael Schutte <michi@uiae.at>
  15
  16 from xml.etree import ElementTree as etree
  17
  18 import sys
  19 import zipfile
  20 from tempfile import mkstemp
  21 import shutil
  22 import os
  23
  24 NAMESPACES = {
  25     "style": "urn:oasis:names:tc:opendocument:xmlns:style:1.0",
  26     "fo": "urn:oasis:names:tc:opendocument:xmlns:xsl-fo-compatible:1.0"
  27 }
  28
  29
  30 def prepstyle(filename):
  31
  32     zin = zipfile.ZipFile(filename)
  33     styles = zin.open("styles.xml")
  34
  35     root = None
  36     # some extra effort to preserve namespace prefixes
  37     for event, elem in etree.iterparse(styles, events=("start", "start-ns")):
  38         if event == "start-ns":
  39             etree.register_namespace(elem[0], elem[1])
  40         elif event == "start":
  41             if root is None:
  42                 root = elem
  43
  44     styles.close()
  45
  46     for el in root.findall(".//style:page-layout-properties",
  47                            namespaces=NAMESPACES):
  48         for attr in list(el.attrib):
  49             if attr.startswith("{%s}" % NAMESPACES["fo"]):
  50                 del el.attrib[attr]
  51
  52     tempname = mkstemp()
  53     zout = zipfile.ZipFile(os.fdopen(tempname[0], "wb"), "w",
  54                            zipfile.ZIP_DEFLATED)
  55
  56     for item in zin.infolist():
  57         if item.filename == "styles.xml":
  58             zout.writestr(item, etree.tostring(root, encoding="UTF-8"))
  59         else:
  60             zout.writestr(item, zin.read(item.filename))
  61
  62     zout.close()
  63     zin.close()
  64     shutil.move(tempname[1], filename)
  65
  66
  67 def main():
  68     args = sys.argv[1:]
  69     if len(args) != 1 or args[0] in ('-h', '--help'):
  70         print(__doc__, file=sys.stderr)
  71         print("Usage: %s STYLE_FILE.odt\n" % sys.argv[0], file=sys.stderr)
  72         sys.exit(1)
  73     filename = args[0]
  74     prepstyle(filename)
  75
  76
  77 if __name__ == '__main__':
  78     main()