Le portail francophone de la géomatique

# -*- coding:Latin-1 -*-
import os, sys
import shapefile

#FONCTIONS
def filesList(inFolder, ext):
    """Lister les fichiers d'un dossier. La liste est filtrée selon une extension donnée"""
    lst = os.listdir(inFolder)
    extLst=[elem for elem in lst if os.path.splitext(elem)[1]==ext]
    extLst.sort()
    return extLst

def addZeros(num, precision):
    """Retourne une chaine représentant le nombre en entrée préfixé
    de zeros afin que la longueur de la chaine soit égale à la precision"""
    numZeros=str(num)
    while len(numZeros)<precision: numZeros='0'+numZeros
    return numZeros


#INPUTS
inFolder="C:\\Documents and Settings\\dominique\\Bureau"
outFolder="C:\\Documents and Settings\\dominique\\Bureau\\output"


#MAIN
shpList=filesList(inFolder, '.shp')
for pathShp in shpList:
    shp=shapefile.Reader(pathShp)
    shapeRec = shp.shapeRecords()
    fields=shp.fields
    for idx,entity in enumerate(shapeRec):
        geomType=entity.shape.shapeType
        NewShp = shapefile.Writer(geomType)
        NewShp.fields = list(fields)
        NewShp._shapes.extend([entity.shape])#argument doit être itérable --> conversion en list nécessaire
        NewShp.records.extend([entity.record])#idem
        #Saving
        basename=os.path.basename(pathShp)
        fileName=basename[:-4]
        extension=basename[len(basename)-4:len(basename)]
        NewShp.save(outFolder+os.sep+fileName+'_'+addZeros(idx,5)+extension)