cmd/pip_download.py

55 lines
1.7 KiB
Python
Raw Normal View History

2019-06-27 16:40:09 +00:00
'''
Wraps `pip download` so that the resulting files, including dependencies,
are automatically organized into a folder called .\package\package-version.
'''
import argparse
import os
import subprocess
import sys
import tempfile
def clean_version(version):
for (index, character) in enumerate(version):
if character not in '.0123456789':
break
else:
return version
return version[:index]
def pip_download(package):
tmpdir = tempfile.TemporaryDirectory(prefix=f'pip_download-{package}')
subprocess.call([sys.executable, '-m', 'pip', 'download', package, '-d', tmpdir.name])
downloaded_files = os.listdir(tmpdir.name)
for filename in downloaded_files:
parts = filename.split('-')
filename_package = parts[0]
if filename_package.lower() == package.lower():
break
else:
raise Exception(f'None of the downloads match the package name {package}? {downloaded_files}')
version = parts[1]
version = clean_version(version)
new_directory = f'{package}\\{package}-{version}'
if not os.path.exists(new_directory):
os.makedirs(new_directory)
for filename in downloaded_files:
os.rename(os.path.join(tmpdir.name, filename), os.path.join(new_directory, filename))
tmpdir.cleanup()
def pip_download_argparse(args):
for package in args.packages:
pip_download(package)
def main(argv):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('packages', nargs='+')
parser.set_defaults(func=pip_download_argparse)
args = parser.parse_args(argv)
args.func(args)
if __name__ == '__main__':
raise SystemExit(main(sys.argv[1:]))