Bug #324191 - Add support for FEATURES=compress-build-logs. The causes

all build logs to be compressed while they are being written. Log file
names have an extension that is appropriate for the compression type.
Currently, only gzip(1) compression is supported, so build logs will
have a '.gz' extension when this feature is enabled.

1 files changed, 14 insertions, 2 deletions

diff --git a/pym/_emerge/SpawnProcess.py b/pym/_emerge/SpawnProcess.py
index 6e0586815..aeb206a06 100644
--- a/pym/_emerge/SpawnProcess.py
+++ b/pym/_emerge/SpawnProcess.py
@@ -6,10 +6,13 @@ from _emerge.PollConstants import PollConstants
 import sys
 from portage.cache.mappings import slot_dict_class
 import portage
+from portage import _encodings
+from portage import _unicode_encode
 from portage import os
 import fcntl
 import errno
 import array
+import gzip
 
 class SpawnProcess(SubProcess):
 
@@ -77,7 +80,12 @@ class SpawnProcess(SubProcess):
 			fd_pipes[1] = slave_fd
 			fd_pipes[2] = slave_fd
 
-			files.log = open(logfile, mode='ab')
+			files.log = open(_unicode_encode(logfile,
+				encoding=_encodings['fs'], errors='strict'), mode='ab')
+			if logfile.endswith('.gz'):
+				files.log = gzip.GzipFile(filename='', mode='ab',
+					fileobj=files.log)
+
 			portage.util.apply_secpass_permissions(logfile,
 				uid=portage.portage_uid, gid=portage.portage_gid,
 				mode=0o660)
@@ -188,7 +196,11 @@ class SpawnProcess(SubProcess):
 								fcntl.fcntl(files.stdout.fileno(),
 								fcntl.F_GETFL) ^ os.O_NONBLOCK)
 
-				buf.tofile(files.log)
+				try:
+					buf.tofile(files.log)
+				except TypeError:
+					# array.tofile() doesn't work with GzipFile
+					files.log.write(buf.tostring())
 				files.log.flush()
 			else:
 				self._unregister()