Fetch the files

1 files changed, 26 insertions, 2 deletions

diff --git a/rpmbuild-remote.py b/rpmbuild-remote.py
index 7d3304c..9df2d32 100644
--- a/rpmbuild-remote.py
+++ b/rpmbuild-remote.py
@@ -116,11 +116,35 @@ def get_files(session, files, dest):
         sftp.get(file, dest)
     sftp.close()
 
+def log_regex(build_output):
+    out = open(build_output, 'r')
+    regex = re.compile("Wrote: (.+)")
+    files = regex.findall(out.readall())
+    out.close()
+
 def get_mock_results(session, dest, sout, serr):
-    pass
+    os.mkdirs(dest)
+    out = open(os.path.join(dest, 'mock.stdout'), 'w+')
+    out.write(sout.readall())
+    out.close()
+    err = open(os.path.join(dest, 'mock.stderr'), 'w+')
+    err.write(serr.readall())
+    err.close()
+    output_dir = re.compile(r'INFO: Results and/or logs in: (.+)').group(1)
+    logs = [os.path.join(output_dir, x)
+            for x in ['build.log', 'root.log', 'state.log']]
+    get_files(session, logs, dest)
+    get_files(os.path.join(dest, 'build.log'), dest)
 
 def get_rpmbuild_results(session, dest, sout, serr):
-    pass
+    os.mkdirs(dest)
+    out = open(os.path.join(dest, 'rpmbuild.stdout', 'w+'))
+    out.write(sout.readall())
+    out.close()
+    err = open(os.path.join(dest, 'rpmbuild.stderr', 'w+'))
+    err.write(serr.readall())
+    err.close()
+    get_files(os.path.join(dest, 'rpmbuild.stdout'), dest)
 
 def execute_jobs(opts, system, jobs):
     session = paramiko.SSHClient()