#!/usr/bin/env python
# -*- coding: utf-8 mode: python -*- vim:sw=4:sts=4:et:ai:si:sta:fenc=utf-8

u"""Ce script est prévu pour être utilisé dans un script CGI.
Il enregistre les fichiers uploadés dans le répertoire spécifié, et éventuellement lance une commande avec comme argument les fichiers enregistrés.
"""

import sys, os, types, cgi, tempfile
from os import path

cgitb = None # marqueur pour savoir si le module cgitb a été importé
#import cgitb; cgitb.enable()

class FileManager(object):
    def __init__(self, destdir=None, parse_env=True):
        if destdir is None: destdir = '/tmp'
        self.fileinfos = []
        self.count = 0
        self.destdir = destdir
        if parse_env: self.parse_env()

    def parse_env(self):
        class FieldStorage(cgi.FieldStorage):
            def make_file(self, binary=None):
                return self.fm._new_file(self)
        FieldStorage.fm = self
        self.form = FieldStorage()

    def _new_file(self, form):
        fp = path.join(self.destdir, 'upload%i' % self.count)
        outf = open(fp, 'w+b')
        self.count += 1
        self.fileinfos.append((form.name, form.filename, fp, form.type))
        return outf

    def write_csv(self, outf):
        close = False
        if type(outf) is types.StringType:
            outf = open(outf, 'w+b')
            close = True
        try:
            outf.write("name,value,file,type\n")
            for fi in self.fileinfos:
                outf.write("%s,%s,%s,%s\n" % fi)
        finally:
            if close: outf.close()

# Analyser les options
from optparse import OptionParser
parser = OptionParser(usage=u"\n\t%prog [options]\n\t%prog [options] -c command [args]", description=__doc__)
parser.add_option('-d', '--destdir', dest='destdir',
                  help=u"Spécifier le répertoire de destination des fichiers uploadés. Ce répertoire est créé si nécessaire.")
parser.add_option('-p', '--printcsv', action='store_true', dest='printcsv',
                  help=u"Afficher la liste des fichiers uploadés au format CSV. Chaque ligne est de la forme 'name,value,file,type' où name est le nom du champ dans le formulaire http, value le nom du fichier tel qu'il a été fourni par le client, file le chemin complet vers le fichier uploadé, et type le type mime du contenu.")
parser.add_option('-c', '--cmd', action='store_true', dest='cmd',
                  help=u"Lancer une commande après avoir enregistré les fichiers. Implique --remove")
parser.add_option('-G', '--infos', action='store_const', const='infos', dest='argtype',
                  help=u"La commande est lancée avec 4 arguments pour chaque fichier uploadé. Les arguments sont fournis dans l'ordre donné par l'option -p. C'est l'option par défaut.")
parser.add_option('-F', '--files', action='store_const', const='files', dest='argtype',
                  help=u"La commande est lancée avec en argument chaque fichier uploadé, correspondant à la colonne file de l'option -p")
parser.add_option('-C', '--csv', action='store_const', const='csv', dest='argtype',
                  help=u"La commande est lancée avec en argument un fichier au format CSV qui liste les fichiers uploadés. Cf l'option -p pour le format de ce fichier.")
parser.set_defaults(argtype='infos')
parser.add_option('--remove', action='store_true', dest='clear',
                  help=u"Supprimer les fichiers uploadés.")
parser.add_option('-k', '--keep', action='store_false', dest='clear',
                  help=u"Ne pas supprimer les fichiers uploadés. Utiliser cette option avec -c pour garder les fichiers uploadés.")
parser.add_option('--devel', action='store_true', dest='devel',
                  help=u"Activer le mode développement: le module cgitb est chargé et activé.")
o, args = parser.parse_args()

if o.devel and cgitb is None:
    import cgitb; cgitb.enable()

# Créer destdir le cas échéant
if o.destdir is not None and not path.isdir(o.destdir):
    os.makedirs(o.destdir)

# Si aucune commande n'est spécifiée, afficher la liste des fichiers au format CSV
if o.printcsv is None and o.cmd is None:
    o.printcsv = True

# Avec -c, par défaut supprimer les fichiers
if o.cmd and o.clear is None:
    o.clear = True

fm = FileManager(o.destdir)

if o.printcsv:
    fm.write_csv(sys.stdout)

if o.cmd:
    tmpfile = None
    if o.argtype == 'infos':
        for fi in fm.fileinfos:
            args.extend(fi)
    elif o.argtype == 'files':
        args.extend(map(lambda fi: fi[2], fm.fileinfos))
    elif o.argtype == 'csv':
        fd, tmpfile = tempfile.mkstemp()
        os.close(fd)
        fm.write_csv(tmpfile)
        args.append(tmpfile)

    os.spawnvp(os.P_WAIT, args[0], args)

    if tmpfile is not None:
        try: os.remove(tmpfile)
        except: pass

if o.clear:
    for fi in fm.fileinfos:
        try: os.remove(fi[0])
        except: pass