proto.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Nov 24 15:35:04 2017

@author: Lorenz Steinert
"""

import re
import os
import sys
import shutil
import codecs
import argparse
from module.esme import FsrLdap, get_date, get_sprecher, get_config, get_share, get_path, get_ldap_server, get_ldap_base


def build_parser():
    """definition for the cli-arguments of the skript"""
    parser = argparse.ArgumentParser(description="Create Protokoll")
    parser.add_argument('-d', '--debug', '-v', '--verbose',
                        action='store_true', help='enable debug mode')
    parser.add_argument('-VV', '--Vollversamlung',
                        action='store_true', help='gennerate VV Protokoll')
    parser.add_argument('-t', '--test',
                        action='store_true', help='enable test mode')
    parser.add_argument('-p', '--path', help='set the protokoll directory')
    parser.add_argument('-c', '--config', help='set the config file')
    parser.add_argument('-s', '--sharedir', help='set the share directory')
    parser.add_argument('-S', '--Server', help='set the ldap Server')
    parser.add_argument('-b', '--base', help='set the ldap Server search base')

    return parser.parse_args()


def gen_head(share_dir, reg, sprecher, protokollant, date):
    """generate the head of the protokoll"""
    #look for the name.tex and if not found ignore it
    if os.path.isfile(os.path.join(share_dir, 'name.tex')):
        with codecs.open(os.path.join(share_dir, 'name.tex'),
                         'r', 'utf-8') as fobj_name:
            f_name = fobj_name.read()
    else:
        print(os.path.abspath(os.path.join(share_dir, 'name.tex'))
              + " not found. Ignoring")
        f_name = ""

    #look for the acroname.tex and the acro.tex if one is not found
    #ignore both
    if (os.path.isfile(os.path.join(share_dir, 'acroname.tex'))
            and os.path.isfile(os.path.join(share_dir, 'acro.tex'))):

        with codecs.open(os.path.join(share_dir, 'acroname.tex'),
                         'r', 'utf-8') as fobj_acroname:
            f_acroname = fobj_acroname.read()

        with codecs.open(os.path.join(share_dir, 'acro.tex'),
                         'r', 'utf-8') as fobj_acro:
            f_acro = fobj_acro.read()
    else:
        print(os.path.abspath(os.path.join(share_dir, 'acroname.tex'))
              + " or "
              + os.path.abspath(os.path.join(share_dir, 'acro.tex'))
              + " not found. Ignoring")
        f_acroname = ""
        f_acro = ""

    #look for teh head.tex and abort if it is not found
    if os.path.isfile(os.path.join(share_dir, 'head.tex')):
        with codecs.open(os.path.join(share_dir, 'head.tex'),
                         'r', 'utf-8') as fobj_head:
            f_head = fobj_head.read()
    else:
        print(os.path.abspath(os.path.join(share_dir, 'head.tex'))
              + " not found. Aborting!")
        sys.exit(1)

    #Replace the paceholder strings with the files
    f_head = reg['FSRN'].sub(f_name, f_head)
    f_head = reg['ACRON'].sub(f_acroname, f_head)
    f_head = reg['ACRO'].sub(f_acro, f_head)
    f_head = reg['SPR'].sub(sprecher, f_head)
    f_head = reg['PRO'].sub(protokollant, f_head)
    f_head = reg['DATE'].sub(date[0], f_head)

    return f_head

def resort_fsr(fsr, num_cols=5):
    """resort list for pascal style sorting"""
    fsr += [""] * (num_cols - len(fsr) % num_cols)
    depth = len(fsr) // num_cols

    tmp = [[] for i in range(depth)]

    for index, elem in zip(range(len(fsr)), fsr):
        tmp[index % depth] += [elem]

    return [j for i in tmp for j in i]

def gen_attendance_tab(fsr, num_cols=5, ext=False):
    """generate the atandance tables"""

    fsr = resort_fsr(fsr, num_cols)

    tabdef = '||' + 'c|l||' * num_cols

    fsr_tab = ''

    if ext:
        fsr_tab += "Externe Referenten\\\\\n\\\\\n" \
                   + r"\scriptsize{\begin{tabular}{" \
                   + tabdef + "}\n\\hline\n"
    else:
        fsr_tab += "FSR-Mitglieder\\\\\n\\\\\n\\scriptsize{\\begin{tabular}{" \
                   + tabdef + "}\n\\hline\n"

    for elem, index in zip(fsr, range(len(fsr))):
        if index % num_cols == 0:
            fsr_tab += " "
        else:
            fsr_tab += "&"

        fsr_tab += "  &" + elem

        if index % num_cols == num_cols - 1:
            fsr_tab += "\\\\\\hline\n"
        else:
            fsr_tab += "\n"

    if ext:
        fsr_tab += "\\end{tabular}}\\normalsize"
    else:
        fsr_tab += r"\end{tabular}}\normalsize\vspace{1em}"
        fsr_tab += "\n\\newline\n"

    return fsr_tab


def gen_foot(share_dir, vollversamlung, reg, sprecher, protokollant, date):
    """generate the foot of the protokoll"""
    #look for the foot*.tex skripts
    # if not found abort
    if vollversamlung:
        if os.path.isfile(os.path.join(share_dir, 'foot_VV.tex')):
            with codecs.open(os.path.join(share_dir, 'foot_VV.tex'),
                             'r', 'utf-8') as fobj_foot:
                f_foot = fobj_foot.read()
        else:
            print(os.path.abspath(os.path.join(share_dir, 'foot_VV.tex'))
                  + " not found. Aborting")
            sys.exit(1)
    else:
        if os.path.isfile(os.path.join(share_dir, 'foot.tex')):
            with codecs.open(os.path.join(share_dir, 'foot.tex'),
                             'r', 'utf-8') as fobj_foot:
                f_foot = fobj_foot.read()
        else:
            print(os.path.abspath(os.path.join(share_dir, 'foot.tex'))
                  + " not found. Aborting")
            sys.exit(1)

    #replace the placeholder strings in f_foot
    f_foot = reg['SPR'].sub(sprecher, f_foot)
    f_foot = reg['PRO'].sub(protokollant, f_foot)
    f_foot = reg['DATE'].sub(date[0], f_foot)

    return f_foot


if __name__ == "__main__":

    RUNDIR = os.path.dirname(os.path.realpath(__file__))
    OPT = build_parser()
    CONFIG = get_config(RUNDIR, OPT.config)
    SHARE_DIR = get_share(RUNDIR, CONFIG, OPT.sharedir)
    LDAP_SERVER = get_ldap_server(CONFIG, OPT.server)
    LDAP_SERVER_BASE = get_ldap_base(CONFIG, OPT.base)

    REG = {'SPR': re.compile('%%SPRECHER'),
           'PRO': re.compile('%%PROTOKOLLANT'),
           'DATE': re.compile('%%DATE'),
           'ACRO': re.compile('%%ACRO'),
           'ACRON': re.compile('%%ACRONAME'),
           'FSRN': re.compile('%%FSRNAME')}

    if CONFIG:
        NUM_COLS = CONFIG['DEFAULT'].getint('NUM_COLS')
        NUM_COLS_EXT = CONFIG['DEFAULT'].getint('NUM_COLS_EXT')
    else:
        NUM_COLS = 5
        NUM_COLS_EXT = 5

    #Set the timeout to 1 if we just want to test stuff
    if OPT.test:
        ESME = FsrLdap(LDAP_SERVER, LDAP_SERVER_BASE, OPT.debug,
                       timeout=1)
    else:
        ESME = FsrLdap(LDAP_SERVER, LDAP_SERVER_BASE, OPT.debug,
                       timeout=CONFIG['SERVER'].getint('connect_timeout'))

    print("\n")
    print("Wenn der Standardwert (in []) stimmt einfach Enter!\n")

    SPRECHER = get_sprecher(CONFIG)
    PROTOKOLLANT = ESME.get_protokollant_ldap(CONFIG)

    DATE = get_date(OPT.test)

    #Fetch the FSR from the Server
    #and if not able to connect fall back to the textfiles
    FSR = [ESME.get_fsr_ldap(SHARE_DIR),
           ESME.get_fsr_extern_ldap(SHARE_DIR)]

    #Generate the Path for the Protokoll in depandance of OPT.VV
    #and make the directory
    if OPT.Vollversamlung:
        OUTDIR = os.path.join(get_path(RUNDIR, CONFIG, OPT.path), DATE[1],
                              "fsr_physik_vollversamlung_" + '-'.join(DATE[1:]))
    else:
        OUTDIR = os.path.join(get_path(RUNDIR, CONFIG, OPT.path), DATE[1],
                              "fsr_physik_sitzung_" + '-'.join(DATE[1:]))
    if not os.path.exists(OUTDIR):
        os.makedirs(OUTDIR)

    #Generate the path of the .tex file
    OUTFILE = os.path.join(OUTDIR, "fsr_physik_protokoll_"
                           + '-'.join(DATE[1:]) + ".tex")

    shutil.copyfile(os.path.join(SHARE_DIR, "FS-Logo.pdf"),
                    os.path.join(OUTDIR, "FS-Logo.pdf"))

    msg = gen_head(SHARE_DIR, REG, SPRECHER, PROTOKOLLANT, DATE)

    msg += "~\\newline\n"

    if not OPT.Vollversamlung:
        msg += gen_attendance_tab(FSR[0], NUM_COLS)
        msg += gen_attendance_tab(FSR[1], NUM_COLS_EXT, ext=True)

    msg += gen_foot(SHARE_DIR, OPT.Vollversamlung, REG,
                    SPRECHER, PROTOKOLLANT, DATE)

    with codecs.open(OUTFILE, 'w', 'utf-8') as fobj_out:
        fobj_out.write(msg)