Pillow/PIL/ContainerIO.py

118 lines
2.7 KiB
Python
Raw Normal View History

2010-07-31 06:52:47 +04:00
#
# The Python Imaging Library.
# $Id$
#
# a class to read from a container file
#
# History:
# 1995-06-18 fl Created
# 1995-09-07 fl Added readline(), readlines()
#
# Copyright (c) 1997-2001 by Secret Labs AB
# Copyright (c) 1995 by Fredrik Lundh
#
# See the README file for information on usage and redistribution.
#
##
# A file object that provides read access to a part of an existing
# file (for example a TAR file).
2014-08-26 17:47:10 +04:00
2015-05-26 17:07:21 +03:00
class ContainerIO(object):
2010-07-31 06:52:47 +04:00
def __init__(self, file, offset, length):
2016-09-24 12:10:46 +03:00
"""
Create file object.
:param file: Existing file.
:param offset: Start of region, in bytes.
:param length: Size of region, in bytes.
"""
2010-07-31 06:52:47 +04:00
self.fh = file
self.pos = 0
self.offset = offset
self.length = length
self.fh.seek(offset)
##
# Always false.
def isatty(self):
return 0
2014-08-26 17:47:10 +04:00
def seek(self, offset, mode=0):
2016-09-24 12:10:46 +03:00
"""
Move file pointer.
:param offset: Offset in bytes.
:param mode: Starting position. Use 0 for beginning of region, 1
for current offset, and 2 for end of region. You cannot move
the pointer outside the defined region.
"""
2010-07-31 06:52:47 +04:00
if mode == 1:
self.pos = self.pos + offset
elif mode == 2:
self.pos = self.length + offset
else:
self.pos = offset
# clamp
self.pos = max(0, min(self.pos, self.length))
self.fh.seek(self.offset + self.pos)
def tell(self):
2016-09-24 12:10:46 +03:00
"""
Get current file pointer.
2010-07-31 06:52:47 +04:00
2016-09-24 12:10:46 +03:00
:returns: Offset from start of region, in bytes.
"""
return self.pos
2010-07-31 06:52:47 +04:00
2014-08-26 17:47:10 +04:00
def read(self, n=0):
2016-09-24 12:10:46 +03:00
"""
Read data.
@def read(bytes=0)
:param bytes: Number of bytes to read. If omitted or zero,
read until end of region.
:returns: An 8-bit string.
"""
2010-07-31 06:52:47 +04:00
if n:
n = min(n, self.length - self.pos)
else:
n = self.length - self.pos
2014-08-26 17:47:10 +04:00
if not n: # EOF
2010-07-31 06:52:47 +04:00
return ""
self.pos = self.pos + n
return self.fh.read(n)
def readline(self):
2016-09-24 12:10:46 +03:00
"""
Read a line of text.
:returns: An 8-bit string.
"""
2010-07-31 06:52:47 +04:00
s = ""
while True:
2010-07-31 06:52:47 +04:00
c = self.read(1)
if not c:
break
s = s + c
if c == "\n":
break
return s
def readlines(self):
2016-09-24 12:10:46 +03:00
"""
Read multiple lines of text.
:returns: A list of 8-bit strings.
"""
2010-07-31 06:52:47 +04:00
l = []
while True:
2010-07-31 06:52:47 +04:00
s = self.readline()
if not s:
break
l.append(s)
return l