refactor ionice() on Linux

1 files changed, 3 insertions, 6 deletions

diff --git a/psutil/_pslinux.py b/psutil/_pslinux.py
index 7c9c669f..6c58cf2c 100644
--- a/psutil/_pslinux.py
+++ b/psutil/_pslinux.py
@@ -1990,12 +1990,9 @@ class Process(object):
                 value = 0
             if value and ioclass == IOPRIO_CLASS_IDLE:
                 raise ValueError("IOPRIO_CLASS_IDLE accepts no value")
-            try:
-                return cext.proc_ioprio_set(self.pid, ioclass, value)
-            except OSError as err:
-                if err.errno == errno.EINVAL and value > 7:
-                    raise ValueError("value not in 0-7 range")
-                raise
+            if value < 0 or value > 7:
+                raise ValueError("value not in 0-7 range")
+            return cext.proc_ioprio_set(self.pid, ioclass, value)
 
     if HAS_PRLIMIT: