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
|
import os
import sys
import tarfile
def is_exe(fpath):
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
def main(args):
if len(args) < 2 or len(args) > 3:
raise Exception(
"Illegal usage: `tar_directory.py archive.tar directory [skip prefix]` or `tar_directory.py archive.tar output_directory --extract`"
)
tar, directory, prefix, extract = args[0], args[1], None, False
if len(args) == 3:
if args[2] == '--extract':
extract = True
else:
prefix = args[2]
for tar_exe in ('/usr/bin/tar', '/bin/tar'):
if not is_exe(tar_exe):
continue
if extract:
dest = os.path.abspath(directory)
if not os.path.exists(dest):
os.makedirs(dest)
os.execv(tar_exe, [tar_exe, '-xf', tar, '-C', dest])
else:
source = os.path.relpath(directory, prefix) if prefix else directory
os.execv(tar_exe, [tar_exe, '-cf', tar] + (['-C', prefix] if prefix else []) + [source])
break
else:
if extract:
dest = os.path.abspath(directory)
if not os.path.exists(dest):
os.makedirs(dest)
with tarfile.open(tar, 'r') as tar_file:
tar_file.extractall(dest)
else:
source = directory
with tarfile.open(tar, 'w') as out:
out.add(os.path.abspath(source), arcname=os.path.relpath(source, prefix) if prefix else source)
if __name__ == '__main__':
main(sys.argv[1:])
|
