scripts/impression/print_status.py

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

"""
print_status.py

Récupère le statut des dernières impressions auprès de l'imprimante et envoie un
mail à l'intérressé quand son impression est sortie, ou envoie un message aux
imprimeurs si l'impression semble avoir échoué.

Auteur : Sylvain Boilard <boilard@crans.org>
License : BSD 2 clauses.

"""

import sys, os
sys.path.append("/usr/scripts/gestion")

import time
import httplib2
import re
import smtplib
from ldap_crans import crans_ldap

last_print_filename = "/var/run/print_status/last_print.txt"
printer_host ="imprimante.adm.crans.org"
smtp_server= "smtp.crans.org"
error_send_to= u"impression@crans.org"
ok_mail=u"""Content-Type: text/plain; charset=utf-8
Content-Disposition: inline
Content-Transfer-Encoding: quoted-printable
From: %s
To: %s
Subject: [Impression] Impression de "%s" terminée.

Bonjour %s,

Le fichier "%s" que tu as envoyé %s
a été imprimé. Tu peux venir le récupérer au local impression, au 4ème étage du
bâtiment J, muni du code secret qui t'a été donné lorsque tu as envoyé ton
document. Si tu as oublié ce code, ou pour tout autre problème lié au service
d'impression, tu peux contacter les imprimeurs à l'adresse %s .

Bonne journée,

-- 
Les imprimeurs du Cr@ns
"""
error_mail=u"""Content-Transfer-Encoding: quoted-printable
Content-Type: text/plain; charset=utf-8
Content-Disposition: inline
From: %s
To: %s
Subject: [Impression][print_status.py] Impression potentiellement échouée.

Bonjour,

Il semble y avoir eu un problème avec l'impression du fichier
"%s" (tâche n°%s) de %s (uid=%s),
lancée %s.
Le statut de la tâche est "%s".

Aucun autre mail ne sera envoyé à ce sujet. Si nécéssaire, veuillez tenir
l'intéressé au courant aux adresses suivantes :

%s

Bonne journée,

-- 
print_status.py
"""

litMonths = [u"Janvier", u"Février", u"Mars", u"Avril", u"Mai", u"Juin", u"Juillet", u"Août", u"Septembre", u"Octobre", u"Novembre", u"Décembre"]

def getMSE():
    """Retourne le nombre de millisecondes depuis le 1 Janvier 1970 sous forme d'une chaîne de caractères."""
    return str(int(time.time()*1000.))

def buildDate(date):
    """Retourne la date et l'heure passée en argument après l'avoir mise en forme."""
    dm, year, hms = date.split(' ')
    day, month = dm.split('/')
    hour, minute, _ = hms.split(':')
    return u"le " + day + u" " + litMonths[int(month) - 1] + u" " + year + u" à " + hour + u":" + minute

def sendMail(from_addr, to_addrs, mail_content):
    """Wrapper pour smtplib.sendmail(). Permet d'attendre un peu en cas d'erreur temporaire (typiquement, si on essaie d'envoyer trop de mails à la minute)."""
    conn = SMTP(smtp_server)
    try:
        conn.sendmail(from_addr, to_addrs, mail_content)
    except smtplib.SMTPSenderRefused as (code, _, _):
        if code == 450:
            time.sleep(65)
            conn.sendmail(from_addr, to_addrs, mail_content)
        else:
            raise
    conn.quit()

http = httplib2.Http()

# On récupère la liste des tâches et on enlève les entêtes.
# Pour faire la requête, on doit récupérer un cookie de session.
headers, _ = http.request("https://" + printer_host + "/twelcome.cgi?CorePGTAG=0&Dummy=" + getMSE())
_, content = http.request("https://" + printer_host + "/pprint.csv?Flag=Csv_Data&LogType=0&Dummy=" + getMSE(), 'GET', headers={'Cookie': headers['set-cookie']})
task_list = content.split('\n')
task_list.pop(0)

# On récupère la dernière tâche à avoir été traitée la dernière fois quel le script a été lancé.
# Si le fichier n'existe pas, on le crée en mettant la première tâche parmi celles que l'on vient de récupérer, et on s'arrête là.
try:
    last_file = open(last_print_filename, "r")
    last = last_file.read()
    last_file.close()
except:
    last_file = open(last_print_filename, "w")
    last_file.write(task_list[0].split(',')[0])
    last_file.close()
    exit(0)

db = crans_ldap()
match_doc_name = re.compile(r"/[^/]*\.pdf")

# On parcoure les tâches récupérées et qui n'ont pas déja été traitées, et on agit en conséquence.
for item in task_list:
    fields = item.split(',', 6)
    # On notera la présence d'une tâche complètement vide à la fin de la liste.
    if fields[0] == last or fields[0] == '':
        break
    if fields[3].strip('"') == "root":
        result = fields[1].strip('"')
        taskID, user, _ = fields[2].strip('"').split(':')
        user = user.split('@').pop() # On récupère le nom du club si besoin.
        date = buildDate(fields[5])
        match_taskID = re.compile(r"impression\(%s\)" % taskID)
        db_query = db.search("uid=%s" % user)
        if db_query['adherent']:
            target = db_query['adherent'][0]
            full_name = target.prenom() + u" " + target.nom()
            send_to = [ target.mail() + u"@crans.org" ]
        else:
            target = db_query['club'][0]
            full_name = u"Club " + target.nom()
            send_to = map(lambda aid : db.search("aid=" + aid)['adherent'][0].mail() + u"@crans.org", target.imprimeurs())
        historique = target.historique()
        historique.reverse() # Ce qu'on cherche a des chances d'être récent et donc d'être à la fin de l'historique.
        for hist_line in historique:
            if match_taskID.search(hist_line):
                filename = match_doc_name.search(hist_line).group()[1:]
                break
        if result == "OK":
            mail_content = ok_mail % (error_send_to, u", ".join(send_to), filename, full_name, filename, date, error_send_to)
            sendMail(error_send_to, send_to, mail_content.encode("utf-8"))
        else:
            mail_content = error_mail % (error_send_to, error_send_to, filename, taskID, full_name, user, date, result, u"\n".join(send_to))
            sendMail(error_send_to, error_send_to, mail_content.encode("utf-8"))

# Éventuellement, on met à jour le numéro de la dernière tâche traitée.
new_last = task_list[0].split(',', 1)[0]
if last != new_last:
    last_file = open(last_print_filename, "w")
    last_file.write(new_last)
    last_file.close()