openmedialibrary_platform_d.../lib/python3.7/site-packages/PIL/TarIO.py

72 lines
1.5 KiB
Python
Raw Permalink Normal View History

2016-02-06 09:36:57 +00:00
#
# The Python Imaging Library.
# $Id$
#
# read files from within a tar file
#
# History:
# 95-06-18 fl Created
# 96-05-28 fl Open files in binary mode
#
# Copyright (c) Secret Labs AB 1997.
# Copyright (c) Fredrik Lundh 1995-96.
#
# See the README file for information on usage and redistribution.
#
2019-01-10 09:03:55 +00:00
import sys
2018-12-31 23:25:26 +00:00
from . import ContainerIO
2016-02-06 09:36:57 +00:00
##
# A file object that provides read access to a given member of a TAR
# file.
class TarIO(ContainerIO.ContainerIO):
def __init__(self, tarfile, file):
2018-12-31 23:25:26 +00:00
"""
Create file object.
2016-02-06 09:36:57 +00:00
2018-12-31 23:25:26 +00:00
:param tarfile: Name of TAR file.
:param file: Name of member file.
"""
2019-01-10 09:03:55 +00:00
self.fh = open(tarfile, "rb")
2016-02-06 09:36:57 +00:00
while True:
2019-01-10 09:03:55 +00:00
s = self.fh.read(512)
2016-02-06 09:36:57 +00:00
if len(s) != 512:
raise IOError("unexpected end of tar file")
name = s[:100].decode('utf-8')
i = name.find('\0')
if i == 0:
raise IOError("cannot find subfile")
if i > 0:
name = name[:i]
size = int(s[124:135], 8)
if file == name:
break
2019-01-10 09:03:55 +00:00
self.fh.seek((size + 511) & (~511), 1)
2016-02-06 09:36:57 +00:00
# Open region
2019-01-10 09:03:55 +00:00
ContainerIO.ContainerIO.__init__(self, self.fh, self.fh.tell(), size)
# Context manager support
def __enter__(self):
return self
def __exit__(self, *args):
self.close()
if sys.version_info.major >= 3:
def __del__(self):
self.close()
def close(self):
self.fh.close()