Source

virtualenv / bin / refresh-support-files.py

#!/usr/bin/env python
"""
Refresh any files in ../virtualenv_support/ that come from elsewhere
"""

import os
import urllib
import sys

here = os.path.dirname(__file__)
support_files = os.path.join(here, '..', 'virtualenv_support')

files = [
    ('http://peak.telecommunity.com/dist/ez_setup.py', 'ez_setup.py'),
    ('http://pypi.python.org/packages/2.6/s/setuptools/setuptools-0.6c11-py2.6.egg', 'setuptools-0.6c11-py2.6.egg'),
    ('http://pypi.python.org/packages/2.5/s/setuptools/setuptools-0.6c11-py2.5.egg', 'setuptools-0.6c11-py2.5.egg'),
    ('http://pypi.python.org/packages/2.4/s/setuptools/setuptools-0.6c11-py2.4.egg', 'setuptools-0.6c11-py2.4.egg'),
    ('http://python-distribute.org/distribute_setup.py', 'distribute_setup.py'),
    ('http://pypi.python.org/packages/source/d/distribute/distribute-0.6.14.tar.gz', 'distribute-0.6.14.tar.gz'),
    ('http://pypi.python.org/packages/source/p/pip/pip-0.7.2.tar.gz', 'pip-0.7.2.tar.gz'),
    ]

def main():
    for url, filename in files:
        print 'fetching', url, '...',
        sys.stdout.flush()
        f = urllib.urlopen(url)
        content = f.read()
        f.close()
        print 'done.'
        filename = os.path.join(support_files, filename)
        if os.path.exists(filename):
            f = open(filename, 'rb')
            cur_content = f.read()
            f.close()
        else:
            cur_content = ''
        if cur_content == content:
            print '  %s up-to-date' % filename
        else:
            print '  overwriting %s' % filename
            f = open(filename, 'wb')
            f.write(content)
            f.close()

if __name__ == '__main__':
    main()