| | |
| | import sys |
| | import os.path |
| | import logging |
| | import tempfile |
| | import shutil |
| |
|
| | import setuptools.archive_util |
| |
|
| |
|
| | logger = logging.getLogger(__name__) |
| |
|
| |
|
| | def main(): |
| | logging.basicConfig() |
| | logger.setLevel(logging.DEBUG) |
| |
|
| | src = sys.argv[1] |
| | dst = sys.argv[2] |
| |
|
| | strip_toplevel_dir = True |
| |
|
| | if not os.path.exists(dst): |
| | os.makedirs(dst) |
| |
|
| | if not os.path.isdir(dst): |
| | logger.error('%s: not a directory', dst) |
| |
|
| | if strip_toplevel_dir: |
| | tempdir = tempfile.mkdtemp() |
| | try: |
| | setuptools.archive_util.unpack_archive(src, tempdir) |
| | toplevel_items = os.listdir(tempdir) |
| | if len(toplevel_items) > 1: |
| | logger.error('%s has no single top-level directory', src) |
| | raise SystemExit(1) |
| | root = os.path.join(tempdir, toplevel_items[0]) |
| | for item in os.listdir(root): |
| | src_item = os.path.join(root, item) |
| | dst_item = os.path.join(dst, item) |
| | if os.path.exists(dst_item): |
| | if os.path.isdir(dst_item): |
| | shutil.rmtree(dst_item) |
| | else: |
| | os.unlink(dst_item) |
| | shutil.move(src_item, dst) |
| | finally: |
| | shutil.rmtree(tempdir) |
| | else: |
| | setuptools.archive_util.unpack_archive(src, dst) |
| |
|
