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

#! /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 "="*78
    print "  Setup program for Seismic Handler v0.1"
    print "="*78
    print "\n This piece of software will try to setup the included source code.\n"

def download(pkg):
    os.system("wget")

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

    copy(os.path.join(data["pwd"], ".."), target, dbg)

def copy(src, trt, dbg=False):
    if dbg:
        os.system("echo 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))