recredite.py -> recredit_masse.py dans impression

impression/recredit_masse.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+""" Pour recréditer facilement des jobs échoués. """
+
+usage = u"""Usage : recredite.py JID1 JID2 JID3
+ou recredite.py --csv FILE.CSV
+
+Pour chaque job, va chercher le montant dans l'historique et recrédite l'adhérent.
+Avec l'option --csv, parse un journal de l'imprimante et recrédite les jobs NG."""
+
+import re
+import time
+import csv
+import sys
+if not "/usr/scripts" in sys.path:
+    sys.path.insert(0, "/usr/scripts")
+
+import lc_ldap.shortcuts
+
+def get_adh_montant(jid, conn=None):
+    """Récupère l'adhérent à recréditer et le montant correspondants à ce jid."""
+    if conn == None:
+        conn = lc_ldap.shortcuts.lc_ldap_admin()
+    adh = conn.search(u"(historique=*debit*impression\(%s\)*)" % jid, mode='rw')
+    if not adh:
+        print "jid non trouvé dans l'historique LDAP : %s" % jid
+        return None
+    adh = adh[0]
+    ligne = [l.value for l in adh["historique"] if "impression(%s)" % jid in l.value][0]
+    montant = re.match(u".*debit (?P<montant>[0-9]+\.[0-9]+) Euros", ligne).groupdict()
+    montant = float(montant["montant"])
+    return (adh, montant)
+
+def recredite(adh, montant, jid):
+    """Effectue le crédit, sauf si il a déjà été fait."""
+    # On vérifie que le remboursement n'a pas déjà été fait
+    deja = [l.value for l in adh["historique"] if "remboursement(%s)" % jid in l.value]
+    if deja:
+        print "[jid=%s] %s (%s) déjà recrédité !" % (jid, adh["uid"][0], adh.dn.split(",")[0])
+        return
+    timestamp = time.strftime("%d/%m/%Y %H:%M")
+    historique = "%s, %s : credit %.2f Euros [remboursement(%s)]" % (jid, timestamp, lc_ldap.shortcuts.current_user, montant, jid)
+    adh["historique"].append(historique)
+    solde = adh["solde"][0].value
+    adh["solde"][0] = solde + montant
+    print "%s (%s) recrédité de %.2f euros (nouveau solde : %s)" % (adh["uid"][0], adh.dn.split(",")[0], montant, adh["solde"][0])
+    adh.save()
+
+def parse_csv(filename):
+    """Parse le .csv du journal de l'imprimante pour en extraire les jid des jobs en NG."""
+    with open(filename) as f:
+        csvreader = csv.reader(f)
+        jobs = [l[2] for l in csvreader if l and l[1] == 'NG']
+    jids = [int(j[1:].split(":")[0]) for j in jobs]
+    return jids
+    
+
+if __name__ == "__main__":
+    if len(sys.argv) == 1:
+        print usage
+        sys.exit(1)
+    if "--csv" in sys.argv:
+        jids = parse_csv(sys.argv[2])
+        print "jids parsés : %s" % jids
+    else:
+        jids = [int(jid) for jid in sys.argv[1:]]
+    conn = lc_ldap.shortcuts.lc_ldap_admin()
+    for jid in jids:
+        jid = int(jid)
+        adh, montant = get_adh_montant(jid, conn)
+        recredite(adh, montant, jid)