fixHTMLCode.py

#!/usr/bin/python

import os
import re
import sys

"""Replace all equation marks in HTML file with <img> tag to display
corresponding PNG file.  This assumes PNG files are in correct sequence.
Also fix any accented characters texinfo does not get right.
"""

# constants

rc = {
    "directory": "images",  # directory with equation images
}

eqmark = "<!-- EquationMark -->"

##############################################################################
# business


eqcnt = 0


def eqreplace(mx):
    """helper function to replace equation marks.

    mx -- regular expression match object

    Return replacement string.
    """
    global eqcnt
    eqcnt += 1
    imgfile = "eq-%02i.png" % eqcnt
    imgurl = os.path.join(rc["directory"], imgfile)
    s = '</p><p align="center"><img src="%s" alt="%s">' % (imgurl, imgurl)
    return s


def replaceEquationMarks(s):
    """Replace equation marks in given string.
    Return modified string.
    """
    s1 = re.sub(eqmark, eqreplace, s)
    return s1


def main():
    for f in sys.argv[1:]:
        with open(f) as fp:
            s = fp.read()
        s1 = replaceEquationMarks(s)
        if s1 != s:
            with open(f, "w") as fpout:
                fpout.write(s1)


if __name__ == "__main__":
    main()