- from https://github.com/pld-linux/pld-builder.new

[TLD.git] / pld-builder.new / PLD_Builder / notify.py

diff --git a/pld-builder.new/PLD_Builder/notify.py b/pld-builder.new/PLD_Builder/notify.py
new file mode 100644 (file)
index 0000000..5b0368f
--- /dev/null
+++ b/pld-builder.new/PLD_Builder/notify.py
@@ -0,0 +1,41 @@
+# vi: encoding=utf-8 ts=8 sts=4 sw=4 et
+
+import StringIO
+
+import mailer
+import gpg
+import util
+import notifyq
+from config import config
+
+class Notifier:
+    def __init__(self, g):
+        self.xml = StringIO.StringIO()
+        self.xml.write("<notification group-id='%s' builder='%s'>\n" % \
+                        (g.id, config.builder))
+
+    def send(self, r):
+        sio = StringIO.StringIO()
+        self.xml.write("</notification>\n")
+        self.xml.seek(0)
+        sio.write(gpg.sign(self.xml.read()))
+        self.xml = None
+        sio.seek(0)
+        notifyq.init(r)
+        notifyq.add(sio)
+        notifyq.flush()
+
+    def add_batch(self, b, s):
+        self.xml.write("  <batch id='%s' status='%s' />\n" % (b.b_id, s))
+
+n = None
+
+def begin(group):
+    global n
+    n = Notifier(group)
+
+def add_batch(batch, status):
+    n.add_batch(batch, status)
+
+def send(r):
+    n.send(r)