Wednesday, 13 January 2016

python - Automating script to set geodatabase?


I got some good help here yesterday. I'm basically trying to set up my script in such a way that once the user just puts in the workspace in the script tool in ArcGIS the script will do the rest. I have been able to create a geodatabase and now I'm trying to run the following code and I'm having trouble. I basically need to be able to change or reset the workspace halfway through to tell it to use the newly created GDB. This is the next piece of code that I'm trying to run:


# Create Raster Dataset
out_path = "Prog.gdb"
out_name = "R_Dataset"
# Run the Create Raster Dataset Script
arcpy.CreateRasterDataset_management (out_path, out_name, "", "8_BIT_UNSIGNED", "", "1", "", "PYRAMIDS -1 NEAREST DEFAULT 75 NO_SKIP", "128 128", "LZ77", "")
# Mosaic Rasters into a the Raster Dataset
inputs = arcpy.ListRasters("", "TIF")

# Set the target to the Raster Dataset that was already created
target = "R_Dataset"
# Run the Mosaic tool
arcpy.Mosaic_management (inputs, target, "LAST", "FIRST", "", "", "NONE", "0", "NONE")

The out_path is basically referencing the GDB that was just created in the workspace folder. It's telling me that this isn't a valid input etc. I was then trying to get it to sort through to Tiffs and automatically mosaic these into the newly created raster dataset.


Any and all help greatly appreciated.


Thanks


Edit: @Aaron


Below is the entire script that I want to use, basically if I can get the user to get the workspace as a parameter the script will then do everything else. Where I'm falling short is that basically the initial (and hopefully only) workspace they define is a folder. I then want the Mosaic to New Raster to select the 4 tiffs in the original workspace folder to create a new raster in the gdb I have already created.



I think then if I could then somehow change the workspace to the gdb (as in the workspace they select + the name of the gdb I have created) it would work as everything from then on in the script happens in the gdb if that makes sense?


    # Create file Geodatabase
# Import ArcGIS modules
import arcpy, os
# Check out the ArcGIS Spatial Analyst Extension
arcpy.CheckOutExtension("spatial")
# Need to be able to OverWrite Outputs
arcpy.env.overwriteOutput = True
#Set the workspace
arcpy.env.workspace = out_folder_path = arcpy.GetParameterAsText(0)

# Set up variables)
out_name = "Prog.gdb"
# Execute CreateFileGDB
arcpy.CreateFileGDB_management(out_folder_path, out_name)
# Set local variables before importing shapefiles to Geodatabase
inFeatures = arcpy.ListFeatureClasses()
outLocation = "Prog.gdb"
# Execute shapefile to Geodatabase
arcpy.FeatureClassToGeodatabase_conversion(inFeatures, outLocation)
# Remove _shp if it is present in feature class name

arcpy.env.workspace = os.path.join(out_folder_path, outLocation)
newFC = arcpy.ListFeatureClasses()
[arcpy.Rename_management(fc, fc.replace("_shp", "")) for fc in newFC if "_shp" in fc]
#Mosaic rasters to a new raster dataset
inputs = arcpy.ListRasters()
arcpy.env.workspace = Location = os.path.join(arcpy.env.workspace)
#Set Name
mosaic_name = "All_Mosaic.tif"
#Run the Mosaic Tool
arcpy.MosaicToNewRaster_management(inputs, Location, mosaic_name, "", "", "", 1, "", "")

#Inputs for the extract by mask tool
in_raster = "All_Mosaic.tif"
in_mask_data = "Study_area_extent"
#Extract by Mask
outExtractByMask = arcpy.sa.ExtractByMask (in_raster, in_mask_data)
# Save the output
outExtractByMask.save("T_Mask")
# Buffer the Thames feature class to create the Flood Risk Area
in_features = "Thames"
out_feature_class = "Flood_R_A"

# Run the Buffer Tool
arcpy.Buffer_analysis (in_features, out_feature_class, "300 Meters", "FULL", "ROUND", "ALL", "")
#Clip the Flood Risk Area to the Study Area Extent
# Set local variables
in_features = "Flood_R_A"
clip_features = "Study_area_extent"
out_feature_class = "Clipped_Flood_R_A"
xy_tolerance = ""
# Execute Clip
arcpy.Clip_analysis(in_features, clip_features, out_feature_class, xy_tolerance)

#Super Output Areas: Add field (this field will be used to calculate the population density of the each of the SOAs
inFeatures= "Super_Output_Areas"
arcpy.AddField_management(inFeatures, "Pop_Density", "DOUBLE", "", "", "", "", "NULLABLE", "NON_REQUIRED", "")
#Calculate field
#Set variables
inFeatures = "Super_Output_Areas"
fieldName1 = "Pop_Density"
arcpy.CalculateField_management(inFeatures, fieldName1, "!LSOA_POP! / !AreaSQKM!", "PYTHON", "")
# The Flood Risk Area and the SOA are going to be intersected in order try and find the population at risk
in_Features = (["Super_Output_Areas", "Clipped_Flood_R_A"])

intersectOutput = "Intersect"
arcpy.Intersect_analysis(in_Features, intersectOutput, "ALL", "", "INPUT")
#Add field. This will add a field to the Intersect file (intersection of SOAs and Flood Risk Area). This field will be used to calculate the population at risk.
inFeatures= "Intersect"
arcpy.AddField_management(inFeatures, "Pop_Risk", "DOUBLE", "", "", "", "", "NULLABLE", "NON_REQUIRED", "")
# Local variables:
Intersect = "Intersect"
# Process: Calculate Field
arcpy.CalculateField_management(Intersect, "Pop_Risk", "!Shape_Area! * !Pop_Density!", "PYTHON", "")
#Summary Statistics for the Total Population Effected

in_table = "Intersect"
out_table = "Pop_Risk_Sum"
stats = "SUM"
casefield = "Pop_Risk"
arcpy.Statistics_analysis(in_table, out_table, "Pop_Risk SUM", "")
# Process: Make Feature Layer
in_features = "Points_of_interest"
out_layer = "POIs_Layer"
arcpy.MakeFeatureLayer_management(in_features, out_layer, "", "", "OBJECTID OBJECTID VISIBLE NONE;Shape Shape VISIBLE NONE;CATEGORY CATEGORY VISIBLE NONE;NAME NAME VISIBLE NONE")
# Process: Select Layer By Location

in_layer = "POIs_Layer"
select_features = "Clipped_Flood_R_A"
arcpy.SelectLayerByLocation_management(in_layer, "INTERSECT", select_features, "", "NEW_SELECTION")
# Process: Copy Features
in_features = "POIs_Layer"
out_feature_class = "Copy_POI_LAYER"
arcpy.CopyFeatures_management(in_features, out_feature_class, "", "0", "0", "0")
# Process: Summary Statistics (2)
in_table = "Copy_POI_LAYER"
out_table = "POI_SUMMARY"

arcpy.Statistics_analysis(in_table, out_table, "CATEGORY COUNT", "CATEGORY")


No comments:

Post a Comment

arcpy - Changing output name when exporting data driven pages to JPG?

Is there a way to save the output JPG, changing the output file name to the page name, instead of page number? I mean changing the script fo...