source: alternc-apps/trunk/bin/update_apps.py @ 2466

#!/usr/bin/python

import adodb, os, sys, getopt

alterncapps  = "/var/alternc/apps"
host         = "localhost"
database     = "alternc"
username     = "alternc"
password     = "nJWuYCgj0K"

def usage():
    print "update_apps.py usage:"
    print "  -c --cron cron mode"
    print "  -u [appname] --update=[appname] update installation of appname"

def generateFileList(path):
    directories = []
    fileslist   = []
    for root, dirs, files in os.walk(path):
        root = root.replace(path , "")

        if root != '':
            directories.append(root)

        for file in files:
            fileslist.append(root + "/" + file )

    return {'files': fileslist, 'dirs': directories}

def createDirIfNotExist(dir):
    if not os.path.exists(dir):
        os.mkdir(dir)
        os.chmod(dir,0770)
        os.chown(dir,33,-1)

def installFile(fromfile, tofile):
    if not os.path.exists(tofile):
        fromfile = os.path.normpath(fromfile)
        tofile   = os.path.normpath(tofile)
        os.symlink(fromfile, tofile)

def updateFiles(fromfiles,tofiles,frompath,path):
    # we recreate the directories
    for dir in fromfiles['dirs']:
        if dir not in tofiles['dirs']:
            createDirIfNotExist(os.path.normpath(path + "/" + dir))

    for file in fromfiles['files']:
        if file not in tofiles['files']:
            installFile(os.path.normpath(frompath + "/" + file),os.path.normpath(path + "/" + file))

    for file in tofiles['files']:
        filepath = os.path.normpath(path + "/" + file)
        if not os.path.exists(filepath):
            os.unlink(filepath)

def installApplication(application, path):
    fromfiles = generateFileList(alterncapps + "/" + application)
    tofiles = generateFileList(path)
    print tofiles
    for dir in files['dirs']:
        createDirIfNotExist(path + dir)

    for file in files['files']:
        installFile(alterncapps + "/" + application + "/" + file, path + "/" + file)

def updateApplication(appname):
    print "Upgrading : "  + appname

    conn = adodb.NewADOConnection('mysql')
    conn.Connect(host, username, password,database);

    files   = generateFileList(alterncapps + "/" + appname)
 
    for row in conn.Execute('SELECT uid, application, path FROM applications WHERE application = %s AND installed = 1',(appname)):
        tofiles = generateFileList(row[2])
        updateFiles(files, tofiles, os.path.normpath(alterncapps + "/" + appname), row[2])


def runCron():
    conn = adodb.NewADOConnection('mysql')
    conn.Connect(host, username, password,database);

    for row in conn.Execute('SELECT uid, application, path FROM applications WHERE installed = 0'):
        uid         = row[0]
        application = row[1]
        path        = row[2]
        installApplication(application,path)
        conn.Execute('UPDATE applications SET installed = 1 WHERE uid = %s AND application = %s AND path = %s',(uid,application,path))

try:
    optlist, args = getopt.getopt(sys.argv[1:],'cu:', ["cron","update="])
    for option, arg in optlist:
        if option in ("--cron","-c"):
            runCron()
        elif option in ("--update","-u"):
            updateApplication(arg)
        else:
            usage()
            exit(2)

except getopt.GetoptError, err:
    print str(err)
    usage()
    sys.exit(2)