在Python中更改进程优先级，跨平台

Question

我有一个Python程序，进行耗时的计算。由于它使用高CPU，并且我希望我的系统保持响应，请将程序的优先级更改为低于正常优先级。

我找到了这个： 在Windows中设置进程优先级-ActiveState

但是我正在寻找跨平台的解决方案。

- Craig McQueen

Answer 1

这是我用来将进程设置为低于正常优先级的解决方案：

lowpriority.py

def lowpriority():
    """ Set the priority of the process to below-normal."""

    import sys
    try:
        sys.getwindowsversion()
    except AttributeError:
        isWindows = False
    else:
        isWindows = True

    if isWindows:
        # Based on:
        #   "Recipe 496767: Set Process Priority In Windows" on ActiveState
        #   http://code.activestate.com/recipes/496767/
        import win32api,win32process,win32con

        pid = win32api.GetCurrentProcessId()
        handle = win32api.OpenProcess(win32con.PROCESS_ALL_ACCESS, True, pid)
        win32process.SetPriorityClass(handle, win32process.BELOW_NORMAL_PRIORITY_CLASS)
    else:
        import os

        os.nice(1)

在Windows和Linux上测试过Python 2.6。

- Craig McQueen

---

不错的解决方案，但为什么要调用sys.getwindowversion()而不是检查sys.platform？ - Winston Ewert

我只想可靠地知道我是否在运行Windows操作系统。sys.platform 有文档记录的返回值，但我不确定它是否可信--它是否真的在64位Windows上返回“win32”？（信不信由你，我还没有尝试在64位Windows上运行！我想现在应该可以在工作中尝试一下） - Craig McQueen

在Win7 64位系统上尝试（使用Python 64位）：sys.platform确实返回 'win32'。 - Jonathan Livni

@CraigMcQueen；确认在2008 R2服务器上--sys.platform返回“win32”。 - Molomby

为什么不直接使用 win32process.SetPriorityClass(win32api.GetCurrentProcess(), win32process.HIGH_PRIORITY_CLASS) 呢？ - panda-34

Answer 2

你可以使用 psutil 模块。

在 POSIX 平台上：

>>> import psutil, os
>>> p = psutil.Process(os.getpid())
>>> p.nice()
0
>>> p.nice(10)  # set
>>> p.nice()
10

在Windows操作系统下：

>>> p.nice(psutil.HIGH_PRIORITY_CLASS)

- Giampaolo Rodolà

---

谢谢提供链接。可惜的是，这两个平台上的API不同（10与psutil.HIGH_PRIORITY_CLASS）。“跨平台”应该意味着“在所有平台上使用相同的API”。我认为这是很遗憾的。 - Craig McQueen

Python社区在可移植性方面做得很糟糕。他们对“可移植”的定义是“我们将实现任何操作系统轻松允许我们执行的操作”。为什么在Windows上无法使用select()处理管道？这是可以做到的，但操作系统不会为您执行它，因此它没有被实现。只要您从未需要任何库，Python就是可移植的，这实际上是永远不可能的。 - user1594322

有些事情基于其本质无法跨平台，与编程语言无关，Python也不例外（主要是因为它无法做到）。 现在，我希望听听您的意见，如何重新设计psutil的良好功能，以便在POSIX和Windows平台上兼容。 如果您的建议合理，我将更改当前的API，并感谢您的贡献；否则，我会认为您只是在捣乱，因为我没有看到您的信息中有任何建设性的建议。 - Giampaolo Rodolà

进程优先级并非“本质上不可移植”；只是没有人费心去抽象出 Linux、MacOS 和 Windows 优先级系统的适当子集。在 Python 的子进程系统中提供更好的进程优先级控制并不难——你只需实现所有操作系统中存在的一组公共优先级控制即可。所有操作系统都理解“高”、“正常”、“低”和“空闲”优先级的概念。你可以将这些通用概念简单地映射到 Linux 和 BSD 派生版的 niceness 范围上。 - johnwbyrd

注意，Windows示例与POSIX示例相反。一个优先级为10的良好值表示低优先级。 - sebeck

Answer 3

在每个类Unix平台上（包括Linux和MacOSX），请参阅os.nice 这里：

os.nice(increment)
Add increment to the process’s “niceness”. Return the new niceness. Availability: Unix.

既然您已经有了一个适用于Windows的配方，那就覆盖了大多数平台--在Windows以外的所有地方都使用正参数调用os.nice函数，在那里使用该配方。据我所知，没有“好包装”的跨平台解决方案（将这个组合打包起来可能很困难，但是，如果只打包它，您会看到多少额外价值？-）

- Alex Martelli

Answer 4

如果您无法访问这些模块中的某些内容，可能可以在Windows上执行以下操作：

import os  
def lowpriority():  
    """ Set the priority of the process to below-normal."""  
    os.system("wmic process where processid=\""+str(os.getpid())+"\" CALL   setpriority \"below normal\"")  

你可以像上面的例子一样，明显地区分操作系统类型以确保兼容性。

- c1c2c3c4c

Answer 5

这里有一段对我来说很好的代码：

import sys
import os
import psutil

os_used = sys.platform
process = psutil.Process(os.getpid())  # Set highest priority for the python script for the CPU
if os_used == "win32":  # Windows (either 32-bit or 64-bit)
    process.nice(psutil.REALTIME_PRIORITY_CLASS)
elif os_used == "linux":  # linux
    process.nice(psutil.IOPRIO_HIGH)
else:  # MAC OS X or other
    process.nice(20)  

如需了解代码中使用的数字和常量的更多信息，请参阅完整文档：
https://psutil.readthedocs.io/en/latest/#

- X0-user-0X