虚拟环境脚本

Author: huzhonglan

create_venv_script.py

@@ -0,0 +1,49 @@
+# coding=utf-8
+
+"""
+virtualenv定制脚本接受3个扩展函数
+extend_parser(optparse_parser):添加额外的选项
+adjust_options(optioins, args): 改变当前的选项
+after_install: 在默认的环境安装好之后，执行其他工作，通过这个函数定制
+"""
+import os
+import logging
+import subprocess
+import virtualenv
+
+
+virtualenv_path = subprocess.check_output(['which', 'virtualenv']).strip()
+EXPORT_TEXT = ''
+ROOT_PATH = '/home/python/venv'
+
+def extend_parse(parser):
+    parser.add_option('-r', '-req', action='append', type='string', dest='reqs',
+                      help="specify additional required packages", default=[])
+
+def adjust_options(options, args):
+    if not args:
+        return
+    base_dir = args[0]
+    args[0] = os.path.join(ROOT_PATH, base_dir)
+
+def after_install(options, home_dir):
+    if not options.reqs:
+        logging.warning('Warn:You maybe need specify some required packages!')
+    for req in options.reqs:
+        subprocess.call(['{} /bin/pip'.format(home_dir), 'install', req])
+
+def main():
+    text = virtualenv.create_bootstrap_script(EXPORT_TEXT,python_version='3.5')
+    print('%s' % virtualenv_path)
+    with open(virtualenv_path, 'w') as f:
+        f.write(text)
+
+
+if __name__ == '__main__':
+    main()
+
+
+"""
+创建一个自动安装了Django, Flask的虚拟环境
+>virtualenv -r django -r flask
+"""