You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.5 KiB
54 lines
1.5 KiB
2 years ago
|
#! /usr/bin/env python
|
||
|
|
# encoding: utf-8
|
||
|
|
|
||
|
|
from waflib import TaskGen, Task, Logs
|
||
|
|
import zipfile
|
||
|
|
|
||
|
|
class ziparchive(Task.Task):
|
||
|
|
color = 'YELLOW'
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
tgt_str = ' '.join([a.path_from(a.ctx.launch_node()) for a in self.outputs])
|
||
|
|
count = len(self.inputs)
|
||
|
|
return '%s: %d files -> %s' % (self.__class__.__name__, count, tgt_str)
|
||
|
|
|
||
|
|
def keyword(self):
|
||
|
|
return 'Creating'
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
outfile = self.outputs[0].path_from(self.outputs[0].ctx.launch_node())
|
||
|
|
comp = zipfile.ZIP_STORED if self.compresslevel == 0 else zipfile.ZIP_DEFLATED
|
||
|
|
|
||
|
|
with zipfile.ZipFile(outfile, mode='w',
|
||
|
|
compression=comp, compresslevel=self.compresslevel,
|
||
|
|
strict_timestamps=False) as zf:
|
||
|
|
|
||
|
|
for src in self.inputs:
|
||
|
|
infile = src.path_from(src.ctx.launch_node())
|
||
|
|
arcfile = src.path_from(self.relative_to)
|
||
|
|
|
||
|
|
Logs.debug('%s: %s <- %s as %s', self.__class__.__name__, outfile, infile, arcfile)
|
||
|
|
zf.write(infile, arcfile)
|
||
|
|
|
||
|
|
@TaskGen.feature('zip')
|
||
|
|
def create_zip_archive(self):
|
||
|
|
compresslevel = getattr(self, 'compresslevel', 6) # 6 is zip default
|
||
|
|
if compresslevel < 0 or compresslevel > 9:
|
||
|
|
self.fatal('Invalid compress level')
|
||
|
|
|
||
|
|
files = getattr(self, 'files', None)
|
||
|
|
if not files:
|
||
|
|
self.fatal('No files to archive')
|
||
|
|
|
||
|
|
relative_to = getattr(self, 'relative_to', None)
|
||
|
|
if not relative_to:
|
||
|
|
self.fatal('No relative directory supplied')
|
||
|
|
|
||
|
|
self.path.get_bld().mkdir()
|
||
|
|
target = self.path.get_bld().make_node(self.name)
|
||
|
|
|
||
|
|
tsk = self.create_task('ziparchive', files, target)
|
||
|
|
|
||
|
|
setattr(tsk, 'compresslevel', compresslevel)
|
||
|
|
setattr(tsk, 'relative_to', relative_to)
|