Add missing venv module

michaeljoseph · Jul 6, 2014 · ca8b607 · ca8b607
1 parent 742c723
commit ca8b607
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/changes/venv.py b/changes/venv.py
@@ -0,0 +1,20 @@
+import os
+import tempfile
+
+import virtualenv
+
+
+def create_venv(tmp_dir=None):
+    if not tmp_dir:
+        tmp_dir = tempfile.mkdtemp()
+    virtualenv.create_environment(tmp_dir, site_packages=False)
+    return tmp_dir
+
+
+def install(package_name, venv_dir):
+    if not os.path.exists(venv_dir):
+        venv_dir = create_venv()
+    virtualenv.install_wheel(
+        [package_name],
+        '%s/bin/python' % venv_dir,
+    )