waf/playground/strip/strip.py

#! /usr/bin/env python

"""
Strip a program/library after it is created. Use this tool as an example.

Usage::

	bld.program(features='strip', source='main.c', target='foo')

If stripping at installation time is preferred, override/modify the method
copy_fun on the installation context::

	import shutil, os
	from waflib import Build
	from waflib.Tools import ccroot
	def copy_fun(self, src, tgt, **kw):
		shutil.copy2(src, tgt)
		os.chmod(tgt, kw.get('chmod', Utils.O644))
		try:
			tsk = kw['tsk']
		except KeyError:
			pass
		else:
			if isinstance(tsk.task, ccroot.link_task):
				self.cmd_and_log('strip %s' % tgt)
	Build.InstallContext.copy_fun = copy_fun
"""

def configure(conf):
	conf.find_program('strip')

from waflib import Task, TaskGen
class strip(Task.Task):
	run_str = '${STRIP} ${SRC}'
	color   = 'BLUE'
	before  = ['inst']

@TaskGen.feature('strip')
@TaskGen.after('apply_link')
def add_strip_task(self):
	try:
		link_task = self.link_task
	except AttributeError:
		return
	self.create_task('strip', link_task.outputs[0])

@TaskGen.feature('cshlib', 'cxxshlib', 'cprogram', 'cxxprogram', 'fcprogram', 'fcshlib')
@TaskGen.after('process_use', 'apply_link')
def set_strip_order(self):
	# as an example
	if getattr(self, 'link_task', None):
		for x in self.tmp_use_seen:
			tg = self.bld.get_tgen_by_name(x)
			if getattr(tg, 'link_task', None):
				self.link_task.set_run_after(tg.link_task)