install.py

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

#!/usr/bin/env python
import os, sys, shutil

# FHS-compliant.
share_target = '/usr/local/share/zohmg'
lib_target   = '/usr/local/lib/zohmg'
doc_target   = share_target + '/doc'
egg_target   = lib_target + '/egg'
jar_target   = lib_target + '/jar'
mapred_target = lib_target + '/mapred'

targets = [share_target, doc_target, lib_target, egg_target, jar_target, mapred_target]


def clean():
    """cleans things a bit."""
    print
    print 'cleaning previous zohmg installation:'

    # remove target directories.
    for dir in targets:
        print " " + dir
        os.system("rm -rf %s" % dir)


def copy_files():
    """populate /usr/local/lib/zohmg and /usr/local/share/zohmg"""
    print
    print "populating zohmg directories:"

    # create target directories.
    for dir in targets:
        if not os.path.isdir(dir):
            os.mkdir(dir)

    # copy docs.
    docs = ['AUTHORS', 'CONTRIBUTE', 'DEPENDENCIES', 'FAQ', 'INSTALL', 'README']
    for doc in docs:
        copy_file(doc, doc, doc_target)

    # copy stuff to share
    shutil.copytree('examples', share_target + '/examples')
    shutil.copytree('graph', share_target + '/graph')
    shutil.copytree('skel-project', share_target + '/skel-project')
    # and to lib
    shutil.copytree("src/zohmg/middleware", lib_target+"/middleware")
    copy_file("bundled eggs", "lib/egg/*.egg", egg_target)
    copy_file("bundled jars", "lib/jar/*.jar", jar_target)
    copy_file("dumbo bootstrapper", "lib/mapred/import.py", mapred_target)


# assumes that setuptools is available.
def python_modules():
    print
    print "building python eggs:"

    egg_log = '/tmp/zohmg-egg-install.log'
    egg_err = '/tmp/zohmg-egg-install.err'
    redirection = ">> %s 2>> %s" % (egg_log, egg_err)
    os.system('date > %s ; date > %s' % (egg_log, egg_err)) # reset logs.

    modules = ['paste', 'pyyaml', 'simplejson']
    print '(assuming setuptools is available.)'
    print '(logging to ' + egg_log + ' and ' + egg_err + ')'
    for module in modules:
        print 'module: ' + module
        r = os.system("easy_install -maxzd %s %s %s" % (egg_target, module, redirection))
        if r != 0:
            print
            print 'trouble!'
            print 'wanted to easy_install modules but failed.'
            print 'logs are at ' + egg_log + ' and ' + egg_err
            # pause.
            print "press ENTER to continue the installation or CTRL-C to break."
            try: sys.stdin.readline()
            except KeyboardInterrupt:
                print "ok."
                sys.exit(1)
    print 'python eggs shelled in ' + egg_target


def setup():
    """calls setup.py"""
    print
    print "installing zohmg egg:"

    # install,
    log = '/tmp/zohmg-install.log'
    err = '/tmp/zohmg-install.err'
    setup_cmd = sys.executable + ' setup.py install > %s 2> %s' % (log, err)
    os.chdir('src')
    r = os.system(setup_cmd)
    if r != 0:
        # try once more immediately; usually works.
        r = os.system(setup_cmd)
        if r != 0:
            print 'trouble!'
            print 'could not install zohmg: python setup.py install'
            print 'log are at ' + log + ' and ' + err
            sys.exit(r)

    # let the user know what happened, clean up.
    os.system("egrep '(Installing|Copying) zohmg' " + log)
    os.system("rm -rf build dist zohmg.egg-info")

def test():
    print 'testing zohmg script: ',
    r = os.system('zohmg 2>&1 > /dev/null')
    if r != 0:
        # fail!
        print 'trouble!'
        print 'test run failed; it seems something is the matter with the installation :-|'
        sys.exit(r)
    print 'ok!'


# copies bundle (file) to destination, printing msg.
def copy_file(msg, file, destination):
    os.system("cp -v %s %s" % (file, destination))


if __name__ == "__main__":

    # check for rootness.
    if os.geteuid() != 0:
        print "you need to be root. please sudo."
        sys.exit(1)

    clean()
    copy_files()
    python_modules()
    setup()
    test()

    print
    print "ok, that should do it!"
    print "now try this:"
    print "$> zohmg help"
    print