setup.py

"""
Setup script
Run me after cloning the CompDam_DGD repo to finish the initial setup:
  1. Hooks for keeping for/version.for up to date
  2. ... Add other features here ...
"""

import platform
import shutil
import os
import sys
import imp
import pip
import subprocess
import json
import shutil

if __name__ == "__main__":

    config = dict()
    config["git"] = False
    config["bash"] = False
    config["abaverify"] = False

    # Check for compatible versions of Python
    if sys.version_info[0:2] != (2,7):
        raise Exception("setup.py run with Python version " + '.'.join(str(s) for s in sys.version_info[0:3]) + ". Version 2.7 is required to setup and run CompDam_DGD.")
    else:
        print "Found python version: {0}".format(str(sys.version))

    # Check for git
    try:
        git_output = subprocess.check_output(["git", "--version"], shell=True)
        print "Found git version: {0}".format(git_output.rstrip())

        # Create git hooks
        shutil.copyfile(os.path.join(os.getcwd(), 'etc', 'hook.sample'), os.path.join(os.getcwd(), '.git', 'hooks', 'post-checkout'))
        shutil.copyfile(os.path.join(os.getcwd(), 'etc', 'hook.sample'), os.path.join(os.getcwd(), '.git', 'hooks', 'post-commit'))
        shutil.copyfile(os.path.join(os.getcwd(), 'etc', 'hook.sample'), os.path.join(os.getcwd(), '.git', 'hooks', 'post-update'))
        shutil.copyfile(os.path.join(os.getcwd(), 'etc', 'hook.sample'), os.path.join(os.getcwd(), '.git', 'hooks', 'post-merge'))
        print "Copied the git hooks to update versioning on git command execution"

        # Set permissions
        if platform.system() == 'Linux':
            os.chmod(os.path.join(os.getcwd(), '.git', 'hooks', 'post-checkout'), 0775)
            os.chmod(os.path.join(os.getcwd(), '.git', 'hooks', 'post-commit'), 0775)
            os.chmod(os.path.join(os.getcwd(), '.git', 'hooks', 'post-update'), 0775)
            os.chmod(os.path.join(os.getcwd(), '.git', 'hooks', 'post-merge'), 0775)
            print "Changed file persions for git hooks"

        # This configures version.for
        os.system('git checkout')
        print "Checked out repository"

        # Successfully set up versioning using git
        config["git"] = True

        # Try bash (for the python extension module)
        try:
            bash_output = subprocess.check_output(["bash", "--version"], shell=True)
            print "Found bash version: {0}".format(bash_output.split('\n')[0])
            config["bash"] = True

        except subprocess.CalledProcessError:
            print "Bash not found"
            print "NOTICE: Bash is required to run the python extension module (useful for development).\n\tIf running windows, install the windows subsystem for linux (WSL).\n\tSee the readme for more information."


    except subprocess.CalledProcessError:
        shutil.copy(os.path.join(os.getcwd(), 'for', 'version.for.nogit'), os.path.join(os.getcwd(), 'for', 'version.for'))
        print "Git not found"
        print "NOTICE: If you intend to make contributions to the CompDam code base, using git is required.\n\tTo setup CompDam using git, type:\n\t\tgit clone https://github.com/nasa/CompDam_DGD.git\n\tSee the readme for more details.\n"

    # Check for the installation of abaverify
    try:
        imp.find_module('abaverify')
        import abaverify
        print "Found abaverify version {0}".format(abaverify.__version__)
        config["abaverify"] = True
    except ImportError:
        print """
abaverify is failed to import. abaverify is a Python package for running verification tests on Abaqus user subroutines.
More information can be found here: https://github.com/nasa/abaverify
        """
        install_abaverify = raw_input("Would you like to install abaverify now? (yes/no) ")
        if 'y' in install_abaverify.lower():
            print "Attempting to install abaverify with pip version {0}".format(pip.__version__)
            pip_cmd = ['install', '--src', os.pardir, '-e','git+https://github.com/nasa/abaverify#egg=abaverify', '--user']
            if int(pip.__version__.split('.')[0]) > 10:
                from pip._internal import main as pipcall
                pipcall(pip_cmd)
            else:
                pip.main(pip_cmd)
            config["abaverify"] = True

    # Write the config file
    with open('etc/config.json', 'w') as f:
        json.dump(config, f, indent=2)

    print "Setup completed successfully"