Fix data-driven pages output bug

plot_heatmaps.py

@@ -110,6 +110,9 @@ def process(yaml_name):
     if is_latest:
         output_dirs.append(output_dir_latest)
 
+    # Activate data driven pages
+    ddp = mxd.dataDrivenPages
+
     # Export to jpg
     for output_dir in output_dirs:
         page_name = ddp.pageRow.Beach
@@ -118,21 +121,6 @@ def process(yaml_name):
         arcpy.mapping.ExportToJPEG(
             mxd, jpg_name, resolution=200, jpeg_quality=80)
 
-    # Activate data driven pages
-    ddp = mxd.dataDrivenPages
-
-    for page_num in range(1, ddp.pageCount + 1):
-        ddp.currentPageID = page_num
-
-        # Export beaches that are covered by current survey
-        if ddp.pageRow.Parent == beach.lower():
-            page_name = ddp.pageRow.Beach
-            jpg_name = os.path.join(output_jpg_dir, page_name + '.jpg')
-
-            # Export to jpg
-            arcpy.mapping.ExportToJPEG(
-                mxd, jpg_name, resolution=200, jpeg_quality=80)
-
 
 def main():
     example_text = """examples: