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

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

diff --git a/pld-builder.new/PLD_Builder/ftp.py b/pld-builder.new/PLD_Builder/ftp.py
new file mode 100644 (file)
index 0000000..0d13be6
--- /dev/null
+++ b/pld-builder.new/PLD_Builder/ftp.py
@@ -0,0 +1,77 @@
+# vi: encoding=utf-8 ts=8 sts=4 sw=4 et
+
+import path
+import os
+import shutil
+import time
+
+from config import config
+import util
+
+class FTP_Queue:
+    def __init__(self):
+        self.queue = None
+        self.some_failed = 0
+        self.status = ""
+
+    def init(self, g=None, rpmqa=False):
+        self.queue = []
+        if rpmqa:
+            self.requester_email = config.admin_email
+            self.url = config.rpmqa_url
+        else:
+            self.requester_email = g.requester_email
+            if "test-build" in g.flags:
+                self.url = config.test_ftp_url
+            else:
+                self.url = config.ftp_url
+
+    def add(self, file, type):
+        # if /dev/null, say bye bye
+        if self.url == "/dev/null":
+            return
+        name = os.path.basename(file)
+        id = util.uuid()
+        shutil.copy(file, path.ftp_queue_dir + '/' + id)
+        self.queue.append({'name': name, 'id': id, 'type': type})
+        st = os.stat(path.ftp_queue_dir + '/' + id)
+        self.status += "%10d %s\n" % (st.st_size, name)
+
+    def flush(self):
+        def desc(l):
+            return """Target: %s/%s
+Builder: %s
+Time: %d
+Type: %s
+Requester: %s
+END
+""" % (self.url, l['name'], config.builder, time.time(), l['type'], self.requester_email)
+
+        for l in self.queue:
+            f = open(path.ftp_queue_dir + '/' + l['id'] + ".desc", "w")
+            f.write(desc(l))
+            f.close()
+
+    def kill(self):
+        for l in self.queue:
+            os.unlink(path.ftp_queue_dir + '/' + l)
+
+queue = FTP_Queue()
+
+def add(f, type="rpm"):
+    queue.add(f, type)
+
+def flush():
+    queue.flush()
+
+def kill():
+    queue.kill()
+
+def init(r=None, rpmqa=False):
+    queue.init(r, rpmqa)
+
+def status():
+    return queue.status
+
+def clear_status():
+    queue.status = ""