mail-all reconfigure postfix

Sinon, il ne peut pas trop envoyer plein de mails...

darcs-hash:20080910105443-bd074-890fc925be9c04215a158acc5c70c5d4c31c316e.gz

gestion/mail-all.py

@@ -8,8 +8,70 @@ import smtplib
 import sys,os
 import ldap_crans
 import time
+import socket
 from email_tools import format_sender
 
+POSTFIX_CONF = "/etc/postfix/main.cf"
+LIMIT_CONF = "smtpd_client_message_rate_limit = 10\n"
+
+def reload_postfix():
+    os.system("/etc/init.d/postfix reload")
+
+def reconf_postfix():
+    # Est-ce qu'on est sur rouge ?!
+    if socket.gethostname() != "rouge":
+        print u"Il y a trop de destinataires, il faut se logger sur rouge pour lever\n(temporairement) la limite."
+        sys.exit(1)
+
+    conf_fd = open (POSTFIX_CONF, 'r')
+    lines = conf_fd.readlines()
+    nlines = []
+    conf_fd.close()
+
+    # On regarde s'il y a la limite dans le fichier de conf de postfix
+    limit="no"
+    for line in lines:
+        if line == LIMIT_CONF:
+            nlines+=u"## mail_all.py a commenté la ligne suivante"
+            nlines+="# smtpd_client_message_rate_limit = 10\n"
+            limit="yes"
+        else:
+            nlines+=line
+
+    if limit == "no":
+        print u"Fatal! Il n'y a pas l'air d'y avoir de limite dans le fichier de conf de\npostfix"
+        sys.exit(1)
+
+    if limit == "yes":
+        # Si oui, on demande à l'utilisateur si on y touche
+        
+        negatif = ["N", "n", ""]
+        positif = ["O", "o", "Y", "y"]
+        
+        poursuivre = "x"
+        while not (poursuivre in negatif + positif):
+            poursuivre = raw_input("Il y a trop de destinataires, il faut réécrire la conf de postfix [o/N] ")
+        
+        if poursuivre in negatif:
+            print "Ok, on ne touche pas au fichier..."
+            sys.exit(0)
+
+        assert (poursuivre in positif)
+
+        # On réécrit la conf
+        try:
+            nconf_fd = open (POSTFIX_CONF, "w")
+            print u"Réécriture de la configuration de postfix"
+            nconf_fd.writelines(nlines)
+            nconf_fd.close ()
+        except IOError:
+            print u"Je n'arrive pas à réécrire la conf de postfix, es-tu root ?"
+            sys.exit(1)
+
+        reload_postfix()
+            
+        return(limit,lines)
+
 if __name__ == "__main__":
     if len(sys.argv) != 3:
         print """Usage:
@@ -26,10 +88,16 @@ Le second param
 doit également contenir les entêtes, à l'exception du destinataire qui sera rajouté
 par le script.
 
- /!\ Ce script ne demande aucune confirmation, il faut veiller à vérifier avec who
+ /!\ Ce script ne demande aucune confirmation, il faut veiller à
-     que l'on cible bien les utilisateurs que l'on veut. Et si on veut vérifier que
+     vérifier avec who que l'on cible bien les utilisateurs que l'on
-     le mail a une bonne tête on se l'envoie d'abord en mettant login=machin
+     veut. Et si on veut vérifier que le mail a une bonne tête on se
-     comme critère de recherche.
+     l'envoie d'abord en mettant login=machin comme critère de
+     recherche.
+
+ /!\ Ce script a pour but de spammer, et spammer c'est mal. Il faut
+     donc contourner les limitations qui ont été mises en place au
+     Cr@ns. Si plus de 10 mails doivent être envoyés, il faut le faire
+     depuis rouge, et ce en root.
 """
 
         sys.exit(0)
@@ -37,39 +105,60 @@ par le script.
     # On en est là
     # On ouvre la base et on cherche
     adherents = ldap_crans.crans_ldap().search(sys.argv[1])['adherent']
-
+    card= len(adherents)
-    print "%d adhérent(s) a/ont été trouvé(s)..." % len(adherents)
+    print "%d adhérent(s) a/ont été trouvé(s)..." % card
     time.sleep(3)                       # On dort un peu, ctrl-c welcome
 
+    limit="no"
+    if card >= 10:
+        (limit, backup_conf) = reconf_postfix()
+
+    # Il faudra quoiqu'il arrive rétablir la conf de postfix
+    # try capture-t-il les SIGTERM ?!
     try:
-        texte = "".join(file(sys.argv[2], "r").readlines())
-    except IOError:
-        print "Impossible d'ouvrir le fichier à envoyer, merci, au revoir."
-        sys.exit(1)
-
-    echecs = []
-    s = smtplib.SMTP()
-    s.connect('smtp.crans.org')
-    for adherent in adherents:
-        mail = adherent.mail().encode("iso-8859-15", "ignore")
-        if "@" not in mail:
-            mail = mail + "@crans.org"
-        print "Envoi du mail à %s <%s>..." % (adherent.Nom().encode("iso-8859-15","ignore"), mail)
         try:
-            recipient = format_sender(u'"%s" <%s>\n' % (adherent.Nom(), mail))
+            texte = "".join(file(sys.argv[2], "r").readlines())
-            s.sendmail('bulk+%s@crans.org' % mail.replace("@",'-at-'),
+        except IOError:
-            (mail,),
+            print "Impossible d'ouvrir le fichier à envoyer, merci, au revoir."
-            "To: %s\n%s" % (recipient, texte))
+            sys.exit(1)
-        except:
-            print "Erreur lors de l'envoi à %s <%s>..." % (adherent.Nom().encode("iso-8859-15","ignore"), mail)
-            echecs.append(mail)
-        else:
-            # Tout va bien
-            pass
     
-    if echecs:
+        echecs = []
-        print "\nIl y a eu des erreurs :"
+        s = smtplib.SMTP()
-        print echecs
+        s.connect('smtp.crans.org')
+        for adherent in adherents:
+            mail = adherent.mail().encode("iso-8859-15", "ignore")
+            if "@" not in mail:
+                mail = mail + "@crans.org"
+            print "Envoi du mail à %s <%s>..." % (adherent.Nom().encode("iso-8859-15","ignore"), mail)
+            try:
+                recipient = format_sender(u'"%s" <%s>\n' % (adherent.Nom(), mail))
+                s.sendmail('bulk+%s@crans.org' % mail.replace("@",'-at-'),
+                (mail,),
+                "To: %s\n%s" % (recipient, texte))
+            except:
+                print "Erreur lors de l'envoi à %s <%s>..." % (adherent.Nom().encode("iso-8859-15","ignore"), mail)
+                echecs.append(mail)
+            else:
+                # Tout va bien
+                pass
+            
+        if echecs:
+            print "\nIl y a eu des erreurs :"
+            print echecs
+    
+        s.close()
+
+    # On rétablit la conf de postfix    
+    finally:
+        if limit == "yes":
+            try:
+                conf_fd= open (POSTFIX_CONF, "w")
+                print u"Restauration de la configuration de postfix"
+                conf_fd.writelines (backup_conf)
+                conf_fd.close()
+                reload_postfix()
+            except IOError:
+                print "Could not rewrite postfix conf..."
+                sys.exit(1)
 
-    s.close()