Le portail francophone de la géomatique

# Paramètres du script
#Couche d'itinéraire
IN_LAYER_ITI = unicode(arcpy.GetParameterAsText(0)).encode('cp1252') # unicode est utilisé pour gérer les problèmes d'encode. Il n'est pas nécessaire de l'employer s'il n'y a pas de caractères spéciaux dans le nom des couches
#Couche des points de repère
IN_LAYER_PTREP = unicode(arcpy.GetParameterAsText(1)).encode('cp1252')
#Répertoire en sortie
OUT_FOLDER = unicode(arcpy.GetParameterAsText(3)).encode('cp1252')

try:
    # Récupération du document courant
    mxd = arcpy.mapping.MapDocument("CURRENT")

    # Vérification de la configuration des pages dynamiques
    try:
        nbPages = mxd.dataDrivenPages.pageCount + 1
    except:
        raise Exception("DataDrivenPagesError")       

    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Création du fichier PDF dans le dossier "+ OUT_FOLDER)
    if os.path.exists(OUT_PDF_FILE):
        try:
            os.remove(OUT_PDF_FILE)
        except:
            raise Exception("RemovePdfFileError")               
        
    finalPdf = arcpy.mapping.PDFDocumentCreate(OUT_PDF_FILE) 
    
    # Génération du plan d'ensemble
    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Récupération de l'étendue de la couche")
    for layer in arcpy.mapping.ListLayers(mxd, IN_LAYER_ITI):
        layerExtent = layer.getExtent()
    
    dataFrame = arcpy.mapping.ListDataFrames(mxd)[0]
    dataFrame.extent = layerExtent
    dataFrame.scale = dataFrame.scale * 1.1

    mxd.title = IN_LAYER_ITI + " - Plan d'ensemble"

    arcpy.RefreshActiveView()
    arcpy.RefreshTOC()

    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Export du plan d'ensemble")
    tmpPdf = os.path.splitext(OUT_PDF_FILE)[0] + "_tmp.pdf"
    if os.path.exists(tmpPdf):
        os.remove(tmpPdf)
        
    arcpy.mapping.ExportToPDF(mxd, tmpPdf)
    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Insertion dans le document final...")
    finalPdf.appendPages(tmpPdf)
    if os.path.exists(tmpPdf):
        os.remove(tmpPdf)
            
    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Application d'un ensemble de définition sur la couche " + IN_LAYER_ITI)
    queryDef = ITI_FLD_ORDRE + ">=" + str(ordreMin - IN_TOLERANCE_ORDRE) + " AND " + ITI_FLD_ORDRE + "<=" + str(ordreMax + IN_TOLERANCE_ORDRE)
    layerIti = arcpy.mapping.ListLayers(mxd, IN_LAYER_ITI)[0]
    layerIti.definitionQuery = queryDef
    
    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Application d'un ensemble de définition sur la couche " + IN_LAYER_PTREP)
    queryDef = FLD_NUMERO + "=" + str(pageNum) + " OR " + FLD_NUMERO + "= 1 + " + str(pageNum) 
    for layer in arcpy.mapping.ListLayers(mxd, IN_LAYER_PTREP):
        layer.definitionQuery = queryDef

    # Modification du titre (il doit être présent dans la mise en page)
    mxd.title = IN_LAYER_ITI + " - Partie " + str(pageNum)

    #Rafraichissement de la vue
    arcpy.RefreshActiveView()
    arcpy.RefreshTOC()

    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Export de la page {0} sur {1}.".format(str(pageNum), str(mxd.dataDrivenPages.pageCount)))

    tmpPdf = os.path.splitext(OUT_PDF_FILE)[0] + "_tmp" + str(pageNum) + ".pdf"
    if os.path.exists(tmpPdf):
        os.remove(tmpPdf)
        
    arcpy.mapping.ExportToPDF(mxd, tmpPdf)
    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Insertion dans le document final...")
    finalPdf.appendPages(tmpPdf)

    arcpy.AddMessage(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Suppression du fichier temporaire...")
    if os.path.exists(tmpPdf):
        os.remove(tmpPdf)    
        
except Exception, ErrorDesc:

    if str(ErrorDesc) == "DataDrivenPagesError":
        ErrorDesc = "Les pages dynamiques ne sont pas configurees dans ce document."
    elif str(ErrorDesc) == "RemovePdfFileError":
        ErrorDesc = "Impossible de supprimer le fichier PDF déjà existant."
    else:
        arcpy.AddError(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Une erreur est survenue")
        ErrorDesc0 = ErrorDesc
        ErrorDesc = ""
        for sErr in ErrorDesc0:
            ErrorDesc = ErrorDesc + sErr

    arcpy.AddError(datetime.now().strftime("%Y/%m/%d-%H:%M:%S") + " - Erreur(s) lors des traitements - " + ErrorDesc)

try :
    blablabla

except Exception, ErrorDesc: 
    tutu

try :
blablabla

except Exception, ErrorDesc: 
    tutu

blabla ......... je recupère les chemins d'accès (file_ligne_all, file_mxd, dir_data_out, ...)
................  et je remplace les \ par des /


curseur = arcpy.SearchCursor(file_ligne_all)
for c_ligne in curseur:

    mxd = arcpy.mapping.MapDocument (file_mxd)


blabla ...... changement de source des couches, ajout des titres, du fond raster ..........


    # export en pdf
    pdf_doc = arcpy.mapping.PDFDocumentCreate(dir_data_out + "/" + c_ligne.LIGNE + ".pdf")
    
    for page_num in range(1, mxd.dataDrivenPages.pageCount + 1):
        mxd.dataDrivenPages.currentPageID = page_num
        file_pdf = dir_data_out + "/" + c_ligne.LIGNE + "_" + str(page_num) + ".pdf"
        arcpy.mapping.ExportToPDF(mxd, file_pdf)
        try :
            pdf_doc.appendPages(file_pdf)
        except:
            arcpy.AddMessage("Erreur: " + file_pdf)
        
    pdf_doc.saveAndClose()
    del pdf_doc

    mxd.saveACopy(dir_data_out + "/atlas_"+ c_ligne.LIGNE + ".mxd")
    del mxd
del curseur