wcs-dynexport : intégration du basculement vers PostgreSQL (suite)

[progfou.git] / coda / coda-filter.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Objectifs :
# 05. si le destinataire est noreply@auf.org, on le remplace par l'émetteur
#     (ANNULÉ car ça ne devrait plus arriver)
#     (REPRIS pour le cas où le destinataire est impression-coda@auf.org)
# 10. envoyer une copie à l'émetteur, sauf s'il est déjà le destinataire
#     (ANNULÉ car envoi déjà vers les bons destinataires)
# 20. ajouter une en-tête X-Coda pour faciliter les filtrages 
# 30. préfixer le sujet par '[CODA] '
# 40. remplacer l'organisation par AuF (forme longue, au lieu de CODA)
# 50. renommer l'attachement en utilisant le sujet
#       Content-Type: application/pdf; name=attachment.pdf
#       Content-Disposition: inline; filename=attachment.pdf
#
import sys
import email
import email.charset
from email.header import decode_header
import smtplib
import re

# préfixe pour le sujet
SUBJECT_PREFIX = "[CODA] "
# contenu pour l'en-tête X-Coda qui sera ajoutée
X_CODA_VALUE = "Filtre AuF pour Coda v2"
# contenu pour l'en-tête Organisation
ORGANIZATION = "Agence universitaire de la Francophonie"
# masque de repérage du type et numéro de document
DOC_PATTERN = r'^(.*:)?\s*(\S+)\s*/\s*(\d+)\s*$'
# format du nouveau nom de document (à partir du type et numéro)
DOC_FORMAT = 'CODA-%s-%08d.pdf'
# réécriture des liens pour coda.auf
BODY_PATTERN_1 = r'\n(http://coda.auf/)(.*)\n'
BODY_FORMAT_1 = r'\n\nAccès depuis une implantation AUF :\n \1\2\n\nAccès depuis Internet :\n https://coda.auf.org/\2\n\n'
# réécriture des liens pour form.coda.auf
BODY_PATTERN_2 = r'\n(http://form.coda.auf/)(.*)\n'
BODY_FORMAT_2 = r'\n\nAccès depuis une implantation AUF :\n \1\2\n\nAccès depuis Internet :\n https://form-coda.auf.org/\2\n\n'

# codes de sortie venant de <sysexits.h>
EX_TEMPFAIL = 75
EX_UNAVAILABLE = 69

# on récupère l'expéditeur et le(s) destinataire(s) en paramètres
#email_from = 'jca.test@auf.org'
email_from = sys.argv[1]
email_to_list = sys.argv[2:]

# on tente de récupérer le message arrivant sur l'entrée standard (pipe)
try:
    msg = email.message_from_file(sys.stdin)
# en cas d'échec on demande à Postfix de considérer que c'est temporaire
except Exception, err:
    print "Error: %s" % err
    sys.exit(EX_TEMPFAIL)

#--------------------------------------------------------------------------
# TRAITEMENTS
#--------------------------------------------------------------------------

# par défaut les textes seront codés en Quoted-Printable
qp_charset = email.charset.Charset('utf-8')
qp_charset.header_encoding = email.charset.QP
qp_charset.body_encoding = email.charset.QP

# 05. si le destinataire est noreply@auf.org, on le remplace par l'émetteur
#     (ANNULÉ car ça ne devrait plus arriver)
#     (REPRIS pour le cas où le destinataire est impression-coda@auf.org)
if 'impression-coda@auf.org' in email_to_list:
    old_to = 'impression-coda@auf.org'
    email_to_list.remove(old_to)
    email_to_list.append(email_from)
    msg.replace_header('To', email_from)
    msg.add_header('X-Coda-Was-To', old_to)

# 10. on ajoute l'émetteur aux destinataires, sauf s'il est déjà le destinataire
#     (ANNULÉ car envoi déjà vers les bons destinataires)
#if email_from not in email_to_list and not email_from.endswith('@ca.auf.org'):
#    email_to_list.append(email_from)
#    msg.add_header('Cc', email_from)

# 20. on ajoute une en-tête pour aider aux filtrages
msg.add_header('X-Coda', X_CODA_VALUE)

# 30. on préfixe le sujet actuel par '[CODA] '
subject = msg['Subject']
try:
    parts = decode_header(subject)
    subject = u' '.join([unicode(s, e or 'ascii') for s,e in parts])
except:
    pass
msg.replace_header('Subject', SUBJECT_PREFIX + subject)

# 40. on remplace l'organisation par l'AuF
del msg['ORGANISATON'] # non, ce n'est pas une faute de frappe…
msg.add_header('Organization', ORGANIZATION)

# 50. on renomme l'attachement

# construction du nouveau nom de fichier
# on tente de repérer le type et numéro de document
try:
    m = re.match(DOC_PATTERN, subject)
    prefix, com_type, com_num = m.groups()
    new_filename = DOC_FORMAT % (com_type, int(com_num))
# en cas d'échec on ignore le problème
except Exception, err:
    print "Error: %s" % err
    new_filename = 'document-coda.pdf'

# parcours des différentes parties du courriel
for part in msg.walk():
    # traitement du corps de message pour y remplacer :
    #   http://coda.auf/coda/…
    # par :
    #   Accès depuis une implantation AUF : http://coda.auf/coda/…
    #   Accès depuis Internet : https://coda.auf.org/coda/…
    if part.get_content_type() == 'text/plain':
        text = part.get_payload(decode=True)
        new_text = re.sub(BODY_PATTERN_1, BODY_FORMAT_1, text)
        new_text = re.sub(BODY_PATTERN_2, BODY_FORMAT_2, new_text)
        if new_text != text:
            del part['Content-Transfer-Encoding']
            part.set_payload(new_text, qp_charset)
        continue
    # traitement uniquement si c'est un attachement de PDF
    if part.get_content_type() != 'application/pdf':
        continue
    # adaptation du nom de fichier dans le Content-Type
    try:
        raw_param = part.get_param('name')
        param = email.utils.collapse_rfc2231_value(raw_param)
        if param == 'attachment.pdf':
            part.set_param('name', new_filename)
    except:
        pass
    # adaptation du nom de fichier dans le Content-Disposition
    try:
        raw_param = part.get_param('filename', header='Content-Disposition')
        param = email.utils.collapse_rfc2231_value(raw_param)
        if param == 'attachment.pdf':
            part.set_param('filename', new_filename, header='Content-Disposition')
    except:
        pass

#--------------------------------------------------------------------------

# on se connecte à Postfix via le port de retour de filtrage
client = smtplib.SMTP('localhost', 10026)
#client.set_debuglevel(1)

# on tente d'envoyer le courriel modifié
try:
    client.sendmail(email_from, email_to_list, msg.as_string())
# en cas d'échec on demande à Postfix de considérer que c'est temporaire
except Exception, err:
    print "Error: %s" % err
    sys.exit(EX_TEMPFAIL)

# on termine proprement
client.quit()
sys.exit(0)