sbin/create_python_env.py
author ymh <ymh.work@gmail.com>
Tue, 01 Jun 2010 19:07:23 +0200
changeset 35 8b65c9054eac
parent 34 05d10640919a
child 36 bbe6f42d42b2
permissions -rw-r--r--
correct errors
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
"""
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
     2
Call this like ``python create_python_env.py``; it will
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     3
refresh the blinkster-boot.py script
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     4
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     5
-prerequisite:
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     6
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     7
- virtualenv
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     8
- distribute
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
     9
- psycopg2 requires the PostgreSQL libpq libraries and the pg_config utility
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    10
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    11
- virtualenv --distribute --no-site-packages venv
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    12
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    13
- python blinkster-boot.py --distribute --no-site-packages --index-url=http://pypi.websushi.org/ --clear bvenv
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    14
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    15
"""
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    16
import os
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    17
import subprocess
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    18
import re
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    19
import sys
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    20
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    21
here = os.path.dirname(os.path.abspath(__file__))
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    22
base_dir = here
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    23
script_name = os.path.join(base_dir, 'blinkster-boot.py')
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    24
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    25
import virtualenv
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    26
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    27
# things to install
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    28
# - psycopg2 -> pip
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    29
# - PIL -> pip
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    30
# - pyxml -> pip
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    31
# - 4Suite-xml - easy_install ftp://ftp.4suite.org/pub/4Suite/4Suite-XML-1.0.2.tar.bz2
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    32
# - pylucene  - script
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    33
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    34
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    35
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    36
EXTRA_TEXT = """
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    37
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    38
import shutil
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    39
import tarfile
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    40
import urllib
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    41
import platform
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    42
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    43
PSYCOPG2_URL = 'http://initd.org/pub/software/psycopg/psycopg2-2.2.1.tar.gz'
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    44
FOURSUITE_XML_URL = 'ftp://ftp.4suite.org/pub/4Suite/4Suite-XML-1.0.2.tar.bz2'
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    45
PY_LUCENE_URL = 'http://apache.crihan.fr/dist/lucene/pylucene/pylucene-3.0.1-1-src.tar.gz'
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    46
PIL_URL = 'http://effbot.org/downloads/Imaging-1.1.7.tar.gz'
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    47
INDEX_URL = 'http://pypi.python.org/simple/'
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    48
PYXML_URL = "http://sourceforge.net/projects/pyxml/files/pyxml/0.8.4/PyXML-0.8.4.tar.gz/download"
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    49
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    50
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    51
def extend_parser(parser):
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    52
    parser.add_option(
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    53
        '--index-url',
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    54
        metavar='INDEX_URL',
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    55
        dest='index_url',
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    56
        default='',
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    57
        help='base URL of Python Package Index')
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    58
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    59
def adjust_options(options, args):
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    60
    pass
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    61
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    62
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    63
def after_install(options, home_dir):
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    64
    base_dir = os.path.dirname(home_dir)
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    65
    src_dir = join(home_dir, 'src')
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    66
    tmp_dir = join(home_dir, 'tmp')
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    67
    ensure_dir(src_dir)
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    68
    ensure_dir(tmp_dir)
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    69
    logger.indent += 2
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    70
    try:
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    71
        call_subprocess([os.path.abspath(os.path.join(home_dir, 'bin', 'pip')), 'install', '-E', os.path.abspath(home_dir), PSYCOPG2_URL],
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    72
                        cwd=os.path.abspath(tmp_dir),
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    73
                        filter_stdout=filter_python_develop,
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    74
                        show_stdout=True)
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    75
        call_subprocess([os.path.abspath(os.path.join(home_dir, 'bin', 'pip')), 'install', '-E', os.path.abspath(home_dir), PIL_URL],
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    76
                        cwd=os.path.abspath(tmp_dir),
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    77
                        filter_stdout=filter_python_develop,
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    78
                        show_stdout=True)
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    79
        call_subprocess([os.path.abspath(os.path.join(home_dir, 'bin', 'pip')), 'install', '-E', os.path.abspath(home_dir), PYXML_URL],
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    80
                        cwd=os.path.abspath(tmp_dir),
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    81
                        filter_stdout=filter_python_develop,
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    82
                        show_stdout=True)
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    83
        call_subprocess([os.path.abspath(os.path.join(home_dir, 'bin', 'easy_install')), FOURSUITE_XML_URL],
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    84
                        cwd=os.path.abspath(tmp_dir),
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
    85
                        filter_stdout=filter_python_develop,
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
    86
                        show_stdout=True)
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
    87
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    88
        #get pylucene
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    89
        pylucene_src = os.path.join(src_dir,"pylucene.tar.gz")
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    90
        urllib.urlretrieve(PY_LUCENE_URL, pylucene_src)
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    91
        tf = tarfile.open(pylucene_src,'r:gz')
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    92
        pylucene_base_path = os.path.join(src_dir,"pylucene") 
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    93
        tf.extractall(pylucene_base_path)
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    94
        tf.close()
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    95
        
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    96
        pylucene_src_path = os.path.join(pylucene_base_path, os.listdir(pylucene_base_path)[0])
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
    97
        
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    98
        #install jcc
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
    99
        call_subprocess([os.path.abspath(os.path.join(home_dir, 'bin', 'python')), 'setup.py', 'install'],
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   100
                        cwd=os.path.abspath(os.path.join(pylucene_src_path,"jcc")),
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   101
                        filter_stdout=filter_python_develop,
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   102
                        show_stdout=True)
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   103
        #install pylucene       
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   104
        #modify makefile
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   105
        makefile_path = os.path.join(pylucene_src_path,"Makefile")
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   106
        shutil.move( makefile_path, makefile_path+"~" )
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   107
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   108
        destination= open( makefile_path, "w" )
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   109
        source= open( makefile_path+"~", "r" )
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   110
        destination.write("PREFIX_PYTHON="+os.path.abspath(home_dir)+"\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   111
        destination.write("ANT=ant\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   112
        destination.write("PYTHON=$(PREFIX_PYTHON)/bin/python\\n")
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   113
        system_str = platform.system()
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   114
        if system_str == "Darwin":
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   115
            if sys.version_info > (2,6):
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   116
                destination.write("JCC=$(PYTHON) -m jcc.__main__ --shared --arch x86_64 --arch i386\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   117
            else:
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   118
                destination.write("JCC=$(PYTHON) -m jcc --shared --arch x86_64 --arch i386\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   119
            destination.write("NUM_FILES=2\\n")
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   120
        elif system_str == "Windows":
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   121
            destination.write("JCC=$(PYTHON) -m jcc.__main__ --shared --arch x86_64 --arch i386\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   122
            destination.write("NUM_FILES=2\\n")
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   123
        else:
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   124
            destination.write("JCC=$(PYTHON) -m jcc --shared\\n")
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   125
            destination.write("NUM_FILES=2\\n")
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   126
        for line in source:
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   127
            destination.write( line )
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   128
        source.close()
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   129
        destination.close()
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   130
        os.remove(makefile_path+"~" )
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   131
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   132
        call_subprocess(['make', 'install'],
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   133
                        cwd=os.path.abspath(pylucene_src_path),
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   134
                        filter_stdout=filter_python_develop,
35
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   135
                        show_stdout=True)
8b65c9054eac correct errors
ymh <ymh.work@gmail.com>
parents: 34
diff changeset
   136
        shutil.rmtree(src_dir)
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   137
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   138
    finally:
34
05d10640919a improve virtual env script
ymh <ymh.work@gmail.com>
parents: 33
diff changeset
   139
        logger.indent -= 2
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   140
    script_dir = join(base_dir, 'bin')
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   141
    logger.notify('Run "%s Package" to install new packages that provide builds'
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
   142
                  % join(script_dir, 'easy_install'))
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   143
32
e28089cee66c update python env script
ymh <ymh.work@gmail.com>
parents: 31
diff changeset
   144
def ensure_dir(dir):
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   145
    if not os.path.exists(dir):
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   146
        logger.info('Creating directory %s' % dir)
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   147
        os.makedirs(dir)
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   148
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   149
def filter_python_develop(line):
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   150
    if not line.strip():
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   151
        return Logger.DEBUG
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   152
    for prefix in ['Searching for', 'Reading ', 'Best match: ', 'Processing ',
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   153
                   'Moving ', 'Adding ', 'running ', 'writing ', 'Creating ',
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   154
                   'creating ', 'Copying ']:
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   155
        if line.startswith(prefix):
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   156
            return Logger.DEBUG
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   157
    return Logger.NOTIFY
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   158
"""
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   159
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   160
def main():
33
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   161
    python_version = ".".join(map(str,sys.version_info[0:2]))
b25bb1dd177c improve virtualenv script
ymh <ymh.work@gmail.com>
parents: 32
diff changeset
   162
    text = virtualenv.create_bootstrap_script(EXTRA_TEXT, python_version=python_version)
31
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   163
    if os.path.exists(script_name):
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   164
        f = open(script_name)
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   165
        cur_text = f.read()
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   166
        f.close()
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   167
    else:
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   168
        cur_text = ''
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   169
    print 'Updating %s' % script_name
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   170
    if cur_text == 'text':
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   171
        print 'No update'
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   172
    else:
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   173
        print 'Script changed; updating...'
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   174
        f = open(script_name, 'w')
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   175
        f.write(text)
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   176
        f.close()
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   177
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   178
if __name__ == '__main__':
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   179
    main()
a727d83c8f88 add create_python_env file
ymh <ymh.work@gmail.com>
parents:
diff changeset
   180