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
|
#!/usr/bin/env python3
import argparse
import os
import subprocess
import sys
import tarfile
def is_exe(fpath):
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
def parse_args():
parser = argparse.ArgumentParser(description='Extract a tarball and remove it upon successful extraction.')
parser.add_argument('--input', required=True, help='Path to the input tarball')
parser.add_argument('--output', required=True, help='Path to the output directory')
return parser.parse_args()
def unpack_dir(tared_dir, dest_path):
tared_dir = os.path.abspath(tared_dir)
if not os.path.exists(dest_path):
os.makedirs(dest_path)
for tar_exe in ('/usr/bin/tar', '/bin/tar'):
if is_exe(tar_exe):
tar_cmd = [tar_exe, '--extract', '--file', tared_dir, '-C', dest_path, '--no-same-owner']
if sys.platform == 'linux':
tar_cmd.append('--delay-directory-restore')
subprocess.run(tar_cmd, check=True, text=True)
break
else:
with tarfile.open(tared_dir, 'r') as tar_file:
tar_file.extractall(dest_path)
def main():
args = parse_args()
# Validate input file
if not os.path.isfile(args.input):
sys.exit(f"Input file does not exist: {args.input}")
# Create output directory (if it doesn't exist)
try:
os.makedirs(args.output, exist_ok=True)
except OSError as e:
sys.exit(f"Failed to create output directory: {e}")
unpack_dir(args.input, args.output)
# Remove input file only if extraction was successful
try:
os.remove(args.input)
except OSError as e:
sys.exit(f"Failed to remove input file: {e}")
if __name__ == '__main__':
main()
|
