DeepFaceLab/core/osex.py
Colombo c900c65c31 _
2020-02-29 15:20:17 +04:00

44 lines
No EOL
1.5 KiB
Python

import os
import sys
if sys.platform[0:3] == 'win':
from ctypes import windll
from ctypes import wintypes
def set_process_lowest_prio():
try:
if sys.platform[0:3] == 'win':
GetCurrentProcess = windll.kernel32.GetCurrentProcess
GetCurrentProcess.restype = wintypes.HANDLE
SetPriorityClass = windll.kernel32.SetPriorityClass
SetPriorityClass.argtypes = (wintypes.HANDLE, wintypes.DWORD)
SetPriorityClass ( GetCurrentProcess(), 0x00000040 )
elif 'darwin' in sys.platform:
os.nice(10)
elif 'linux' in sys.platform:
os.nice(20)
except:
print("Unable to set lowest process priority")
def set_process_dpi_aware():
if sys.platform[0:3] == 'win':
windll.user32.SetProcessDPIAware(True)
def get_screen_size():
if sys.platform[0:3] == 'win':
user32 = windll.user32
return user32.GetSystemMetrics(0), user32.GetSystemMetrics(1)
elif 'darwin' in sys.platform:
pass
elif 'linux' in sys.platform:
pass
return (1366, 768)
def linux_ignore_UserWarning():
if sys.platform[0:3] != 'win':
# fix for Linux , Ignoring :
# /usr/lib/python3.6/multiprocessing/semaphore_tracker.py:143:
# UserWarning: semaphore_tracker: There appear to be 1 leaked semaphores to clean up at shutdown
import warnings
warnings.filterwarnings(action='ignore', message='semaphore_tracker')