nutools/lib/ulib/support/cgiupload.py

#!/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.

Retourner 0 si tous les fichiers ont été traités, 1 si au moins un fichier a été rejeté, 2 si aucun fichier ne correspondait (ou si aucun fichier n'a été fourni)
"""

import sys, os, re, 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):
    fileinfos = None
    tries = None
    count = None
    destdir = None
    spec = None
    type = None
    exitcode = None

    def __init__(self, destdir=None, spec=None, type=None, parse_env=True):
        if destdir is None: destdir = '/tmp'
        self.fileinfos = []
        self.tries = 0
        self.count = 0
        self.destdir = destdir
        if spec is not None: spec = re.compile(spec)
        self.spec = spec
        self.type = type
        if parse_env: self.parse_env()

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

    def _new_file(self, form):
        self.tries += 1
        if self.spec is not None and self.spec.match(form.filename) is None: return None
        if self.type is not None and self.type != form.type: return None
        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()

    def exitcode(self):
        if self.count == 0: return 2
        elif self.count != self.tries: return 1
        else: return 0

# 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('-e', '--spec', dest='spec',
                  help=u"N'accepter que des fichiers dont le nom correspond à l'expression régulière spécifiée")
parser.add_option('-t', '--type', dest='type',
                  help=u"N'accepter que des fichiers dont le type mime correspond à la valeur spécifiée")
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, o.spec, o.type)

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

sys.exit(fm.exitcode())