From 01ff50234b1c7ef1f5157e4c56fdb9c7ab327228 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Thu, 27 Jun 2019 09:40:09 -0700
Subject: [PATCH] Add pip_download.

---
 pip_download.py | 55 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 pip_download.py

diff --git a/pip_download.py b/pip_download.py
new file mode 100644
index 0000000..36318f1
--- /dev/null
+++ b/pip_download.py
@@ -0,0 +1,55 @@
+'''
+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])
+    input()
+    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:]))