Home | History | Annotate | Download | only in system 
      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())
     65