arcpy開(kāi)發(fā)實(shí)例之a(chǎn)dd-in代碼
import arcpy
import pythonaddins
class BtnEditFeature(object):
? ? """Implementation for pydemoaddin_addin.button_edit (Button)"""
? ? def __init__(self):
? ? ? ? self.enabled = True
? ? ? ? self.checked = False
? ? def onClick(self):
? ? ? ? with arcpy.da.UpdateCursor("points",["name","OID@"]) as cursor:
? ? ? ? ? ? for row in cursor:
? ? ? ? ? ? ? ? row[0]="point_"+str(row[1]);
? ? ? ? ? ? ? ? cursor.updateRow(row);
class BtnNewFeatureClass(object):
? ? """Implementation for pydemoaddin_addin.button_newfeatureclass (Button)"""
? ? def __init__(self):
? ? ? ? self.enabled = True
? ? ? ? self.checked = False
? ? def onClick(self):
? ? ? ? if arcpy.Exists("d:/temp/points.shp"):
? ? ? ? ? ? arcpy.management.Delete("d:/temp/points.shp");
? ? ? ? arcpy.management.CreateFeatureclass("d:/temp","points.shp","POINT");
? ? ? ? arcpy.management.AddField("d:/temp/points.shp","name","TEXT",20);
? ? ? ? pythonaddins.MessageBox("Create point.shp successfully!","TIP");
class ToolDrawPoint(object):
? ? """Implementation for pydemoaddin_addin.tool_drawpoint (Tool)"""
? ? def __init__(self):
? ? ? ? self.enabled = True
? ? ? ? self.cursor = 3;
? ? ? ? self.shape = "NONE" # Can set to "Line", "Circle" or "Rectangle" for interactive shape drawing and to activate the onLine/Polygon/Circle event sinks.
? ?
? ? def onMouseDownMap(self, x, y, button, shift):
? ? ? ? with arcpy.da.InsertCursor("d:/temp/points.shp",['SHAPE@XY']) as dc:
? ? ? ? ? ? dc.insertRow([(x,y)]);
? ? ? ? ? ? print ("One point at the mouse position({0},{1}) has been added.".format(x,y);
? ? ? ? mxd = arcpy.mapping.MapDocument("current");
? ? ? ? df = mxd.activeDataFrame;
? ? ? ? ext = df.extent;
? ? ? ? extent = ext
? ? ? ? ext.XMin = ext.XMin-0.1
? ? ? ? df.extent = ext;
? ?
class ToolIdentify(object):
? ? """Implementation for pydemoaddin_addin.tool_identify (Tool)"""
? ? def __init__(self):
? ? ? ? self.enabled = True
? ? ? ? self.shape = "NONE" # Can set to "Line", "Circle" or "Rectangle" for interactive shape drawing and to activate the onLine/Polygon/Circle event sinks.
? ?
? ? def onMouseDownMap(self, x, y, button, shift):
? ? ? ?
? ? ? ? mxd = arcpy.mapping.MapDocument("current");
? ? ? ? df = mxd.activeDataFrame;
? ? ? ? lyr = arcpy.mapping.ListLayers(mxd,"points",df)[0];
? ? ? ? pt = arcpy.Point(x,y);
? ? ? ? geom = arcpy.PointGeometry(pt)
? ? ? ?
? ? ? ? arcpy.SelectLayerByLocation_management( "points","WITHIN_A_DISTANCE" ,geom,0.1);
? ? ? ? selectedCount = arcpy.GetCount_management("points");
? ? ? ?
? ? ? ? if selectedCount>0: ? ? ? ? ? ?
? ? ? ? ? ? for f in lyr.getSelectionSet():
? ? ? ? ? ? ? ? print f;
? ?
class ToolRemove(object):
? ? """Implementation for pydemoaddin_addin.tool_remove (Tool)"""
? ? def __init__(self):
? ? ? ? self.enabled = True
? ? ? ? self.shape = "NONE" # Can set to "Line", "Circle" or "Rectangle" for interactive shape drawing and to activate the onLine/Polygon/Circle event sinks.
? ?
? ? def onRectangle(self, rectangle_geometry):
? ? ? ? mxd = arcpy.mapping.MapDocument("current");
? ? ? ? df = mxd.activeDataFrame;
? ? ? ? lyr = arcpy.mapping.ListLayers(mxd,"points",df)[0];
? ? ? ? rect = rectangle_geometry;
? ? ? ? arcpy.SelectLayerByLocation_management( "points","WITHIN" ,rect);
? ? ? ? selectedCount = arcpy.GetCount_management("points");
? ? ? ?
? ? ? ? if selectedCount>0:
? ? ? ? ? ? arcpy.da.UpdateCursor("points",)
? ? ? ? ? ? with arcpy.da.UpdateCursor("points","OID@") as cursor:
? ? ? ? ? ? for f in lyr.getSelectionSet():
? ? ? ? ? ? ? ? for row in cursor:
? ? ? ? ? ? ? ? ? ? if row[0]==f:
? ? ? ? ? ? ? ? ? ? ? ? cursor.deleteRow(); ? ? ? ? ?
? ? ? ? ? ? ? ?
? ? ? ? ? ? ? ?
