Updated db_migrator.py for switchport mode

MuhammadUmarAsad · MuhammadUmarAsad · commit b4e46dbdea95 · 2023-02-15T16:43:19.000+05:00
diff --git a/scripts/db_migrator.py b/scripts/db_migrator.py
@@ -444,6 +444,18 @@ def migrate_config_db_port_table_for_auto_neg(self):
                 elif value['autoneg'] == '0':
                     self.configDB.set(self.configDB.CONFIG_DB, '{}|{}'.format(table_name, key), 'autoneg', 'off')
 
+    def migrate_config_db_port_table_for_switchport_mode(self):
+        port_table = self.configDB.get_table('PORT')
+        vlan_member_table = self.configDB.get_table('VLAN_MEMBER')
+        for key, value in port_table.items():
+            if 'mode' in value:
+                self.configDB.set(self.configDB.CONFIG_DB, '{}|{}'.format("PORT", key), 'mode', value['mode'])
+            else:
+                if key in vlan_member_table.keys():
+                    self.configDB.set(self.configDB.CONFIG_DB, '{}|{}'.format("PORT", key), 'mode', 'trunk')
+                else:
+                    self.configDB.set(self.configDB.CONFIG_DB, '{}|{}'.format('PORT', key), 'mode', 'routed')
+
     def migrate_qos_db_fieldval_reference_remove(self, table_list, db, db_num, db_delimeter):
         for pair in table_list:
             table_name, fields_list = pair
@@ -853,6 +865,7 @@ def version_4_0_0(self):
         This is the latest version for master branch
         """
         log.log_info('Handling version_4_0_0')
+        self.migrate_config_db_port_table_for_switchport_mode()
         return None
 
     def get_version(self):
