Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Option to run process with a nice level (Linux), or lower priority (Windows) #2304

Open
wants to merge 7 commits into
base: nightly
Choose a base branch
from
Open

Option to run process with a nice level (Linux), or lower priority (Windows) #2304

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
d92fd65
Add process priority run command "--nice/-ni", which sets process pri…
planetrocky Oct 31, 2024
c9c8742
add Python package pywin32 for Windows platforms only.
planetrocky Nov 1, 2024
d7b803d
Merge remote-tracking branch 'upstream/nightly' into nightly
planetrocky Nov 1, 2024
d05324f
add Python package pywin32 for Windows platforms only.
planetrocky Nov 1, 2024
5b9c228
update changelog for the low priority feature
planetrocky Nov 2, 2024
7b22c69
Added -ni/--nice (KOMETA_NICE) to Run Commands & Environment Variable…
planetrocky Nov 2, 2024
be50efb
inline code escapes for spelling
planetrocky Nov 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 27 additions & 0 deletions kometa.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@
"read-only-config": {"args": "ro", "type": "bool", "help": "Run without writing to the config"},
"divider": {"args": "d", "type": "str", "default": "=", "help": "Character that divides the sections (Default: '=')"},
"width": {"args": "w", "type": "int", "default": 100, "help": "Screen Width (Default: 100)"},
"nice": {"args": "ni", "type": "bool", "help": "Run with lower priority"},
}

parser = argparse.ArgumentParser()
Expand Down Expand Up @@ -257,6 +258,31 @@ def new_send(*send_args, **kwargs):
plexapi.BASE_HEADERS["X-Plex-Client-Identifier"] = str(uuid_num)
ImageFile.LOAD_TRUNCATED_IMAGES = True

def low_priority():
""" Set the priority of the process to below-normal."""
try:
sys.getwindowsversion()
except AttributeError:
isWindows = False
else:
isWindows = True

if isWindows:
import win32api, win32process, win32con
# priorityclasses:
# win32process.IDLE_PRIORITY_CLASS,
# win32process.BELOW_NORMAL_PRIORITY_CLASS,
# win32process.NORMAL_PRIORITY_CLASS,
# win32process.ABOVE_NORMAL_PRIORITY_CLASS,
# win32process.HIGH_PRIORITY_CLASS,
# win32process.REALTIME_PRIORITY_CLASS
win32process.SetPriorityClass(win32api.GetCurrentProcess(), win32process.BELOW_NORMAL_PRIORITY_CLASS)
else:
os.nice(10)

if run_args["nice"]:
low_priority()

def process(attrs):
with ProcessPoolExecutor(max_workers=1) as executor:
executor.submit(start, *[attrs])
Expand Down Expand Up @@ -284,6 +310,7 @@ def start(attrs):
logger.info(f" Platform: {platform.platform()}")
logger.info(f" Total Memory: {round(psutil.virtual_memory().total / (1024.0 ** 3))} GB")
logger.info(f" Available Memory: {round(psutil.virtual_memory().available / (1024.0 ** 3))} GB")
logger.info(f" Process Priority: {'low' if run_args["nice"] else 'normal'}")
if not is_docker and not is_linuxserver:
try:
with open(os.path.abspath(os.path.join(os.path.dirname(__file__), "requirements.txt")), "r") as file:
Expand Down