Le portail francophone de la géomatique

# -*- coding:Latin1 -*-

#Import Standard library modules
import sys, os
try:
    #pour les utilisateurs ArcGIS 9.2/9.3
    import arcgisscripting
    gp = arcgisscripting.create()
    print "\n" + "Utilisation de ArcGIS 9.2 ou supérieure avec arcgisscripting..." + "\n"
    gp.AddMessage("\n" + "Utilisation de ArcGIS 9.2 ou supérieure avec arcgisscripting..." + "\n")
except:
    #pour les utilisateurs ArcGIS 9.0/9.1
    import win32com.client
    gp = win32com.client.Dispatch("esriGeoprocessing.GpDispatch.1")
    print "\n" + "Utilisation de ArcGIS 9.0/9.1 ou supérieure avec win32com.client.Dispatch..." + "\n"
    gp.AddMessage("\n" + "Utilisation de ArcGIS 9.0/9.1 ou supérieure avec win32com.client.Dispatch..." + "\n")

def listFiles(dir_name, subdir, *args):
    '''Return a list of file names found in directory 'dir_name'
    If 'subdir' is True, recursively access subdirectories under 'dir_name'.
    Additional arguments, if any, are file extensions to match filenames. Matched
        file names are added to the list.
    If there are no additional arguments, all files found in the directory are
        added to the list.
    Example usage: fileList = dirEntries(r'H:\TEMP', False, 'txt', 'py')
        Only files with 'txt' and 'py' extensions will be added to the list.
    Example usage: fileList = dirEntries(r'H:\TEMP', True)
        All files and all the files in subdirectories under H:\TEMP will be added
        to the list.
    '''
    fileslist = []
    for file in os.listdir(dir_name):
        dirfile = os.path.join(dir_name, file)
        if os.path.isfile(dirfile):
            if not args:
                fileslist.append(dirfile)
            else:
                if os.path.splitext(dirfile)[1][1:] in args:
                    fileslist.append(dirfile)
        # recursively access file names in subdirectories
        elif os.path.isdir(dirfile) and subdir:
            print "Accès au répertoire :", dirfile
            fileslist.extend(listFiles(dirfile, subdir, *args))
    return fileslist



# MAIN
if __name__ == '__main__':

    # Allow output file to be overwritten if it already exists
    gp.OverwriteOutput = 1


    #Arguments
    #startdir = sys.argv[1]#à l'ancienne
    startdir = gp.getparameterastext(0) 'localisation racine
    subdir = gp.getparameterastext(1) 'true pour la récursivité 
    #startdir = "D:\temp"
    #subdir = True

    filetype = 'shp' #filtre de type d'extension à prendre en compte dans le listing
try:
    print "*"*10
    print "Processus : conversion des shapes contenu dans \n" + startdir
    gp.AddMessage("*"*10)
    gp.AddMessage("Processus : conversion des shapes contenu dans \n" + startdir)

    #Lister des fichiers à insérer en base
    myFileList = []
    myFileList = listFiles(startdir, subdir, filetype);
    
    #Recupere l espace de travail
    for f in myFileList:
        filepath = os.path.dirname(f);
        filename = os.path.basename(f).split(".")[0];
        fileext = os.path.basename(f).split(".")[1];

        #traitement sur le fichier
        print "export du fichier :", os.path.basename(f)
        gp.AddMessage("export du fichier :" + os.path.basename(f))
        gp.AddToolbox("C:/Program Files/ArcGIS/ArcToolbox/Toolboxes/[b]TOOLBOX.tbx[/b]")
        [b]ici ton traitement[/b]


    
    print "Processus : Fin du traitement"
    print "*"*10
    gp.AddMessage("Processus : Fin du traitement")
    gp.AddMessage("*"*10)
    

except:
    gp.AddMessage(gp.GetMessages(2))
    print gp.GetMessages(2)