Source code for invirtualenv_plugins.rpm_scripts.post_install

#!/usr/bin/env python
from __future__ import print_function
import logging
import os
import shutil
import sys

from configparser import ConfigParser, NoOptionError
from invirtualenv.deploy import build_deploy_virtualenv


def update_config(venv_dir, config_filename=None):
    if not config_filename:
        config_filename = 'deploy.conf'
    config = ConfigParser()
    config.read(config_filename)
    config.set('global', 'virtualenv_deploy_dir', venv_dir)
    with open(config_filename, 'w') as configfile:
        config.write(configfile)


def get_config_flag(section, option, config_filename=None):
    if not config_filename:
        config_filename = 'deploy.conf'
    config = ConfigParser()
    config.read(config_filename)
    try:
        return config.getboolean(section, option)
    except NoOptionError:
        return


if __name__ == "__main__":
    # logging.basicConfig(level=logging.DEBUG)
    logger = logging.getLogger(sys.argv[0])

    logger.debug('Running post install steps, from directory %r' % os.getcwd())

    if not os.path.exists('deploy.conf'):
        print("No 'deploy.conf' found.  Doing nothing", file=sys.stderr)
        sys.exit(1)

    logger.debug('Using the following deploy.conf: %s' % open('deploy.conf').read())

    venv_directory = None
    try:
        venv_directory = build_deploy_virtualenv(update_existing=True)
        update_config(venv_directory)
    except:
        print('Unable to create the python virtualenv', file=sys.stderr)
        logger.exception('The virtualenv create failed, removing venv_directory')
        if venv_directory and os.path.exists(venv_directory):
            print('Removing the %r virtualenv directory' % venv_directory)
            shutil.rmtree(venv_directory)
        sys.exit(1)

    try:
        if get_config_flag('global', 'link_bin_files'):
            logger.debug('Linking files in the virtualenv bin directory to %r', os.path.dirname(sys.executable))
            try:
                from invirtualenv.deploy import link_deployed_bin_files
                link_deployed_bin_files(venv_directory, '/usr/bin')
            except ImportError:
                print('WARNING: The installed version of invirtualenv does not support linking bin files')
    except:
        logger.exception('An error occurred linking bin files into %r', os.path.dirname(sys.executable))

    print('Created virtualenv %r' % venv_directory)
    sys.exit(0)