[port_util] Allow system with no ports in config db run without errors

Signed-off-by: liora <[email protected]>
sonic-net · Aug 31, 2021 · 57f53b2 · 57f53b2
1 parent 40637b8
commit 57f53b2
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/src/swsssdk/port_util.py b/src/swsssdk/port_util.py
@@ -57,20 +57,14 @@ def get_index_from_str(if_name):
         if match:
             return int(match.group(1)) + baseidx
 
-def get_interface_oid_map(db):
+def get_interface_oid_map(db, blocking=True):
     """
         Get the Interface names from Counters DB
     """
-    if_name_map = {}
-    if_id_map = {}
-
     db.connect('COUNTERS_DB')
-    if db.exists('COUNTERS_DB', 'COUNTERS_PORT_NAME_MAP'):
-        if_name_map = db.get_all('COUNTERS_DB', 'COUNTERS_PORT_NAME_MAP', blocking=True)
-
-    if db.exists('COUNTERS_DB', 'COUNTERS_LAG_NAME_MAP'):
-        if_lag_name_map = db.get_all('COUNTERS_DB', 'COUNTERS_LAG_NAME_MAP', blocking=True)
-        if_name_map.update(if_lag_name_map)
+    if_name_map = db.get_all('COUNTERS_DB', 'COUNTERS_PORT_NAME_MAP', blocking)
+    if_lag_name_map = db.get_all('COUNTERS_DB', 'COUNTERS_LAG_NAME_MAP', blocking)
+    if_name_map.update(if_lag_name_map)
 
     if not if_name_map:
         return {}, {}