diff mbox series

[V2,3/5] nics/net_device: add attribute pkg and get method

Message ID 20210318092215.27270-4-haiyangx.zhao@intel.com (mailing list archive)
State Changes Requested
Headers
Series framework: add a proposal of recognizing pkgs |

Commit Message

Zhao, HaiyangX March 18, 2021, 9:22 a.m. UTC 
  add attribute pkg to record nic current package and add the related get method.

Signed-off-by: Haiyang Zhao <haiyangx.zhao@intel.com>
---
 nics/net_device.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)
diff mbox series

Patch

diff --git a/nics/net_device.py b/nics/net_device.py
index bdc9d37d..6a26d1eb 100644
--- a/nics/net_device.py
+++ b/nics/net_device.py
@@ -71,6 +71,7 @@  class NetDevice(object):
         self.intf2_name = None
         self.get_interface_name()
         self.socket = self.get_nic_socket()
+        self.pkg = {}
 
     def stop(self):
         pass
@@ -118,6 +119,27 @@  class NetDevice(object):
         """
         return self.crb.get_pci_dev_driver(self.domain_id, self.bus_id, self.devfun_id)
 
+    def get_nic_pkg(self):
+        """
+        Get the NIC pkg.
+        """
+        out = self.__send_expect('dmesg | grep "DDP package" | tail -1', '# ')
+        if 'could not load' in out:
+            print(RED(out))
+            print(RED('Warning: The loaded DDP package version may not as you expected'))
+            try:
+                pkg_info = out.split('. ')[1].lower()
+                self.pkg['type'] = re.findall(".*package '(.*)'", pkg_info)[0].strip()
+                self.pkg['version'] = re.findall("version(.*)", pkg_info)[0].strip()
+            except:
+                print(RED('Warning: get pkg info failed'))
+        else:
+            pkg_info = out.split(': ')[-1].lower().split('package version')
+            if len(pkg_info) > 1:
+                self.pkg['type'] = pkg_info[0].strip()
+                self.pkg['version'] = pkg_info[1].strip()
+        return self.pkg
+
     def get_nic_socket(self):
         """
         Get socket id of specified pci device.
@@ -428,7 +450,6 @@  class NetDevice(object):
         self.__send_expect("ifconfig %s down" % intf, "# ")
         self.__send_expect("ifconfig %s up" % intf, "# ")
 
-
     @nic_has_driver
     def disable_ipv6(self):
         """