# Copyright 2010 Baptiste Lepilleur and The JsonCpp Authors

# Distributed under MIT license, or public domain if desired and

# recognized in your jurisdiction.

# See file LICENSE for detail or copy at http://jsoncpp.sourceforge.net/LICENSE

from contextlib import closing

import os

import tarfile

TARGZ_DEFAULT_COMPRESSION_LEVEL = 9

def make_tarball(tarball_path, sources, base_dir, prefix_dir=''):

"""Parameters:

tarball_path: output path of the .tar.gz file

sources: list of sources to include in the tarball, relative to the current directory

base_dir: if a source file is in a sub-directory of base_dir, then base_dir is stripped

from path in the tarball.

prefix_dir: all files stored in the tarball be sub-directory of prefix_dir. Set to ''

to make them child of root.

"""

base_dir = os.path.normpath(os.path.abspath(base_dir))

def archive_name(path):

"""Makes path relative to base_dir."""

path = os.path.normpath(os.path.abspath(path))

common_path = os.path.commonprefix((base_dir, path))

archive_name = path[len(common_path):]

if os.path.isabs(archive_name):

archive_name = archive_name[1:]

return os.path.join(prefix_dir, archive_name)

def visit(tar, dirname, names):

for name in names:

path = os.path.join(dirname, name)

if os.path.isfile(path):

path_in_tar = archive_name(path)

tar.add(path, path_in_tar)

compression = TARGZ_DEFAULT_COMPRESSION_LEVEL

with closing(tarfile.TarFile.open(tarball_path, 'w:gz',

compresslevel=compression)) as tar:

for source in sources:

source_path = source

if os.path.isdir(source):

for dirpath, dirnames, filenames in os.walk(source_path):

visit(tar, dirpath, filenames)

else:

path_in_tar = archive_name(source_path)

tar.add(source_path, path_in_tar) # filename, arcname

def decompress(tarball_path, base_dir):

"""Decompress the gzipped tarball into directory base_dir.

"""

with closing(tarfile.TarFile.open(tarball_path)) as tar:

tar.extractall(base_dir)