Spaces:
Runtime error
Runtime error
File size: 926 Bytes
c310e19 |
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 |
#!/usr/bin/env python3
import os
import shlex
import shutil
import subprocess
def extract_archive(dataset_archive, tmp_data_path):
if not os.path.isfile(dataset_archive):
return False
dataset_ext = os.path.splitext(dataset_archive)[1]
if dataset_ext != ".gz" and dataset_ext != ".tar":
return False
if os.path.isdir(tmp_data_path):
shutil.rmtree(tmp_data_path, ignore_errors=True)
os.makedirs(tmp_data_path)
if dataset_ext == ".gz":
tar_opt = "-xzf"
else:
tar_opt = "-xf"
extract_cmd = ("tar {} {} -C {}").format(tar_opt, dataset_archive, tmp_data_path)
subprocess.call(shlex.split(extract_cmd))
return True
def tar_file(tar_path, tmp_path):
tar_name = tar_path.split('/')[-1]
if extract_archive(tar_path, tmp_path):
print('extract ' + tar_name + 'successfully!')
else:
print("fail to extract " + tar_name)
|