source: SH_SHM/trunk/setup/setup.py @ 343

#! /usr/bin/env python

import os
import sys

default_location = "/usr/local/sh"
downloads = {
    "locsat": "ftp://ftp.szgrf.bgr.de/pub/software/locsat-linux.tar.gz",
    "fk": "ftp://ftp.szgrf.bgr.de/pub/software/fk-linux.tar.gz"
}

def fetchinfo(**kwargs):
    """
    Fetch information about user etc.
    """
    dbg = kwargs["debug"]

    home = os.getenv("HOME")
    pwd = os.path.realpath(os.path.split(sys.argv[0])[0])
    tmp = os.getenv("TEMP")
    if not tmp:
        tmp = pwd

    uid = os.getuid()
    root = uid == 0
    if root:
        shcroot = default_location
    else:
        shcroot = os.path.join(home, "sh")

    del kwargs
    return locals()

def banner():
    print "="*70
    print "  Setup program v0.1 for Seismic Handler (SH/SHM)"
    print "="*70
    print "\n This piece of software will try to setup the included source code.\n"

def download(target, pkg, dbg):
    print "Downloading %s..." % pkg

    verbose = dbg and "" or "-q "

    if os.system("wget %s-P %s %s" % (verbose, target, downloads[pkg])):
        abort("Cannot download %s. Exit here." % pkg)

def setup(**kwargs):
    """
    Setup Seismic Handler from source.
    """
    banner()
    dbg = kwargs["debug"]

    data = fetchinfo(debug=dbg)
    if dbg:
        print data

    if not data["root"]:
        res = query("You have no administrative rights. "
                    "Perform user installation [Yn]? ")
        if res.lower() == "n":
            abort()

    while 1:
        target = query("Installation directory [%s, q = exit]: " % data["shcroot"])
        if target.lower() == "q":
            abort()

        if not target:
            target = data["shcroot"]

        if not os.path.exists(target):
            try:
                os.mkdir(target)
            except Exception, e:
                if dbg:
                    print e

                print "Cannot create target directory: %s" % target
                continue

        break

    if dbg:
        print "Installation target: %s" % target

    copy(os.path.join(data["pwd"], ".."), os.path.join(target, "sh"), dbg)
    download(target, "locsat", dbg)
    download(target, "fk", dbg)

    exitcode = install(target, dbg)
    if exitcode == 99:
        abort("Installation aborted.")
    elif exitcode:
        abort("Installation errors occured. Please check output.")

    customize(target, dbg)

def customize(target, dbg):
    """
    Offers usage of new configuration files.

    Save user defined configuration files, if choosen.
    """

    flist = ["STATINF.DAT", "sensitivities.txt", "filter_lookup.txt"]

    print "\nSeismic Handler comes with updated station configuration files."
    a = raw_input("Do you want to use them (backup will be saved) [Yn]? ")

    if a == "n":
        print "\nPlease check the following files manually (sh/inputs):"
        for f in flist:
            print f, "vs.", f+".dist"
        return

    for f in flist:
        tname = os.path.join(target, "sh/inputs", f)
        if os.path.exists(tname):
            print "Backing up %s... (to %s)" % (f, f+".yours")
            os.system("cp %s %s" % (tname, tname+".yours"))
        os.system("cp %s %s" % (tname+".dist", tname))

def install(target, dbg):
    exitcode = \
         os.system(os.path.join(target, "sh/setup/SHM-install.csh %s" % target))

    return exitcode >> 8

def copy(src, trt, dbg=False):
    if dbg:
        os.system("cp -rvd %s %s" % (src, trt))
    else:
        os.system("cp -rd %s %s" % (src, trt))

def abort(text=None):
    if not text:
        text = "Installation aborted by user."

    quit(text)

def query(text):
    """
    Query user.
    """
    return raw_input(text)

if __name__ == "__main__":
    from optparse import OptionParser
    parser = OptionParser()
    parser.add_option("--debug", dest="debug", action="store_true",
                      help="Turn on debug option.", default=False)

    (options, args) = parser.parse_args()

    setup(**vars(options))