176 lines
5.9 KiB
Python
176 lines
5.9 KiB
Python
|
# -*- python -*-
|
||
|
#
|
||
|
# Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010 The SCons Foundation
|
||
|
#
|
||
|
# Permission is hereby granted, free of charge, to any person obtaining
|
||
|
# a copy of this software and associated documentation files (the
|
||
|
# "Software"), to deal in the Software without restriction, including
|
||
|
# without limitation the rights to use, copy, modify, merge, publish,
|
||
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
||
|
# permit persons to whom the Software is furnished to do so, subject to
|
||
|
# the following conditions:
|
||
|
#
|
||
|
# The above copyright notice and this permission notice shall be included
|
||
|
# in all copies or substantial portions of the Software.
|
||
|
#
|
||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
|
||
|
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
|
||
|
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
||
|
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
||
|
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||
|
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
#
|
||
|
|
||
|
__doc__ = """
|
||
|
Textfile/Substfile builder for SCons.
|
||
|
|
||
|
Create file 'target' which typically is a textfile. The 'source'
|
||
|
may be any combination of strings, Nodes, or lists of same. A
|
||
|
'linesep' will be put between any part written and defaults to
|
||
|
os.linesep.
|
||
|
|
||
|
The only difference between the Textfile builder and the Substfile
|
||
|
builder is that strings are converted to Value() nodes for the
|
||
|
former and File() nodes for the latter. To insert files in the
|
||
|
former or strings in the latter, wrap them in a File() or Value(),
|
||
|
respectively.
|
||
|
|
||
|
The values of SUBST_DICT first have any construction variables
|
||
|
expanded (its keys are not expanded). If a value of SUBST_DICT is
|
||
|
a python callable function, it is called and the result is expanded
|
||
|
as the value. Values are substituted in a "random" order; if any
|
||
|
substitution could be further expanded by another subsitition, it
|
||
|
is unpredictible whether the expansion will occur.
|
||
|
"""
|
||
|
|
||
|
__revision__ = "src/engine/SCons/Tool/textfile.py 5134 2010/08/16 23:02:40 bdeegan"
|
||
|
|
||
|
import SCons
|
||
|
|
||
|
import os
|
||
|
import re
|
||
|
|
||
|
from SCons.Node import Node
|
||
|
from SCons.Node.Python import Value
|
||
|
from SCons.Util import is_String, is_Sequence, is_Dict
|
||
|
|
||
|
def _do_subst(node, subs):
|
||
|
"""
|
||
|
Fetch the node contents and replace all instances of the keys with
|
||
|
their values. For example, if subs is
|
||
|
{'%VERSION%': '1.2345', '%BASE%': 'MyProg', '%prefix%': '/bin'},
|
||
|
then all instances of %VERSION% in the file will be replaced with
|
||
|
1.2345 and so forth.
|
||
|
"""
|
||
|
contents = node.get_text_contents()
|
||
|
if not subs: return contents
|
||
|
for (k,v) in subs:
|
||
|
contents = re.sub(k, v, contents)
|
||
|
return contents
|
||
|
|
||
|
def _action(target, source, env):
|
||
|
# prepare the line separator
|
||
|
linesep = env['LINESEPARATOR']
|
||
|
if linesep is None:
|
||
|
linesep = os.linesep
|
||
|
elif is_String(linesep):
|
||
|
pass
|
||
|
elif isinstance(linesep, Value):
|
||
|
linesep = linesep.get_text_contents()
|
||
|
else:
|
||
|
raise SCons.Errors.UserError(
|
||
|
'unexpected type/class for LINESEPARATOR: %s'
|
||
|
% repr(linesep), None)
|
||
|
|
||
|
# create a dictionary to use for the substitutions
|
||
|
if 'SUBST_DICT' not in env:
|
||
|
subs = None # no substitutions
|
||
|
else:
|
||
|
d = env['SUBST_DICT']
|
||
|
if is_Dict(d):
|
||
|
d = list(d.items())
|
||
|
elif is_Sequence(d):
|
||
|
pass
|
||
|
else:
|
||
|
raise SCons.Errors.UserError('SUBST_DICT must be dict or sequence')
|
||
|
subs = []
|
||
|
for (k,v) in d:
|
||
|
if callable(v):
|
||
|
v = v()
|
||
|
if is_String(v):
|
||
|
v = env.subst(v)
|
||
|
else:
|
||
|
v = str(v)
|
||
|
subs.append((k,v))
|
||
|
|
||
|
# write the file
|
||
|
try:
|
||
|
fd = open(target[0].get_path(), "wb")
|
||
|
except (OSError,IOError), e:
|
||
|
raise SCons.Errors.UserError("Can't write target file %s" % target[0])
|
||
|
# separate lines by 'linesep' only if linesep is not empty
|
||
|
lsep = None
|
||
|
for s in source:
|
||
|
if lsep: fd.write(lsep)
|
||
|
fd.write(_do_subst(s, subs))
|
||
|
lsep = linesep
|
||
|
fd.close()
|
||
|
|
||
|
def _strfunc(target, source, env):
|
||
|
return "Creating '%s'" % target[0]
|
||
|
|
||
|
def _convert_list_R(newlist, sources):
|
||
|
for elem in sources:
|
||
|
if is_Sequence(elem):
|
||
|
_convert_list_R(newlist, elem)
|
||
|
elif isinstance(elem, Node):
|
||
|
newlist.append(elem)
|
||
|
else:
|
||
|
newlist.append(Value(elem))
|
||
|
def _convert_list(target, source, env):
|
||
|
if len(target) != 1:
|
||
|
raise SCons.Errors.UserError("Only one target file allowed")
|
||
|
newlist = []
|
||
|
_convert_list_R(newlist, source)
|
||
|
return target, newlist
|
||
|
|
||
|
_common_varlist = ['SUBST_DICT', 'LINESEPARATOR']
|
||
|
|
||
|
_text_varlist = _common_varlist + ['TEXTFILEPREFIX', 'TEXTFILESUFFIX']
|
||
|
_text_builder = SCons.Builder.Builder(
|
||
|
action = SCons.Action.Action(_action, _strfunc, varlist = _text_varlist),
|
||
|
source_factory = Value,
|
||
|
emitter = _convert_list,
|
||
|
prefix = '$TEXTFILEPREFIX',
|
||
|
suffix = '$TEXTFILESUFFIX',
|
||
|
)
|
||
|
|
||
|
_subst_varlist = _common_varlist + ['SUBSTFILEPREFIX', 'TEXTFILESUFFIX']
|
||
|
_subst_builder = SCons.Builder.Builder(
|
||
|
action = SCons.Action.Action(_action, _strfunc, varlist = _subst_varlist),
|
||
|
source_factory = SCons.Node.FS.File,
|
||
|
emitter = _convert_list,
|
||
|
prefix = '$SUBSTFILEPREFIX',
|
||
|
suffix = '$SUBSTFILESUFFIX',
|
||
|
src_suffix = ['.in'],
|
||
|
)
|
||
|
|
||
|
def generate(env):
|
||
|
env['LINESEPARATOR'] = os.linesep
|
||
|
env['BUILDERS']['Textfile'] = _text_builder
|
||
|
env['TEXTFILEPREFIX'] = ''
|
||
|
env['TEXTFILESUFFIX'] = '.txt'
|
||
|
env['BUILDERS']['Substfile'] = _subst_builder
|
||
|
env['SUBSTFILEPREFIX'] = ''
|
||
|
env['SUBSTFILESUFFIX'] = ''
|
||
|
|
||
|
def exists(env):
|
||
|
return 1
|
||
|
|
||
|
# Local Variables:
|
||
|
# tab-width:4
|
||
|
# indent-tabs-mode:nil
|
||
|
# End:
|
||
|
# vim: set expandtab tabstop=4 shiftwidth=4:
|