Add get_ports_by_switch() helper method

Change-Id: I7c78d607a323e02443c1ff582b8a9b9b4ad23402

1 files changed, 3 insertions, 0 deletions

diff --git a/db/db.py b/db/db.py
index 014d7de..97d4b18 100644
--- a/db/db.py
+++ b/db/db.py
@@ -215,6 +215,9 @@ class VlanDB:
     def get_port_name(self, port_id):
         return self._get_element("port_name", "port", "port_id", port_id)
 
+    def get_ports_by_switch(self, switch_id):
+        return self._get_multi_elements("port_id", "port", "switch_id", switch_id)
+
     def get_vlan_name(self, vlan_id):
         return self._get_element("vlan_name", "vlan", "vlan_id", vlan_id)