Basic functionality for save_failed_tree plugin

plugins/builder/save_failed_tree.py

@@ -0,0 +1,40 @@
+import fnmatch
+import os
+import tarfile
+import koji.tasks as tasks
+from __main__ import BuildRoot
+
+__all__ = ('SaveFailedTreeTask',)
+
+def omit_ccache(tarinfo):
+    if fnmatch.fnmatch(tarinfo.name, '*/tmp/krb5cc') or \
+       fnmatch.fnmatch(tarinfo.name, '*/etc/*.keytab'):
+        return None
+    else:
+        return tarinfo
+
+
+
+class SaveFailedTreeTask(tasks.BaseTaskHandler):
+    Methods = ['saveFailedTree']
+    _taskWeight = 3.0
+
+
+    def handler(self, taskID, full=False):
+        self.logger.debug("Starting saving buildroots for task %d [full=%s]" % (taskID, full))
+        tar_path = os.path.join(self.workdir, 'broots-task-%s.tar.gz' % taskID)
+        f = tarfile.open(tar_path, "w:gz")
+        for broot in self.session.listBuildroots(taskID=taskID):
+            broot = BuildRoot(self.session, self.options, broot['id'])
+            path = broot.rootdir()
+            if full:
+                self.logger.debug("Adding buildroot (full): %s" % path)
+            else:
+                path = os.path.join(path, 'builddir')
+                self.logger.debug("Adding buildroot: %s" % path)
+            f.add(path, filter=omit_ccache)
+        f.close()
+        self.logger.debug("Uploading %s to hub." % tar_path)
+        self.uploadFile(tar_path)
+        os.unlink(tar_path)
+        self.logger.debug("Finished saving buildroots for task %d" % taskID)