Python3: resources and libraries

[csit.git] / resources / libraries / python / DpdkUtil.py

diff --git a/resources/libraries/python/DpdkUtil.py b/resources/libraries/python/DpdkUtil.py
index bbd6987..3a04cbd 100644 (file)
--- a/resources/libraries/python/DpdkUtil.py
+++ b/resources/libraries/python/DpdkUtil.py
@@ -17,7 +17,7 @@ from resources.libraries.python.OptionString import OptionString
 from resources.libraries.python.ssh import exec_cmd_no_error
 
 
-class DpdkUtil(object):
+class DpdkUtil:
     """Utilities for DPDK."""
 
     @staticmethod
@@ -29,17 +29,19 @@ class DpdkUtil(object):
         :returns: EAL parameters.
         :rtype: OptionString
         """
-        options = OptionString(prefix='-')
-        options.add('v')
+        options = OptionString(prefix=u"-")
+        options.add(u"v")
         # Set the hexadecimal bitmask of the cores to run on.
-        options.add_with_value_from_dict('l', 'eal_corelist', kwargs)
+        options.add_with_value_from_dict(u"l", u"eal_corelist", kwargs)
         # Set master core.
-        options.add_with_value('-master-lcore', '0')
+        options.add_with_value(u"-master-lcore", u"0")
         # Load an external driver. Multiple -d options are allowed.
         options.add_with_value_if_from_dict(
-            'd', '/usr/lib/librte_pmd_virtio.so', 'eal_driver', kwargs, True)
+            u"d", u"/usr/lib/librte_pmd_virtio.so", u"eal_driver", kwargs, True
+        )
         options.add_if_from_dict(
-            '-in-memory', 'eal_in_memory', kwargs, False)
+            u"-in-memory", u"eal_in_memory", kwargs, False
+        )
         return options
 
     @staticmethod
@@ -51,50 +53,54 @@ class DpdkUtil(object):
         :returns: PMD parameters.
         :rtype: OptionString
         """
-        options = OptionString(prefix='--')
+        options = OptionString(prefix=u"--")
         # Set the forwarding mode: io, mac, mac_retry, mac_swap, flowgen,
         # rxonly, txonly, csum, icmpecho, ieee1588
         options.add_equals_from_dict(
-            'forward-mode', 'pmd_fwd_mode', kwargs, 'io')
+            u"forward-mode", u"pmd_fwd_mode", kwargs, u"io"
+        )
         # Set the number of packets per burst to N.
-        options.add_equals('burst', 64)
+        options.add_equals(u"burst", 64)
         # Set the number of descriptors in the TX rings to N.
-        options.add_equals_from_dict('txd', 'pmd_txd', kwargs, 1024)
+        options.add_equals_from_dict(u"txd", u"pmd_txd", kwargs, 1024)
         # Set the number of descriptors in the RX rings to N.
-        options.add_equals_from_dict('rxd', 'pmd_rxd', kwargs, 1024)
+        options.add_equals_from_dict(u"rxd", u"pmd_rxd", kwargs, 1024)
         # Set the number of queues in the TX to N.
-        options.add_equals_from_dict('txq', 'pmd_txq', kwargs, 1)
+        options.add_equals_from_dict(u"txq", u"pmd_txq", kwargs, 1)
         # Set the number of queues in the RX to N.
-        options.add_equals_from_dict('rxq', 'pmd_rxq', kwargs, 1)
+        options.add_equals_from_dict(u"rxq", u"pmd_rxq", kwargs, 1)
         # Set the hexadecimal bitmask of offloads.
-        options.add_equals_from_dict('tx-offloads', 'pmd_tx_offloads', kwargs)
+        options.add_equals_from_dict(u"tx-offloads", u"pmd_tx_offloads", kwargs)
         # Set the number of mbufs to be allocated in the mbuf pools.
-        options.add_equals_from_dict('total-num-mbufs', 'pmd_num_mbufs', kwargs)
+        options.add_equals_from_dict(
+            u"total-num-mbufs", u"pmd_num_mbufs", kwargs
+        )
         # Disable hardware VLAN.
         options.add_if_from_dict(
-            'disable-hw-vlan', 'pmd_disable_hw_vlan', kwargs, True)
+            u"disable-hw-vlan", u"pmd_disable_hw_vlan", kwargs, True
+        )
         # Set the MAC address XX:XX:XX:XX:XX:XX of the peer port N
-        options.add_equals_from_dict('eth-peer', 'pmd_eth_peer_0', kwargs)
-        options.add_equals_from_dict('eth-peer', 'pmd_eth_peer_1', kwargs)
+        options.add_equals_from_dict(u"eth-peer", u"pmd_eth_peer_0", kwargs)
+        options.add_equals_from_dict(u"eth-peer", u"pmd_eth_peer_1", kwargs)
         # Set the max packet length.
-        options.add_equals_from_dict('max-pkt-len', 'pmd_max_pkt_len', kwargs)
+        options.add_equals_from_dict(u"max-pkt-len", u"pmd_max_pkt_len", kwargs)
         # Set the number of forwarding cores based on coremask.
-        options.add_equals_from_dict('nb-cores', 'pmd_nb_cores', kwargs)
+        options.add_equals_from_dict(u"nb-cores", u"pmd_nb_cores", kwargs)
         return options
 
     @staticmethod
     def get_testpmd_cmdline(**kwargs):
         """Get DPDK testpmd command line arguments.
 
-        :param args: Key-value testpmd parameters.
-        :type args: dict
+        :param kwargs: Key-value testpmd parameters.
+        :type kwargs: dict
         :returns: Command line string.
         :rtype: OptionString
         """
         options = OptionString()
-        options.add('testpmd')
+        options.add(u"testpmd")
         options.extend(DpdkUtil.get_eal_options(**kwargs))
-        options.add('--')
+        options.add(u"--")
         options.extend(DpdkUtil.get_pmd_options(**kwargs))
         return options
 
@@ -103,14 +109,14 @@ class DpdkUtil(object):
         """Start DPDK testpmd app on VM node.
 
         :param node: VM Node to start testpmd on.
-        :param args: Key-value testpmd parameters.
+        :param kwargs: Key-value testpmd parameters.
         :type node: dict
         :type kwargs: dict
         """
         cmd_options = OptionString()
-        cmd_options.add("/start-testpmd.sh")
+        cmd_options.add(u"/start-testpmd.sh")
         cmd_options.extend(DpdkUtil.get_eal_options(**kwargs))
-        cmd_options.add('--')
+        cmd_options.add(u"--")
         cmd_options.extend(DpdkUtil.get_pmd_options(**kwargs))
         exec_cmd_no_error(node, cmd_options, sudo=True, disconnect=True)
 
@@ -122,5 +128,5 @@ class DpdkUtil(object):
         :type node: dict
         :returns: nothing
         """
-        cmd = "/stop-testpmd.sh"  # Completed string, simpler than OptionString.
+        cmd = u"/stop-testpmd.sh"  # Completed string, simple one.
         exec_cmd_no_error(node, cmd, sudo=True, disconnect=True)