rework of API, added socket connection status, made more consistent with Feature style

freecad/kiconnect/api.py

@@ -9,39 +9,39 @@ from .bases import BaseObject, BaseViewProvider
 class APIObject(BaseObject):
     TYPE = 'KiConnect::API'
 
-    def __init__(self, parent, feature):
-        super(APIObject, self).__init__(parent, feature)
+    def __init__(self, feature):
+        super(APIObject, self).__init__(feature)
 
         feature.addProperty('App::PropertyFile', 'Socket', 'KiConnect', 'Path to the KiCAD Socket File').Socket = '/tmp/kicad/api.lock'
+        feature.addProperty('App::PropertyBool', 'Connected', 'KiConnect', 'Is socket connected')
 
-    def execute(self, feature):
-        self.parent.ping_connection()
+        self.onDocumentRestored(feature)
 
-
-class APIViewProvider(BaseViewProvider):
-    pass
-
-
-    def __init__(self):
-        self.feature = App.ActiveDocument.addObject('App::FeaturePython', 'API')
-
-        APIObject(self, self.feature)
-        APIViewProvider(self, self.feature.ViewObject)
+    def onDocumentRestored(self, feature):
+        super(APIObject, self).onDocumentRestored(feature)
 
         self.kicad = KiCad()
-        self.ping_connection()
+        self.ping_connection(feature)
+
+        parent = feature.getParent()
+        if not parent: return
+
+        boards = [ board for board in parent.Group if hasattr(board, 'Type') and board.Type == 'KiConnect::Board' ]
+
 
     @property
     def is_connected(self):
-        return self._connected
+        return self.feature.Connected
 
-    def ping_connection(self):
+    def ping_connection(self, feature):
         try:
             self.kicad.ping()
             feature.Connected = True
+            feature.Label2 = 'Connected'
         except Exception as e:
             print(e)
             feature.Connected = False
+            feature.Label2 = 'Disconnected'
     pass
 
 
@@ -49,14 +49,15 @@ class APIViewProvider(BaseViewProvider):
     ICON = 'icon_footprint_browser.svg'
     TYPE = 'KiConnect::API'
 
-    def __init__(self, parent, viewprovider):
-        super(APIViewProvider, self).__init__(parent, viewprovider)
+    def __init__(self, viewprovider):
+        super(APIViewProvider, self).__init__(viewprovider)
 
 
-class API():
-    def __init__(self):
-        self.feature = App.ActiveDocument.addObject('App::FeaturePython', 'API')
+def makeAPI(parent):
+    feature = App.ActiveDocument.addObject('App::FeaturePython', 'API')
+    parent.addObject(feature)
 
-        self.object = APIObject(self, self.feature)
-        self.viewobject = APIViewProvider(self, self.feature.ViewObject)
+    APIObject(feature)
+    APIViewProvider(feature.ViewObject)
 
+    return feature