Minor tune up

mcneel · Nov 1, 2018 · 4ff937c · 4ff937c
1 parent 32dfded
commit 4ff937c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rhinopython/SampleEventHandler.py b/rhinopython/SampleEventHandler.py
@@ -18,13 +18,19 @@ def __init__(self):
         Rhino.RhinoDoc.ReplaceRhinoObject += self.OnReplaceRhinoObject
         Rhino.RhinoDoc.DeleteRhinoObject += self.OnDeleteRhinoObject
         Rhino.RhinoDoc.UndeleteRhinoObject += self.OnUndeleteRhinoObject
+        self.Enabled = True
 
     # Disables the event handlers
     def Disable(self):
         Rhino.RhinoDoc.AddRhinoObject -= self.OnAddRhinoObject
         Rhino.RhinoDoc.ReplaceRhinoObject -= self.OnReplaceRhinoObject
         Rhino.RhinoDoc.DeleteRhinoObject -= self.OnDeleteRhinoObject
         Rhino.RhinoDoc.UndeleteRhinoObject -= self.OnUndeleteRhinoObject
+        self.Enabled = False
+
+    # Returns the enabled state
+    def IsEnabled(self):
+        return self.Enabled
 
     # AddRhinoObject event handler
     def OnAddRhinoObject(self, sender, e):