mirror of
https://github.com/apache/cloudstack.git
synced 2025-11-04 00:02:37 +01:00
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
import Utils
|
|
import Options
|
|
import tarfile
|
|
from TaskGen import feature, before
|
|
import Task
|
|
import os, sys
|
|
|
|
# this is a clever little thing
|
|
# given a list of nodes, build or source
|
|
# construct a tar file containing them
|
|
# rooted in the parameter root =, specified in the task generator
|
|
# and renaming the names of the files according to a rename(x) function passed to the task generator as well
|
|
# if a build node's result of rename() has the same name as a source node, the build node will take precedence
|
|
# for as long as the build node appears later than the source node (this is an implementation detail of waf we are relying on)
|
|
def tar_up(task):
|
|
tgt = task.outputs[0].bldpath(task.env)
|
|
if os.path.exists(tgt): os.unlink(tgt)
|
|
if tgt.lower().endswith(".bz2"): z = tarfile.open(tgt,"w:bz2")
|
|
elif tgt.lower().endswith(".gz"): z = tarfile.open(tgt,"w:gz")
|
|
elif tgt.lower().endswith(".tgz"): z = tarfile.open(tgt,"w:gz")
|
|
else: z = tarfile.open(tgt,"w")
|
|
fileset = {}
|
|
for inp in task.inputs:
|
|
src = inp.srcpath(task.env)
|
|
if src.startswith(".."):
|
|
srcname = Utils.relpath(src,os.path.join("..",".")) # file in source dir
|
|
else:
|
|
srcname = Utils.relpath(src,os.path.join(task.env.variant(),".")) # file in artifacts dir
|
|
srcname = srcname.split(os.path.sep,len(task.generator.root.split(os.path.sep)))[-1]
|
|
if task.generator.rename: srcname = task.generator.rename(srcname)
|
|
fileset[srcname] = src
|
|
for srcname,src in fileset.items():
|
|
ti = tarfile.TarInfo(srcname)
|
|
ti.mode = 0755
|
|
ti.size = os.path.getsize(src)
|
|
openmode = 'r'
|
|
if Options.platform == 'win32': openmode = openmode + 'b'
|
|
f = file(src,openmode)
|
|
z.addfile(ti,fileobj=f)
|
|
f.close()
|
|
z.close()
|
|
if task.chmod: os.chmod(tgt,task.chmod)
|
|
return 0
|
|
|
|
def apply_tar(self):
|
|
Utils.def_attrs(self,fun=tar_up)
|
|
self.default_install_path=0
|
|
lst=self.to_list(self.source)
|
|
self.meths.remove('apply_core')
|
|
self.dict=getattr(self,'dict',{})
|
|
out = self.path.find_or_declare(self.target)
|
|
ins = []
|
|
for x in Utils.to_list(self.source):
|
|
node = self.path.find_resource(x)
|
|
if not node:raise Utils.WafError('cannot find input file %s for processing'%x)
|
|
ins.append(node)
|
|
tsk=self.create_task('tar',ins,out)
|
|
tsk.fun=self.fun
|
|
tsk.dict=self.dict
|
|
tsk.install_path=self.install_path
|
|
tsk.chmod=self.chmod
|
|
if not tsk.env:
|
|
tsk.debug()
|
|
raise Utils.WafError('task without an environment')
|
|
|
|
Task.task_type_from_func('tar',func=tar_up)
|
|
feature('tar')(apply_tar)
|
|
before('apply_core')(apply_tar)
|