Application: gvSIG desktop

#!/usr/bin/python

import sys

import os

import os.path

import fnmatch

import shutil

import zipfile 

import stat

import getopt

from os.path import dirname

RWALL = stat.S_IWOTH | stat.S_IROTH | stat.S_IWUSR | stat.S_IRUSR | stat.S_IWGRP | stat.S_IRGRP 

VERBOSE = False

VERSION = os.path.basename(os.getcwd())

GVSPKG_ROOT = os.path.abspath("../..")

GVSPKG_ROOT = "/home/groups/gvsig-desktop/www/gvSIG-desktop"

def getVersion():

    return VERSION

def getPackagesRoot():

    return GVSPKG_ROOT

def getPool():

    return getPackagesRoot() + "/pool"

def getDist():

    return getPackagesRoot() + "/dists/" +getVersion()

def message(msg):

    if VERBOSE:

        print msg

def msgerror(msg):

    print msg

def findfiles(root,filename):

    result=list()

    for wroot, dirs, files in os.walk(root):

        for f in files :

            if fnmatch.fnmatch(f,filename):

                result.append(wroot+"/"+f)

    result.sort()

    return result

class IndexList(list):

    def load(self, fname):

        message( "Loading index list from '%s'." % fname)

        f=file(fname,"r")

        lines=f.readlines()

        f.close()

        for line in lines:

          if line[-1] == "\n":

            line = line[:-1]

          self.append(line)

    def save(self,fname):

        message( "Saving index list from '%s'." % fname)

        f=file(fname,"w")

        for index in self:

          f.write("%s\n" % index)

        f.close()

        try:

            os.chmod(fname,RWALL)

        except Exception, ex:

            msgerror("Can't change permissions of file '%s', error %s" % (fname, str(ex)))

    def build(self, pool, version):

        message( "Creating index list for version '%s' from '%s'" % (version, pool) )

        packages=dict()

        for root, dirs, files in os.walk(pool):

            for f in files :

                if fnmatch.fnmatch(f,"gvSIG-desktop-"+version+"-*.gvspki"):

                    fullpath = root+"/"+f

                    if packages.get(dirname(fullpath)) == None:

                        packages[dirname(fullpath)]=fullpath

                    else:

                        s = packages[dirname(fullpath)]

                        if s<fullpath :

                            packages[dirname(fullpath)]=fullpath

        self.extend(packages.values())

        self.sort()

def lsi(args):

    try:

        opts, args = getopt.getopt(args[1:], "l", ["long-format"])

    except getopt.GetoptError, err:

        # print help information and exit:

        print str(err) # will print something like "option -a not recognized"

        help(args)

        sys.exit(2)

    long_format=False

    for opt, arg in opts:

        if opt in ("-l", "--long-format"):

            long_format = True

        else:

            assert False, "unhandled option"

    indexes = IndexList()

    indexes.build(getPool(), getVersion())

    for index in indexes:

        pkg_name = os.path.basename(index)

        print pkg_name

        if long_format:

            show(["show", pkg_name])

def mks(args):

    try:

        opts, args = getopt.getopt(args[1:], "c", ["clear-list"])

    except getopt.GetoptError, err:

        # print help information and exit:

        print str(err) # will print something like "option -a not recognized"

        help(args)

        sys.exit(2)

    clear_list=False

    for opt, arg in opts:

        if opt in ("-c", "--clear-list"):

            clear_list = True

        else:

            assert False, "unhandled option"

    indexes = IndexList()

    packages_txt = getDist() +"/packages.txt"

    packages_gvspki = getDist() +"/packages.gvspki"

    packages_gvspks = getDist() +"/packages.gvspks"

    message( "Creating 'packages.gvspki' for version '%s'" % getVersion() )

    if not os.path.exists(getDist()):

        msgerror("Can't locate version folder '%s'." % getVersion())

        sys.exit(3)

    if os.path.exists(packages_txt) and not clear_list:

        indexes.load(packages_txt)

    else:

        indexes.build(getPool(), getVersion())

        indexes.save(packages_txt)

    message( "Writing 'packages.gvspki' to '%s'" % packages_gvspki )

    set = zipfile.ZipFile(packages_gvspki,"w",zipfile.ZIP_STORED)

    for index in indexes:

        try:

            set.write(index, os.path.basename(index))

        except Exception, ex:

            msgerror("Can't add index '%s', error %s" % (index, str(ex)))

    set.close()

    try:

        os.chmod(packages_gvspki,RWALL)

    except Exception, ex:

        msgerror("Can't change permissions of file '%s', error %s" % (packages_gvspki, str(ex)))

    message( "Writing 'packages.gvspks' to '%s'" % packages_gvspks )

    set = zipfile.ZipFile(packages_gvspks,"w",zipfile.ZIP_STORED)

    for index in indexes:

        index = index.replace(".gvspki",".gvspkg")

        try:

            set.write(index, os.path.basename(index))

        except Exception, ex:

            msgerror("Can't add package '%s', error %s" % (index, str(ex)))

    set.close()

    try:

        os.chmod(packages_gvspks,RWALL)

    except Exception, ex:

        msgerror("Can't change permissions of file '%s', error %s" % (packages_gvspks, str(ex)))

    message( "Created")

