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 |