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 __future__ import with_statement
     25 
     26 from webkitpy.common.system.filesystem import FileSystem
     27 
     28 
     29 class FileSetFileHandle(object):
     30     """Points to a file that resides in a file set"""
     31     def __init__(self, fileset, filename, filesystem=None):
     32         self._filename = filename
     33         self._fileset = fileset
     34         self._contents = None
     35         self._filesystem = filesystem or FileSystem()
     36 
     37     def __str__(self):
     38         return "%s:%s" % (self._fileset, self._filename)
     39 
     40     def close(self):
     41         pass
     42 
     43     def contents(self):
     44         if self._contents is None:
     45             self._contents = self._fileset.read(self._filename)
     46         return self._contents
     47 
     48     def save_to(self, path, filename=None):
     49         if filename is None:
     50             self._fileset.extract(self._filename, path)
     51             return
     52         with self._filesystem.mkdtemp() as temp_dir:
     53             self._fileset.extract(self._filename, temp_dir)
     54 
     55             src = self._filesystem.join(temp_dir, self._filename)
     56             dest = self._filesystem.join(path, filename)
     57             self._filesystem.copyfile(src, dest)
     58 
     59     def delete(self):
     60         self._fileset.delete(self._filename)
     61 
     62     def name(self):
     63         return self._filename
     64 
     65     def splitext(self):
     66         return self._filesystem.splitext(self.name())
     67