CODEWITHSUNDEEP
python
Craig McQueen
Craig McQueen

Reputation: 43446

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.

Upvotes: 41

Views: 55850

Answers (5)

X0-user-0X
X0-user-0X

Reputation: 868

Here a code that work good for me:

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)

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

Upvotes: 2

Craig McQueen
Craig McQueen

Reputation: 43446

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.

Upvotes: 48

c1c2c3c4c
c1c2c3c4c

Reputation: 41

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.

Upvotes: 4

Alex Martelli
Alex Martelli

Reputation: 881705

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?-)

Upvotes: 11

Giampaolo Rodolà
Giampaolo Rodolà

Reputation: 13066

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)

Upvotes: 30

Related Questions