add method of related port/device to CollectUtils

diff --git a/linkpair/collect/utils.py b/linkpair/collect/utils.py
index 08bf138..b8d4e92 100755 (executable)
--- a/linkpair/collect/utils.py
+++ b/linkpair/collect/utils.py
@@ -78,6 +78,25 @@ class CollectUtils(object):
         else:
             self._devices[device.label] = device
 
+    def get_linkpair_by_device_port(self, dev1_name, port1_name):
+        for linkpair in self._linkpairs:
+            if linkpair.device1 == dev1_name:
+                if linkpair.port1 == port1_name:
+                    return linkpair
+
+    def exist_linkpair_by_device_port(self, dev1_name, port1_name):
+        for linkpair in self._linkpairs:
+            if linkpair.device1 == dev1_name:
+                if linkpair.port1 == port1_name:
+                    return True
+        return False
+    
+    def exist_port(self, device_name):
+        if device_name in self._devices:
+            return True
+        else:
+            return False
+
     def add_port(self, port):
         if port.label in self._ports:
             self._ports[port.label].metadata.update(port.metadata)
@@ -90,6 +109,12 @@ class CollectUtils(object):
         else:
             return self.PORT_NOT_FOUND
 
+    def exist_port(self, port_name):
+        if port_name in self._ports:
+            return True
+        else:
+            return False
+
     def get_linkpairs(self):
         return self._linkpairs