Python3: resources and libraries

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

diff --git a/resources/libraries/python/LimitUtil.py b/resources/libraries/python/LimitUtil.py
index 45aeaff..c34109c 100644 (file)
--- a/resources/libraries/python/LimitUtil.py
+++ b/resources/libraries/python/LimitUtil.py
@@ -15,10 +15,10 @@
 
 from resources.libraries.python.ssh import exec_cmd_no_error
 
-__all__ = ["LimitUtil"]
+__all__ = [u"LimitUtil"]
 
 
-class LimitUtil(object):
+class LimitUtil:
     """Class contains methods for getting or setting process resource limits."""
 
     @staticmethod
@@ -30,10 +30,8 @@ class LimitUtil(object):
         :type node: dict
         :type pid: int
         """
-        command = 'prlimit --noheadings --pid={pid}'.format(pid=pid)
-
-        message = 'Node {host} failed to run: {command}'.\
-            format(host=node['host'], command=command)
+        command = f"prlimit --noheadings --pid={pid}"
+        message = f"Node {node[u'host']} failed to run: {command}"
 
         exec_cmd_no_error(node, command, sudo=True, message=message)
 
@@ -50,11 +48,7 @@ class LimitUtil(object):
         :type resource: str
         :type limit: str
         """
-        command = 'prlimit --{resource}={limit} --pid={pid}'.format(
-            resource=resource, limit=limit, pid=pid)
-
-        message = 'Node {host} failed to run: {command}'.\
-            format(host=node['host'], command=command)
+        command = f"prlimit --{resource}={limit} --pid={pid}"
+        message = f"Node {node[u'host']} failed to run: {command}"
 
         exec_cmd_no_error(node, command, sudo=True, message=message)
-