简体   繁体   中英

Change process priority in Python, cross-platform

I've got a Python program that does time-consuming computations. Since it uses high CPU, and I want my system to remain responsive, I'd like the program to change its priority to below-normal.

I found this: Set Process Priority In Windows - ActiveState

But I'm looking for a cross-platform solution.

Here's the solution I'm using to set my process to below-normal priority:

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)

Tested on Python 2.6 on Windows and Linux.

You can use psutil module.

On POSIX platforms:

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

On Windows:

>>> p.nice(psutil.HIGH_PRIORITY_CLASS)

On every Unix-like platform (including Linux and MacOsX), see os.nice here :

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

Since you already have a recipe for Windows, that covers most platforms -- call os.nice with a positive argument everywhere but Windows, use that recipe there. There is no "nicely packaged" cross-platform solution AFAIK (would be hard to package this combo up, but, how much extra value would you see in just packaging it?-)

If you don't have access to some of these modules you can potentially do it in Windows with:

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\"")  

You can obviously distinguish OS types as with examples above for compatibility.

Here a code that work good for me:

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)  

For futher informations on the number and constant used in the code the full documentation is here:
https://psutil.readthedocs.io/en/latest/#

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM