80 lines
1.8 KiB
Python
Executable File
80 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# by Erik Osheim
|
|
#
|
|
# This program unboxes various archive
|
|
|
|
import os
|
|
import shutil
|
|
from subprocess import call
|
|
import sys
|
|
import tempfile
|
|
|
|
# list each kind of archive in terms of extension, and then program and
|
|
# arguments to use. Use None to indicate where the filename should be placed.
|
|
handlers = {
|
|
'.zip': ['unzip', None],
|
|
'.jar': ['unzip', None],
|
|
'.tgz': ['tar', 'xvzf', None],
|
|
'.tbz': ['tar', 'xvjf', None],
|
|
'.tar.gz': ['tar', 'xvzf', None],
|
|
'.tar.bz2': ['tar', 'xvjf', None],
|
|
#'.tar.xz': ['tar', 'xvJz', None],
|
|
}
|
|
|
|
# program usage
|
|
def usage(err=None):
|
|
retval = 0
|
|
if err:
|
|
retval = 1
|
|
print 'error: %s' % err
|
|
print 'usage: unbox ARCHIVE'
|
|
sys.exit(retval)
|
|
|
|
if __name__ == "__main__":
|
|
args = sys.argv[1:]
|
|
if not args:
|
|
usage()
|
|
elif len(args) > 1:
|
|
usage('too many arguments')
|
|
path = args[0]
|
|
if not os.path.exists(path):
|
|
usage('path %r does not exist' % path)
|
|
|
|
cmd = None
|
|
for ext in handlers:
|
|
if path.endswith(ext):
|
|
cmd = handlers[ext]
|
|
base = path[:-len(ext)]
|
|
if not cmd:
|
|
usage("can't handle path %r" % path)
|
|
|
|
d = tempfile.mkdtemp(prefix='unbox-')
|
|
|
|
fd, path2 = tempfile.mkstemp(dir=d)
|
|
os.write(fd, open(path, 'rb').read())
|
|
os.close(fd)
|
|
|
|
cwd = os.getcwd()
|
|
print d
|
|
os.chdir(d)
|
|
|
|
base2 = os.path.basename(path2)
|
|
|
|
# put the path into the command
|
|
cmd = [x or base2 for x in cmd]
|
|
|
|
retcode = call(cmd)
|
|
if retcode != 0:
|
|
print 'error: %s failed (%d)' % (cmd[0], retcode)
|
|
sys.exit(1)
|
|
|
|
os.unlink(path2)
|
|
|
|
items = os.listdir(d)
|
|
if len(items) == 1:
|
|
shutil.move(d + '/' + items[0], cwd)
|
|
shutil.rmtree(d)
|
|
else:
|
|
shutil.move(d, cwd + '/' + base)
|