refact header file

giampaolo · Oct 19, 2023 · 8d0a043 · 8d0a043
1 parent 05c5234
commit 8d0a043
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 8 deletions.
diff --git a/psutil/_psutil_linux.c b/psutil/_psutil_linux.c
@@ -22,7 +22,7 @@
 
 static PyMethodDef mod_methods[] = {
     // --- per-process functions
-#if PSUTIL_HAVE_IOPRIO
+#ifdef PSUTIL_HAVE_IOPRIO
     {"proc_ioprio_get", psutil_proc_ioprio_get, METH_VARARGS},
     {"proc_ioprio_set", psutil_proc_ioprio_set, METH_VARARGS},
 #endif

diff --git a/psutil/arch/linux/proc.c b/psutil/arch/linux/proc.c
@@ -12,7 +12,7 @@
 #include "../../_psutil_common.h"
 
 
-#if PSUTIL_HAVE_IOPRIO
+#ifdef PSUTIL_HAVE_IOPRIO
 enum {
     IOPRIO_WHO_PROCESS = 1,
 };

diff --git a/psutil/arch/linux/proc.h b/psutil/arch/linux/proc.h
@@ -9,14 +9,17 @@
 #include <sched.h> // CPU_ALLOC
 
 // Linux >= 2.6.13
-#define PSUTIL_HAVE_IOPRIO defined(__NR_ioprio_get) && defined(__NR_ioprio_set)
+#if defined(__NR_ioprio_get) && defined(__NR_ioprio_set)
+    #define PSUTIL_HAVE_IOPRIO
+
+    PyObject *psutil_proc_ioprio_get(PyObject *self, PyObject *args);
+    PyObject *psutil_proc_ioprio_set(PyObject *self, PyObject *args);
+#endif
 
 // Should exist starting from CentOS 6 (year 2011).
 #ifdef CPU_ALLOC
     #define PSUTIL_HAVE_CPU_AFFINITY
-#endif
 
-PyObject *psutil_proc_cpu_affinity_get(PyObject *self, PyObject *args);
-PyObject *psutil_proc_cpu_affinity_set(PyObject *self, PyObject *args);
-PyObject *psutil_proc_ioprio_get(PyObject *self, PyObject *args);
-PyObject *psutil_proc_ioprio_set(PyObject *self, PyObject *args);
+    PyObject *psutil_proc_cpu_affinity_get(PyObject *self, PyObject *args);
+    PyObject *psutil_proc_cpu_affinity_set(PyObject *self, PyObject *args);
+#endif