src/third_party/WebKit/Tools/Scripts/webkitpy/common/system/fileset.py

   1 # Copyright (C) 2010 Google Inc. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or without
   4 # modification, are permitted provided that the following conditions
   5 # are met:
   6 #
   7 # 1.  Redistributions of source code must retain the above copyright
   8 #     notice, this list of conditions and the following disclaimer.
   9 # 2.  Redistributions in binary form must reproduce the above copyright
  10 #     notice, this list of conditions and the following disclaimer in the
  11 #     documentation and/or other materials provided with the distribution.
  12 #
  13 # THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  14 # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  15 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  16 # DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  17 # DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  18 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  19 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  20 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  21 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  22 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  23
  24 from webkitpy.common.system.filesystem import FileSystem
  25
  26
  27 class FileSetFileHandle(object):
  28     """Points to a file that resides in a file set"""
  29     def __init__(self, fileset, filename, filesystem=None):
  30         self._filename = filename
  31         self._fileset = fileset
  32         self._contents = None
  33         self._filesystem = filesystem or FileSystem()
  34
  35     def __str__(self):
  36         return "%s:%s" % (self._fileset, self._filename)
  37
  38     def close(self):
  39         pass
  40
  41     def contents(self):
  42         if self._contents is None:
  43             self._contents = self._fileset.read(self._filename)
  44         return self._contents
  45
  46     def save_to(self, path, filename=None):
  47         if filename is None:
  48             self._fileset.extract(self._filename, path)
  49             return
  50         with self._filesystem.mkdtemp() as temp_dir:
  51             self._fileset.extract(self._filename, temp_dir)
  52
  53             src = self._filesystem.join(temp_dir, self._filename)
  54             dest = self._filesystem.join(path, filename)
  55             self._filesystem.copyfile(src, dest)
  56
  57     def delete(self):
  58         self._fileset.delete(self._filename)
  59
  60     def name(self):
  61         return self._filename
  62
  63     def splitext(self):
  64         return self._filesystem.splitext(self.name())