formations/setup_main_tex_file.py

#!/usr/bin/python
# -*- coding:utf-8 -*-

import argparse
import re
import random
import time
from os import path, makedirs
from jinja2 import Environment, FileSystemLoader

def get_unique_name(base):
    now = time.localtime()
    year = str(now[0])
    month = str(now[1]).rjust(2, '0')
    day = str(now[2]).rjust(2, '0')
    rand = str(randint(0, 100)).rjust(2, '0')
    return '-'.join([base, year, month, day, rand]).decode('utf-8')


def escape_tex(value):
    newval = value
    for pattern, replacement in LATEX_SUBS:
        newval = pattern.sub(replacement, newval)
    return newval


def main():

    def init(args):
        """
        init function
        """
        root = '../'
        if args.directory:
            root = root + re.sub(r'[\w]+/?', '../', args.directory)
        else:
            root = '../'
        
        name = args.name
        if name:
            if path.splitext(name)[1] == '':
                name = name.decode('utf-8') + u'.tex'
        else:
            name = get_unique_name('formation')
        
        title = args.title
        if not title:
            title = u'FIXME'
        else:
            title = title.decode('utf-8')
        
        author = args.author
        if not author:
            author = u'Cadoles'
        else:
            author = author.decode('utf-8')
        
        directory = args.directory
        if not directory:
            directory = ''

        env = {'root': root,
                'title': title,
                'author': author}
        resources = [(path.join('presentations', directory), 'main.tex'),]
                     #('slides', 'license-cc-by-sa-2.0.tex'),
                     #('slides', 'preambule.tex'),
                     #('slides', 'title.tex')]
        for directory, template_file in resources:
            template = jinja_env.get_template(template_file)
            rendered_template = template.render(**env)
            if not path.exists(directory):
                makedirs(directory)
            with open(path.join(directory, name), 'w') as rendered_file:
                rendered_file.write(rendered_template.encode('utf-8'))

    def update(args):
        """
        update function
        """
        re_class = re.compile(r'\\documentclass\{(?P<document_class>.*)\}')
        #skbconfig_re = re.compile(r'\\skbconfig\[\s*root\s*=\s*(?P<root>.*),\s*rep\s*=\s*(?P<rep>.*),\s*pub\s*=\s*(?P<pub>.*),\s*fig\s*=\s*(?P<fig>.*),\s*sli\s*=\s*(?P<sli>.*),\s*acr\s*=\s*(?P<acr>.*),\s*bib\s*=\s*(?P<bib>.*),\s*\]\{skblocal.tex\}', re.M)
        skbconfig_re = re.compile(r'\\skbconfig\[\n\s*root\s*=\s*(?P<root>.*),\n\s*rep\s*=\s*(?P<rep>.*),\n\s*pub\s*=\s*(?P<pub>.*),\n\s*fig\s*=\s*(?P<fig>.*),\n\s*sli\s*=\s*(?P<sli>.*),\n\s*acr\s*=\s*(?P<acr>.*),\n\s*bib\s*=\s*(?P<bib>.*)\n\s*\]\{skblocal.tex\}', re.M)
        skbinput_re = re.compile(r'\\skbinput\[from=(?P<rep>.*)\]\{(?P<tex>.*?)\}')
        with open(args.master, 'r') as master:
            tex_master = master.read()
            tex_class = re_class.search(tex_master)
            tex_skbconfig = skbconfig_re.search(tex_master)
            tex_skbinputs = skbinput_re.finditer(tex_master)

        for skbinput in tex_skbinputs:
            rep = path.dirname(skbinput.group('tex'))
            rep = path.join(tex_skbconfig.group(skbinput.group('rep')), rep)
            basename = '{0}.tex'.format(path.basename(skbinput.group('tex')))
            dest = path.join(rep, basename)
            if not path.isfile(dest):
                if not path.isdir(rep):
                    makedirs(rep)
                template = jinja_env.get_template('frame.tex')
                env = {'title': '', 'subtitle': '',
                       'name': dest}
                rendered_template = template.render(**env)
                with open(dest, 'w') as rendered_file:
                    rendered_file.write(rendered_template.encode('utf-8'))
                

    def outline(args):
        """
        outline creation
        """
        section_re = re.compile(r'\\section\{\(?P<name>.*?)}')
        part_re = re.compile(r'\\part\{\(?P<name>.*?)}')
        subsection_re = re.compile(r'\\subsection\{\(?P<name>.*?)}')
        with open(args.master, 'r') as master_tex:
            master = master_tex.read()
        parts = part_re.finditer(master)
        section = section_re.finditer(master)
        subsection_re = subsection_re.finditer(master)


    jinja_loader = FileSystemLoader('./templates')
    jinja_env = Environment(loader=jinja_loader,
                            block_start_string='((*',
                            block_end_string='*))',
                            variable_start_string='(((',
                            variable_end_string=')))',
                            comment_start_string='((=',
                            comment_end_string='=))',
                            trim_blocks=True)
    jinja_env.filters['escape_tex'] = escape_tex

    parser = argparse.ArgumentParser(description="Préparation des fichiers tex")
    subparsers = parser.add_subparsers(help='Aide des sous-commandes')
    parser_init = subparsers.add_parser('init', help='Initialisation du fichier maître')
    parser_init.add_argument('-n', '--name', help="Nom du fichier à créer", required=True)
    parser_init.add_argument('-a', '--author', help="Auteur de la formation")
    parser_init.add_argument('-t', '--title', help="Titre de la formation")
    parser_init.add_argument('-d', '--directory', help="Sous-répertoires où créer le fichier", required=True)
    parser_init.set_defaults(func=init)
    parser_update = subparsers.add_parser('update', help='Mise à jour des fichiers inclus')
    parser_update.add_argument('-m', '--master', help="Emplacement du fichier maître", required=True)
    parser_update.set_defaults(func=update)
    parser_outline = subparsers.add_parser('outline', help="Création du programme à partir du fichier maître")
    parser_outline.add_argument('-m', '--master', help="Emplacement du fichier maître", required=True)
    parser_outline.set_defaults(func=outline)
    args = parser.parse_args()
    args.func(args)
    
                            



if __name__ == '__main__':
    main()