1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
"""
Tar file support
"""
#*****************************************************************************
# Copyright (C) 2016 Volker Braun <vbraun.name@gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
# http://www.gnu.org/licenses/
#*****************************************************************************
from __future__ import print_function
import os
import copy
import tarfile
import stat
from sage_bootstrap.uncompress.filter_os_files import filter_os_files
class SageTarFile(tarfile.TarFile):
"""
Sage as tarfile.TarFile, but applies the user's current umask to the
permissions of all extracted files and directories.
This mimics the default behavior of the ``tar`` utility.
See http://trac.sagemath.org/ticket/20218#comment:16 for more background.
"""
def __new__(cls, *args, **kwargs):
# This is is that SageTarFile() is equivalent to TarFile.open() which
# is more flexible than the basic TarFile.__init__
inst = tarfile.TarFile.open(*args, **kwargs)
inst.__class__ = cls
return inst
def __init__(self, *args, **kwargs):
# Unfortunately the only way to get the current umask is to set it
# and then restore it
self.umask = os.umask(0o777)
os.umask(self.umask)
@classmethod
def can_read(cls, filename):
"""
Given an archive filename, returns True if this class can read and
process the archive format of that file.
"""
return tarfile.is_tarfile(filename)
@property
def names(self):
"""
List of filenames in the archive.
Filters out names of OS-related files that shouldn't be in the
archive (.DS_Store, etc.)
"""
return filter_os_files(self.getnames())
def chmod(self, tarinfo, target):
tarinfo = copy.copy(tarinfo)
tarinfo.mode &= ~self.umask
tarinfo.mode &= ~(stat.S_ISUID | stat.S_ISGID)
return super(SageTarFile, self).chmod(tarinfo, target)
def extractall(self, path='.', members=None):
"""
Same as tarfile.TarFile.extractall but allows filenames for
the members argument (like zipfile.ZipFile).
"""
if members:
name_to_member = dict([member.name, member] for member in self.getmembers())
members = [m if isinstance(m, tarfile.TarInfo)
else name_to_member[m]
for m in members]
return super(SageTarFile, self).extractall(path=path, members=members)
def extractbytes(self, member):
"""
Return the contents of the specified archive member as bytes.
If the member does not exist, returns None.
"""
if member in self.getnames():
reader = self.extractfile(member)
return reader.read()
|
