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(); ? ? ? ? ?

? ? ? ? ? ? ? ?

? ? ? ? ? ? ? ?