bootstrap.py

#!/usr/bin/env python
# bootstrap.py
# Bootstrap and setup a virtualenv with the specified requirements.txt
import argparse
import os
import stat
import sys
from shutil import move
from subprocess import call
from tempfile import mkstemp


description = """
Set up my development environment for me!
"""

project_name = 'modelbouwdag'

parser = argparse.ArgumentParser(description=description)
parser.add_argument('target', choices=['production', 'staging', 'test', 'dev'],
                    help='production/staging/dev')
parser.add_argument('--project', default=project_name,
                    help='Name of the project in your src directory, "%s" by default' % project_name)
parser.add_argument('--env', default='env',
                    help='Directory name for virtualenv, "env" by default')

args = parser.parse_args()


def replace_or_append(file_path, search_val, replace_val):
    file_handle, abs_path = mkstemp()
    new_file = open(abs_path, 'w')
    old_file = open(file_path, 'r')
    found = False
    for line in old_file:
        if line.startswith(search_val):
            new_file.write(replace_val)
            found = True
        else:
            new_file.write(line)
    if not found:
        new_file.write("\n" + replace_val)
    new_file.close()
    os.close(file_handle)
    old_file.close()
    os.remove(file_path)
    move(abs_path, file_path)
    os.chmod(file_path, 436)


def replace_wsgi_settings(target):
    path = os.path.join('src', project_name, 'wsgi.py')
    replace_or_append(
        path, 'os.environ.setdefault',
        'os.environ.setdefault("DJANGO_SETTINGS_MODULE", "%s.conf.%s")\n' % (project_name, target)
    )


def replace_manage_settings(target):
    path = os.path.join('src', project_name, 'manage.py')
    replace_or_append(
        path, '    os.environ.setdefault',
        '    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "%s.conf.%s")\n' % (project_name, target)
    )


def append_settings_activate(project, target, env):
    if os.name == 'posix':
        path = '%s/bin/activate' % env
        replace_or_append(path, 'export DJANGO_SETTINGS_MODULE=',
                          'export DJANGO_SETTINGS_MODULE=\'%s.conf.%s\'\n' %
                          (project, target))
    elif os.name == 'nt':
        path = '%s\\Scripts\\activate.bat' % env
        replace_or_append(path, 'set DJANGO_SETTINGS_MODULE=',
                          'set DJANGO_SETTINGS_MODULE=%s.conf.%s\n' %
                          (project, target))
        path = '%s\\Scripts\\deactivate.bat' % env
        replace_or_append(path, 'set DJANGO_SETTINGS_MODULE=',
                          'set DJANGO_SETTINGS_MODULE=\n')


def main():
    virtualenv = args.env
    if not hasattr(sys, 'real_prefix'):
        print('\n== Creating virtual environment ==\n')
        call('virtualenv {0} --python=python3 --prompt="({1}-{2}) "'.format(
            virtualenv, args.project, args.target
        ), shell=True)
    print('\n== Set "%s.conf.%s" as default settings ==\n' % (args.project, args.target))
    append_settings_activate(args.project, args.target, args.env)

    if os.name == 'posix':
        # Make manage.py executable
        st = os.stat('src/manage.py')
        os.chmod('src/manage.py', st.st_mode | stat.S_IEXEC)
        django_admin_symlink = os.path.join(virtualenv, 'bin', 'django')
        if not os.path.exists(django_admin_symlink):
            os.symlink('../../src/manage.py', django_admin_symlink)

# Disabled: we have a separate wsgi script per target for now
#    replace_wsgi_settings(args.target)

    print('\n== Compiling base requirements ==\n')
    if os.name == 'posix':
        bin_dir = 'bin'
    elif os.name == 'nt':
        bin_dir = 'Scripts'
    pip_compile = os.path.join(virtualenv, bin_dir, 'pip-compile')
    os.chdir('requirements')
    call('{0} main.in'.format(pip_compile), shell=True)
    os.chdir('..')

    print('\n== Installing %s requirements ==\n' % args.target)
    if os.name == 'posix':
        pip_path = os.path.join(virtualenv, 'bin', 'pip')
        cmd_tpl = '{pip} install -r requirements/{target}.txt'
    elif os.name == 'nt':
        pip_path = os.path.join(virtualenv, 'Scripts', 'pip')
        cmd_tpl = '{pip} install -r requirements\\{target}.txt'
    call(cmd_tpl.format(pip=pip_path, target=args.target), shell=True)

if __name__ == '__main__':
    main()
    sys.exit(0)