diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/archive.py | 29 | 
1 files changed, 17 insertions, 12 deletions
diff --git a/scripts/archive.py b/scripts/archive.py index ac62dc2..dbe54d9 100644 --- a/scripts/archive.py +++ b/scripts/archive.py @@ -16,33 +16,38 @@ def read_file(path, use_crlf):  	return data -def write_zip(target, arcprefix, sources): -	with zipfile.ZipFile(target, 'w', zipfile.ZIP_DEFLATED) as archive: +def write_zip(target, arcprefix, timestamp, sources): +	with zipfile.ZipFile(target, 'w') as archive:  		for source in sorted(sources):  			data = read_file(source, use_crlf = True)  			path = os.path.join(arcprefix, source) -			archive.writestr(path, data) +			info = zipfile.ZipInfo(path) +			info.date_time = time.localtime(timestamp) +			info.compress_type = zipfile.ZIP_DEFLATED +			archive.writestr(info, data) -def write_tar(target, arcprefix, sources, compression): +def write_tar(target, arcprefix, timestamp, sources, compression):  	with tarfile.open(target, 'w:' + compression) as archive:  		for source in sorted(sources):  			data = read_file(source, use_crlf = False)  			path = os.path.join(arcprefix, source)  			info = tarfile.TarInfo(path)  			info.size = len(data) -			info.mtime = time.time() +			info.mtime = timestamp  			archive.addfile(info, StringIO.StringIO(data)) -if len(sys.argv) < 4: -	raise RuntimeError('Usage: python archive.py <target> <archive prefix> <source files>') +if len(sys.argv) < 5: +	raise RuntimeError('Usage: python archive.py <target> <archive prefix> <timestamp> <source files>') -target = sys.argv[1] -arcprefix = sys.argv[2] -sources = sys.argv[3:] +target, arcprefix, timestamp = sys.argv[1:4] +sources = sys.argv[4:] + +# tarfile._Stream._init_write_gz always writes current time to gzip header +time.time = lambda: timestamp  if target.endswith('.zip'): -	write_zip(target, arcprefix, sources) +	write_zip(target, arcprefix, int(timestamp), sources)  elif target.endswith('.tar.gz') or target.endswith('.tar.bz2'): -	write_tar(target, arcprefix, sources, compression = os.path.splitext(target)[1][1:]) +	write_tar(target, arcprefix, int(timestamp), sources, compression = os.path.splitext(target)[1][1:])  else:  	raise NotImplementedError('File type not supported: ' + target)  | 
