pytemplate.py 文件源码

def main(options):
    logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)

    global arguments
    arguments = ""
    if (options.arg):
        exec("%s = %s" % ("tmp",  options.arg))
        arguments = tmp

    logging.info("Load Environment")
    env = Environment(loader=FileSystemLoader(".", encoding='utf-8'), extensions=["jinja2.ext.do",])
    env.globals.update(load_csv = load_csv)
    env.globals.update(load_xml = load_xml)
    env.globals.update(load_sqlite = load_sqlite)
    env.globals.update(load_text = load_text)
    env.globals.update(le = le)
    env.globals.update(log = log)
    env.globals.update(file_md5 = file_md5)
    env.globals.update(file_stat = file_stat)
    env.globals.update(getarg = getarg)


    if (options.format == "odt"):
        logging.info("Read ODT template")
        engine = secretary.Renderer(environment=env)
        template = open(options.template, 'rb')
        logging.info("Template rendering...")
        output_data = engine.render(template)
        logging.info("Template rendering done")
        output_file = open(options.output, 'wb')
        output_file.write(output_data)
        output_file.close()
        logging.info("Done")

    elif (options.format == "text"):
        logging.info("Read TEXT template")
        template = env.get_template(options.template)
        logging.info("Template rendering...")
        output_data = template.render( )
        logging.info("Template rendering done")
        output_file = open(options.output, 'wb')
        output_file.write(output_data.encode('utf8'))
        output_file.close()
        logging.info("Done")
    else:
        logging.info("Error format")