cpp/scons/scons-local-2.0.0.final.0/SCons/Tool/zip.py

   1 """SCons.Tool.zip
   2
   3 Tool-specific initialization for zip.
   4
   5 There normally shouldn't be any need to import this module directly.
   6 It will usually be imported through the generic SCons.Tool.Tool()
   7 selection method.
   8
   9 """
  10
  11 #
  12 # Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010 The SCons Foundation
  13 #
  14 # Permission is hereby granted, free of charge, to any person obtaining
  15 # a copy of this software and associated documentation files (the
  16 # "Software"), to deal in the Software without restriction, including
  17 # without limitation the rights to use, copy, modify, merge, publish,
  18 # distribute, sublicense, and/or sell copies of the Software, and to
  19 # permit persons to whom the Software is furnished to do so, subject to
  20 # the following conditions:
  21 #
  22 # The above copyright notice and this permission notice shall be included
  23 # in all copies or substantial portions of the Software.
  24 #
  25 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  26 # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  27 # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  28 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  29 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  30 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  31 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  32 #
  33
  34 __revision__ = "src/engine/SCons/Tool/zip.py 5023 2010/06/14 22:05:46 scons"
  35
  36 import os.path
  37
  38 import SCons.Builder
  39 import SCons.Defaults
  40 import SCons.Node.FS
  41 import SCons.Util
  42
  43 try:
  44     import zipfile
  45     internal_zip = 1
  46 except ImportError:
  47     internal_zip = 0
  48
  49 if internal_zip:
  50     zipcompression = zipfile.ZIP_DEFLATED
  51     def zip(target, source, env):
  52         compression = env.get('ZIPCOMPRESSION', 0)
  53         zf = zipfile.ZipFile(str(target[0]), 'w', compression)
  54         for s in source:
  55             if s.isdir():
  56                 for dirpath, dirnames, filenames in os.walk(str(s)):
  57                     for fname in filenames:
  58                         path = os.path.join(dirpath, fname)
  59                         if os.path.isfile(path):
  60                             zf.write(path)
  61             else:
  62                 zf.write(str(s))
  63         zf.close()
  64 else:
  65     zipcompression = 0
  66     zip = "$ZIP $ZIPFLAGS ${TARGET.abspath} $SOURCES"
  67
  68
  69 zipAction = SCons.Action.Action(zip, varlist=['ZIPCOMPRESSION'])
  70
  71 ZipBuilder = SCons.Builder.Builder(action = SCons.Action.Action('$ZIPCOM', '$ZIPCOMSTR'),
  72                                    source_factory = SCons.Node.FS.Entry,
  73                                    source_scanner = SCons.Defaults.DirScanner,
  74                                    suffix = '$ZIPSUFFIX',
  75                                    multi = 1)
  76
  77
  78 def generate(env):
  79     """Add Builders and construction variables for zip to an Environment."""
  80     try:
  81         bld = env['BUILDERS']['Zip']
  82     except KeyError:
  83         bld = ZipBuilder
  84         env['BUILDERS']['Zip'] = bld
  85
  86     env['ZIP']        = 'zip'
  87     env['ZIPFLAGS']   = SCons.Util.CLVar('')
  88     env['ZIPCOM']     = zipAction
  89     env['ZIPCOMPRESSION'] =  zipcompression
  90     env['ZIPSUFFIX']  = '.zip'
  91
  92 def exists(env):
  93     return internal_zip or env.Detect('zip')
  94
  95 # Local Variables:
  96 # tab-width:4
  97 # indent-tabs-mode:nil
  98 # End:
  99 # vim: set expandtab tabstop=4 shiftwidth=4: