Application: gvSIG desktop » gvSIG scripting

from gvsig import *

from gvsig.commonsdialog import *

from gvsig.libs.formpanel import FormPanel, getResource

import os

import os.path

import shutil

import sys

import subprocess 

import threading

import logging

from java.awt import Dimension

from org.gvsig.scripting.swing.api import ScriptingSwingLocator

class MakeDocs(FormPanel, threading.Thread):

  def __init__(self,workspace,projects,outputfolder):

    threading.Thread.__init__(self)

    FormPanel.__init__(self, getResource(__file__,"makedocs.xml"))

    self.setName("Javadoc creator")

    self.setPreferredSize(700,230)

    self.canceled = False

    self.workspace = workspace

    self.projects = projects

    self.outputfolder = outputfolder

    self.txtWorkspace.setText(workspace)

    self.txtOutputFolder.setText(outputfolder)

  def findJavaSources(self,workspace,project):

    projectFolder = os.path.join(workspace,project)

    sources = list()

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

        for file in files:

            if file == "pom.xml":

              if root[-5:] != ".main":

                javaSources = os.path.join(root,"src","main","java")

                if os.path.isdir(javaSources):

                  sources.append(javaSources)

                  self.message2(root)

    return sources

  def findJavadocSources(self,workspace,projects):

    sources = list()

    for project, url in projects:

      sources.extend(self.findJavaSources(workspace,project))

      if project == "org.gvsig.scripting":

        sources.append(os.path.join(

          workspace,project,"org.gvsig.scripting.app","org.gvsig.scripting.app.mainplugin",

          "src","main","resources-plugin","scripting","lib","gvsig","javadocs"

        ))

    return sources

  def calculateCountFromSources(self,sources):

    countfiles = 0

    folders = set()

    for sourceFolder in sources:

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

        for f in files:

            if f.endswith(".java"):

              countfiles+=1

              self.message2("[%05d] %s" %  (countfiles, os.path.join(root,f)))

        if len(files)>0:

          package = root[len(sourceFolder)+1:]

          folders.add(package)

    return countfiles, len(folders)

  def calculateHTMLFileList(self,htmlroot, listfname):

    listfile= open(listfname,"w")

    countfiles = 0

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

      for f in files:

        if f.startswith("package-"):

          continue

        if f.endswith(".html"):

          countfiles+=1

          listfile.write(os.path.join(root,f))

          listfile.write("\n")

          self.message2("[%05d] %s" %  (countfiles, os.path.join(root,f)))

    listfile.close()

    return countfiles

  def message(self,msg):

    self.lblStatus.setText(msg)

  def message2(self,msg):

    self.lblStatus2.setText(msg)

  def mkjavadoc(self,workspace,projects, outputpath):

    try:

      self.pgbProgreso.setIndeterminate(True)

      self.message("Creando carpeta de salida...")

      if not os.path.exists(outputpath):

        os.makedirs(outputpath)

      os.chdir(outputpath)

      if not os.path.exists("html"):

        os.makedirs("html")

      os.chdir("html")

      if self.chkGenerarJavadocs.isSelected():

        self.message("Eliminando javadoc existentes...")

        os.chdir("..")

        shutil.rmtree("html",True)

        os.makedirs("html")

        os.chdir("html")

        self.message("Localizando fuentes...")

        sources = self.findJavadocSources(workspace,projects)

        self.message("Preparando procesado...")

        maxfiles, maxpackages = self.calculateCountFromSources(sources)

        argsfile = file("javadoc_args","w")

        argsfile.write("-nonavbar\n")

        argsfile.write("-encoding ISO-8859-1\n")

        argsfile.write("-subpackages org:scripting\n")

        argsfile.write("-sourcepath ")

        for source in sources:

          argsfile.write(":%s" % source)

        argsfile.write("\n")

        argsfile.close()

        proc = subprocess.Popen(["javadoc", "@javadoc_args"], stdout=subprocess.PIPE,stderr=subprocess.STDOUT)

        stdoutprocess = proc.stdout 

        generatingcount = 0

        loadingcount=0

        self.pgbProgreso.setMaximum(maxpackages)

        self.pgbProgreso.setValue(0)

        self.pgbProgreso.setIndeterminate(False)

        for line in stdoutprocess:

          if self.canceled:

            break

          #print line[:-1]

          if ": error:" in line :

            self.message("Procesando fuentes...")

            self.message2(line)

          elif line.startswith("Generating "):

            self.message("Generando javadocs...")

            if generatingcount == 0:

              self.pgbProgreso.setMaximum(maxfiles + maxpackages*3 + 9)

            self.message2(line[11:-9])

            if "/package-" in line:

              generatingcount+=1

            else:

              if os.path.basename(line[:-4]).count(".")==1:

                generatingcount +=1

            self.pgbProgreso.setValue(generatingcount)

          elif line.startswith("Loading source files for package "):

            self.message("Cargando fuentes...")

            loadingcount+=1

            self.pgbProgreso.setValue(loadingcount)

            self.message2(line[33:])

      self.message("Proceso terminado")

      self.message2("")

      self.pgbProgreso.setValue(self.pgbProgreso.getMaximum())

      if self.canceled:

        print "Proceso cancelado"

    except Exception, ex:

      self.message("Proceso abortado")

      self.message2(str(ex))

      print str(ex)

      logging.exception("Proceso abortado")

  def btnSelectWorkspace_click(self, *args):

    f = openFolderDialog("Selecciona la carpeta del workspace")

    if f == None or len(f)<1:

      return

    self.txtWorkspace.setText(f[0].getAbsolutePath())

  def btnSelectOutputFolder_click(self, *args):

    f = openFolderDialog("Selecciona la carpeta de salida")

    if f == None or len(f)<1:

      return

    self.txtOutputFolder.setText(f[0].getAbsolutePath())

  def btnCerrar_click(self,*args):

    self.canceled = True

    self.hide()

  def btnProcesar_click(self,*args):

    self.btnProcesar.setEnabled(False)

    self.btnCerrar.setText("Cancelar")

    self.start()

  def showWindow(self,title="Crear javadocs"):

    windowManager = ScriptingSwingLocator.getUIManager()

    windowManager.showWindow(self.asJComponent(),title)

  def run(self):

    self.mkjavadoc(

      self.workspace,

      self.projects, 

      self.outputfolder

    )

    # Falla al volverse a lanzar el proceso por que no se

    # a rearmado correctamente el thread. Asi que dejo

    # desactivado el boton.

    #self.btnProcesar.setEnabled(True)

def main(*args):

  workspace = "/home/jjdelcerro/data/devel/temp"

  outputfolder = workspace + "/javadocs"

  projects = (

    ("org.gvsig.scripting",None),

    ("org.gvsig.tools",None),

    ("org.gvsig.desktop",None),

    ("org.gvsig.gdal",None),

    ("org.gvsig.raster",None),

    ("org.gvsig.raster.gdal",None),

    ("org.gvsig.geoprocess",None),

    ("org.gvsig.3d",None),

    ("org.gvsig.app.document.layout2.app",None),

    ("org.gvsig.attributeeditor",None),

    ("org.gvsig.chart",None),

    ("org.gvsig.complexlegend",None),

    ("org.gvsig.derivedgeometries",None),

    ("org.gvsig.dgn",None),

    ("org.gvsig.downloader",None),

    ("org.gvsig.dwg",None),

    ("org.gvsig.dxf",None),

    ("org.gvsig.dyschromatopsia",None),

    ("org.gvsig.educa.portableview",None),

    ("org.gvsig.expressionfield",None),

    ("org.gvsig.gpe",None),

    ("org.gvsig.hyperlink.app",None),

    ("org.gvsig.jexcel",None),

    ("org.gvsig.jvmpreferences",None),

    ("org.gvsig.legend.dotdensity.app",None),

    ("org.gvsig.legend.graduatedsymbols.app",None),

    ("org.gvsig.legend.proportionalsymbols.app",None),

    ("org.gvsig.legend.quantitybycategory.app",None),

    ("org.gvsig.legend.vectorfilterexpression.app",None),

    ("org.gvsig.mapsheets.app",None),

    ("org.gvsig.postgresql",None),

    ("org.gvsig.projection.jcrs",None),

    ("org.gvsig.raster.ermapper",None),

    ("org.gvsig.raster.georeferencing",None),

    ("org.gvsig.raster.lizardtech",None),

    ("org.gvsig.raster.multifile",None),

    ("org.gvsig.raster.netcdf",None),

    ("org.gvsig.raster.osm",None),

    ("org.gvsig.raster.postgis",None),

    ("org.gvsig.raster.principalcomponents",None),

    ("org.gvsig.raster.reproject",None),

    ("org.gvsig.raster.roimask",None),

    ("org.gvsig.raster.tasseledcap",None),

    ("org.gvsig.raster.tilecache",None),

    ("org.gvsig.raster.tools",None),

    ("org.gvsig.raster.wms",None),

    ("org.gvsig.raster.wcs",None),

    ("org.gvsig.raster.wmts",None),

    ("org.gvsig.seismic",None),

    ("org.gvsig.sld",None),

    ("org.gvsig.vectorediting",None),

    ("org.gvsig.vectorediting.symmetry",None),

    ("org.gvsig.vectorediting.offset",None),

    ("org.gvsig.wfs.app",None),

    ("org.gvsig.customize.app",None),

    ("org.gvsig.catalog",None),

    ("org.gvsig.gazetteer",None),

    ("org.gvsig.publish",None),

    ("org.gvsig.webmap",None),

    ("org.gvsig.googlemaps",None),

    ("org.gvsig.bingmaps",None),

    ("org.gvsig.toolbox", None),

  )

  makedocs = MakeDocs(workspace,projects,outputfolder)

  makedocs.showWindow("Crear javadocs")

  #makedocs.run()