| OLD | NEW |
| (Empty) |
| 1 # -*- coding: utf-8 -*- | |
| 2 # Copyright 2013 Google Inc. All Rights Reserved. | |
| 3 # | |
| 4 # Licensed under the Apache License, Version 2.0 (the "License"); | |
| 5 # you may not use this file except in compliance with the License. | |
| 6 # You may obtain a copy of the License at | |
| 7 # | |
| 8 # http://www.apache.org/licenses/LICENSE-2.0 | |
| 9 # | |
| 10 # Unless required by applicable law or agreed to in writing, software | |
| 11 # distributed under the License is distributed on an "AS IS" BASIS, | |
| 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 13 # See the License for the specific language governing permissions and | |
| 14 # limitations under the License. | |
| 15 """FilePart implementation for representing part of a file.""" | |
| 16 | |
| 17 from __future__ import absolute_import | |
| 18 | |
| 19 import os | |
| 20 | |
| 21 | |
| 22 class FilePart(file): | |
| 23 """Subclass of the file API for representing part of a file. | |
| 24 | |
| 25 This class behaves as a contiguous subset of a given file (e.g., this object | |
| 26 will behave as though the desired part of the file was written to another | |
| 27 file, and the second file was opened). | |
| 28 """ | |
| 29 | |
| 30 # pylint: disable=super-init-not-called | |
| 31 def __init__(self, filename, offset, length): | |
| 32 """Initializes the FilePart. | |
| 33 | |
| 34 Args: | |
| 35 filename: The name of the existing file, of which this object represents | |
| 36 a part. | |
| 37 offset: The position (in bytes) in the original file that corresponds to | |
| 38 the first byte of the FilePart. | |
| 39 length: The total number of bytes in the FilePart. | |
| 40 """ | |
| 41 self._fp = open(filename, 'rb') | |
| 42 self.length = length | |
| 43 self._start = offset | |
| 44 self._end = self._start + self.length | |
| 45 self._fp.seek(self._start) | |
| 46 | |
| 47 def __enter__(self): | |
| 48 pass | |
| 49 | |
| 50 # pylint: disable=redefined-builtin | |
| 51 def __exit__(self, type, value, traceback): | |
| 52 self.close() | |
| 53 | |
| 54 def tell(self): | |
| 55 return self._fp.tell() - self._start | |
| 56 | |
| 57 def read(self, size=-1): | |
| 58 if size < 0: | |
| 59 size = self.length | |
| 60 size = min(size, self._end - self._fp.tell()) # Only read to our EOF | |
| 61 return self._fp.read(max(0, size)) | |
| 62 | |
| 63 def seek(self, offset, whence=os.SEEK_SET): | |
| 64 if whence == os.SEEK_END: | |
| 65 return self._fp.seek(offset + self._end) | |
| 66 elif whence == os.SEEK_CUR: | |
| 67 return self._fp.seek(offset, whence) | |
| 68 else: | |
| 69 return self._fp.seek(self._start + offset) | |
| 70 | |
| 71 def close(self): | |
| 72 self._fp.close() | |
| 73 | |
| 74 def flush(self, size=None): | |
| 75 raise NotImplementedError('flush is not implemented in FilePart.') | |
| 76 | |
| 77 def fileno(self, size=None): | |
| 78 raise NotImplementedError('fileno is not implemented in FilePart.') | |
| 79 | |
| 80 def isatty(self, size=None): | |
| 81 raise NotImplementedError('isatty is not implemented in FilePart.') | |
| 82 | |
| 83 def next(self, size=None): | |
| 84 raise NotImplementedError('next is not implemented in FilePart.') | |
| 85 | |
| 86 def readline(self, size=None): | |
| 87 raise NotImplementedError('readline is not implemented in FilePart.') | |
| 88 | |
| 89 def readlines(self, size=None): | |
| 90 raise NotImplementedError('readlines is not implemented in FilePart.') | |
| 91 | |
| 92 def xreadlines(self, size=None): | |
| 93 raise NotImplementedError('xreadlines is not implemented in FilePart.') | |
| 94 | |
| 95 def truncate(self, size=None): | |
| 96 raise NotImplementedError('truncate is not implemented in FilePart.') | |
| 97 | |
| 98 def write(self, size=None): | |
| 99 raise NotImplementedError('write is not implemented in FilePart.') | |
| 100 | |
| 101 def writelines(self, size=None): | |
| 102 raise NotImplementedError('writelines is not implemented in FilePart.') | |
| OLD | NEW |