Add the tool to send email
authorMagnus Hagander <magnus@hagander.net>
Mon, 19 Jan 2015 21:02:58 +0000 (22:02 +0100)
committerMagnus Hagander <magnus@hagander.net>
Mon, 19 Jan 2015 21:02:58 +0000 (22:02 +0100)
Forgot to git add that one, it seems..

tools/mail/send_queued_mail.py [new file with mode: 0755]

diff --git a/tools/mail/send_queued_mail.py b/tools/mail/send_queued_mail.py
new file mode 100755 (executable)
index 0000000..5a8005f
--- /dev/null
@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+#
+# Script to send off all queued email.
+#
+# This script is intended to be run frequently from cron. We queue things
+# up in the db so that they get automatically rolled back as necessary,
+# but once we reach this point we're just going to send all of them one
+# by one.
+#
+
+import sys
+import os
+import smtplib
+
+# Set up to run in django environment
+from django.core.management import setup_environ
+sys.path.append(os.path.join(os.path.abspath(os.path.dirname(sys.argv[0])), '../../pgcommitfest'))
+import settings
+setup_environ(settings)
+
+from django.db import connection, transaction
+
+from pgcommitfest.mailqueue.models import QueuedMail
+
+if __name__ == "__main__":
+       # Grab advisory lock, if available. Lock id is just a random number
+       # since we only need to interlock against ourselves. The lock is
+       # automatically released when we're done.
+       curs = connection.cursor()
+       curs.execute("SELECT pg_try_advisory_lock(72181379)")
+       if not curs.fetchall()[0][0]:
+               print "Failed to get advisory lock, existing send_queued_mail process stuck?"
+               connection.close()
+               sys.exit(1)
+
+       for m in QueuedMail.objects.all():
+               # Yes, we do a new connection for each run. Just because we can.
+               # If it fails we'll throw an exception and just come back on the
+               # next cron job. And local delivery should never fail...
+               smtp = smtplib.SMTP("localhost")
+               smtp.sendmail(m.sender, m.receiver, m.fullmsg.encode('utf-8'))
+               smtp.close()
+               m.delete()
+               transaction.commit()
+       connection.close()