Script to install package using .travis.yml

travis_setup.py

#!/usr/bin/env python

'''
    Install package from .travis.yml

    usage: travis_setup.py

    @author     Tarun Bhardwaj
    @license    FreeBSD Licence
'''

import os

try:
    import yaml
except ImportError, e:
    os.system('pip install pyyaml')
    import yaml

def run(cmd):
    '''
        Execute cmd on terminal.

        :param cmd: terminal command/list or commands
    '''
    if cmd:
        map(os.system, cmd if isinstance(cmd, list) else [cmd])

file = open(".travis.yml", "r")
setup = yaml.load(file)

run(setup.get('before_install'))
run(setup.get('install'))