def show(args):

    index_name=args[1]

    message( "Show package.info from '%s'" % index_name)

    files = findfiles(getPool(), index_name)

    if len(files) != 1:

        msgerror("Can't locate package '%s' in pool '%s'." % (index_name, getPool()))

        return

    index_path = files[0]

    outputfolder="/tmp/gvspkg.%s" % os.getpid()

    os.mkdir(outputfolder)

    os.system('unzip -q %s -d %s' % (index_path,outputfolder))

    files = findfiles(outputfolder, "package.info")

    if len(files) != 1:

        msgerror("Can't locate package.info in pool '%s'." % (index_name))

        return

    package_info = files[0]

    f = file(package_info,"r")

    s = f.read()

    f.close()

    shutil.rmtree(outputfolder)

    print s

def edit(args):

    index_name=args[1]

    message( "Show package.info from '%s'" % index_name)

    files = findfiles(getPool(), index_name)

    if len(files) != 1:

        msgerror("Can't locate package '%s' in pool '%s'." % (index_name, getPool()))

        return

    index_path = files[0]

    outputfolder="/tmp/gvspkg.%s" % os.getpid()

    os.mkdir(outputfolder)

    os.system('unzip -q %s -d %s' % (index_path,outputfolder))

    files = findfiles(outputfolder, "package.info")

    if len(files) != 1:

        msgerror("Can't locate package.info in pool '%s'." % (index_name))

        return

    package_info = files[0]

    code = package_info.split("/")[-2]

    os.system('vi "%s"' % package_info)

    temp_index_name = "/tmp/packages.gvspki.%s" % os.getpid()

    temp_index = zipfile.ZipFile(temp_index_name,"w",zipfile.ZIP_STORED)

    temp_index.write(package_info, "%s/package.info" % (code))

    temp_index.close()

    shutil.rmtree(outputfolder)

    os.remove(index_path)

    shutil.move(temp_index_name,index_path)

def help(args):

    print """

usage: gvspkg [OPTIONS] COMMANDS

OPTIONS:

-h|--help

    Muestra esta ayuda

-v

    Activa el modo verbose

--version version

    Fija la version con la que van a trabajar los comandos indicados.

-p|--package-root package-root

    Fija la carpeta en la que va a buscar el raiz del sistema de paquetes

    con el que trabajar

COMMANDS:

lsi [OPTIONS]

    Lista los indices disponibles de la version

    OPTIONS:

    -l | --long-format

        Muestra para cada paquete la informacion del package.info

mks [OPTIONS]

     Crea el fichero packages.gvspki con los indices de la

     version.

     OPTIONS:

     -c | --clear-list

        Elimina la lista de paquetes a utilizar recreandola a partir

        de los paquetes del pool tomando el ultimo build de cada paquete.

show package-index

    Muestra el package.info del indice indicado como parametro.

edit package-index

    Edita el package.info del indice indicado como parametro.

Por defecto la version la obtiene del nombre de la carpeta 

corriente (%s).

El directorio root de la estructura de packetes lo buscara en el

sitio indicado por la variable de entorno GVPKG_ROOT, y si esta no

esta establecida usara "%s"

    """ % (VERSION, GVSPKG_ROOT)

def main():

    global VERSION

    global VERBOSE

    global GVSPKG_ROOT

    try:

        opts, args = getopt.getopt(sys.argv[1:], "hvr:", ["version=", "package-root=","help"])

    except getopt.GetoptError, err:

        # print help information and exit:

        print str(err) # will print something like "option -a not recognized"

        help(None)

        sys.exit(2)

    for opt, arg in opts:

        if opt in ("-h", "--help"):

            help(args)

            sys.exit()

        elif opt in ("-v"):

            VERBOSE = True

        elif opt in ("--version"):

            VERSION = arg

        elif opt in ("-r", "--package-root"):

            GVSPKG_ROOT = arg

        else:

            assert False, "unhandled option"

    # 

    message("VERSION=%s" % VERSION)

    message("GVSPKG_ROOT=%s" % GVSPKG_ROOT)

    command = "help"

    if len(args)>0:

        command=args[0]

    if command=="lsi" :

        lsi(args)

    elif command == "mks":

        mks(args)

    elif command == "edit":

        edit(args)

    elif command == "show":

        show(args)

    else:

        help(args)

main()