source: OpenRLabs-Git/deploy/rlabs-docker/web2py-rlabs/gluon/contrib/markmin/markmin2pdf.py

"""
Created by Massimo Di Pierro
License BSD
"""
from __future__ import print_function

import subprocess
import os
import os.path
import re
import sys
from tempfile import mkstemp, mkdtemp, NamedTemporaryFile
from gluon.contrib.markmin.markmin2latex import markmin2latex

__all__ = ['markmin2pdf']


def removeall(path):
    ERROR_STR = """Error removing %(path)s, %(error)s """

    def rmgeneric(path, __func__):
        try:
            __func__(path)
        except OSError as xxx_todo_changeme:
            (errno, strerror) = xxx_todo_changeme.args
            print(ERROR_STR % {'path': path, 'error': strerror})

    files = [path]

    while files:
        file = files[0]
        if os.path.isfile(file):
            f = os.remove
            rmgeneric(file, os.remove)
            del files[0]
        elif os.path.isdir(file):
            nested = os.listdir(file)
            if not nested:
                rmgeneric(file, os.rmdir)
                del files[0]
            else:
                files = [os.path.join(file, x) for x in nested] + files


def latex2pdf(latex, pdflatex='pdflatex', passes=3):
    """
    calls pdflatex in a tempfolder

    Arguments:

    - pdflatex: path to the pdflatex command. Default is just 'pdflatex'.
    - passes:   defines how often pdflates should be run in the texfile.
    """

    pdflatex = pdflatex
    passes = passes
    warnings = []

    # setup the envoriment
    tmpdir = mkdtemp()
    texfile = open(tmpdir + '/test.tex', 'wt',encoding="utf-8")
    texfile.write(latex)
    texfile.seek(0)
    texfile.close()
    texfile = os.path.abspath(texfile.name)

    # start doing some work
    for i in range(0, passes):
        logfd, logname = mkstemp()
        outfile = os.fdopen(logfd)
        try:
            ret = subprocess.call([pdflatex,
                                   '-interaction=nonstopmode',
                                   '-output-format', 'pdf',
                                   '-output-directory', tmpdir,
                                   texfile],
                                  cwd=os.path.dirname(texfile), stdout=outfile,
                                  stderr=subprocess.PIPE)
        finally:
            outfile.close()
        re_errors = re.compile('^\!(.*)$', re.M)
        re_warnings = re.compile('^LaTeX Warning\:(.*)$', re.M)
        flog = open(logname,encoding="utf-8")
        try:
            loglines = flog.read()
        finally:
            flog.close()
        errors = re_errors.findall(loglines)
        warnings = re_warnings.findall(loglines)
        os.unlink(logname)

    pdffile = texfile.rsplit('.', 1)[0] + '.pdf'
    data = None

    if os.path.isfile(pdffile):
        with open(pdffile,'rb') as fpdf :
            data = fpdf.read()

    removeall(tmpdir)

    return data, warnings, errors


def markmin2pdf(text, image_mapper=lambda x: None, extra={}):
    return latex2pdf(markmin2latex(text, image_mapper=image_mapper, extra=extra))


if __name__ == '__main__':
    import sys
    import doctest
    import markmin2html

    if sys.argv[1:2] == ['-h']:
        data, warnings, errors = markmin2pdf(markmin2html.__doc__)
        if errors:
            print('ERRORS:' + '\n'.join(errors))
            print('WARNGINS:' + '\n'.join(warnings))
        else:
            print(data)
    elif len(sys.argv) > 1:
        fargv = open(sys.argv[1], 'rb')
        try:
            data, warnings, errors = markmin2pdf(fargv.read())
        finally:
            fargv.close()
        if errors:
            print('ERRORS:' + '\n'.join(errors))
            print('WARNGINS:' + '\n'.join(warnings))
        else:
            print(data)
    else:
        doctest.testmod()