|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from contextlib import contextmanager |
|
|
|
|
|
import ctypes |
|
|
import ctypes.util |
|
|
|
|
|
objc = ctypes.cdll.LoadLibrary(ctypes.util.find_library('objc')) |
|
|
_ = ctypes.cdll.LoadLibrary(ctypes.util.find_library('Foundation')) |
|
|
|
|
|
void_p = ctypes.c_void_p |
|
|
ull = ctypes.c_uint64 |
|
|
|
|
|
objc.objc_getClass.restype = void_p |
|
|
objc.sel_registerName.restype = void_p |
|
|
objc.objc_msgSend.restype = void_p |
|
|
objc.objc_msgSend.argtypes = [void_p, void_p] |
|
|
|
|
|
msg = objc.objc_msgSend |
|
|
|
|
|
def _utf8(s): |
|
|
"""ensure utf8 bytes""" |
|
|
if not isinstance(s, bytes): |
|
|
s = s.encode('utf8') |
|
|
return s |
|
|
|
|
|
def n(name): |
|
|
"""create a selector name (for methods)""" |
|
|
return objc.sel_registerName(_utf8(name)) |
|
|
|
|
|
def C(classname): |
|
|
"""get an ObjC Class by name""" |
|
|
ret = objc.objc_getClass(_utf8(classname)) |
|
|
assert ret is not None, "Couldn't find Class %s" % classname |
|
|
return ret |
|
|
|
|
|
|
|
|
|
|
|
NSActivityIdleDisplaySleepDisabled = (1 << 40) |
|
|
NSActivityIdleSystemSleepDisabled = (1 << 20) |
|
|
NSActivitySuddenTerminationDisabled = (1 << 14) |
|
|
NSActivityAutomaticTerminationDisabled = (1 << 15) |
|
|
NSActivityUserInitiated = (0x00FFFFFF | NSActivityIdleSystemSleepDisabled) |
|
|
NSActivityUserInitiatedAllowingIdleSystemSleep = (NSActivityUserInitiated & ~NSActivityIdleSystemSleepDisabled) |
|
|
NSActivityBackground = 0x000000FF |
|
|
NSActivityLatencyCritical = 0xFF00000000 |
|
|
|
|
|
def beginActivityWithOptions(options, reason=""): |
|
|
"""Wrapper for: |
|
|
|
|
|
[ [ NSProcessInfo processInfo] |
|
|
beginActivityWithOptions: (uint64)options |
|
|
reason: (str)reason |
|
|
] |
|
|
""" |
|
|
NSProcessInfo = C('NSProcessInfo') |
|
|
NSString = C('NSString') |
|
|
|
|
|
objc.objc_msgSend.argtypes = [void_p, void_p, void_p] |
|
|
reason = msg(NSString, n("stringWithUTF8String:"), _utf8(reason)) |
|
|
objc.objc_msgSend.argtypes = [void_p, void_p] |
|
|
info = msg(NSProcessInfo, n('processInfo')) |
|
|
objc.objc_msgSend.argtypes = [void_p, void_p, ull, void_p] |
|
|
activity = msg(info, |
|
|
n('beginActivityWithOptions:reason:'), |
|
|
ull(options), |
|
|
void_p(reason) |
|
|
) |
|
|
return activity |
|
|
|
|
|
def endActivity(activity): |
|
|
"""end a process activity assertion""" |
|
|
NSProcessInfo = C('NSProcessInfo') |
|
|
objc.objc_msgSend.argtypes = [void_p, void_p] |
|
|
info = msg(NSProcessInfo, n('processInfo')) |
|
|
objc.objc_msgSend.argtypes = [void_p, void_p, void_p] |
|
|
msg(info, n("endActivity:"), void_p(activity)) |
|
|
|
|
|
_theactivity = None |
|
|
|
|
|
def nope(): |
|
|
"""disable App Nap by setting NSActivityUserInitiatedAllowingIdleSystemSleep""" |
|
|
global _theactivity |
|
|
_theactivity = beginActivityWithOptions( |
|
|
NSActivityUserInitiatedAllowingIdleSystemSleep, |
|
|
"Because Reasons" |
|
|
) |
|
|
|
|
|
def nap(): |
|
|
"""end the caffeinated state started by `nope`""" |
|
|
global _theactivity |
|
|
if _theactivity is not None: |
|
|
endActivity(_theactivity) |
|
|
_theactivity = None |
|
|
|
|
|
def napping_allowed(): |
|
|
"""is napping allowed?""" |
|
|
return _theactivity is None |
|
|
|
|
|
@contextmanager |
|
|
def nope_scope( |
|
|
options=NSActivityUserInitiatedAllowingIdleSystemSleep, |
|
|
reason="Because Reasons" |
|
|
): |
|
|
"""context manager for beginActivityWithOptions. |
|
|
|
|
|
Within this context, App Nap will be disabled. |
|
|
""" |
|
|
activity = beginActivityWithOptions(options, reason) |
|
|
try: |
|
|
yield |
|
|
finally: |
|
|
endActivity(activity) |
|
|
|
|
|
__all__ = [ |
|
|
"NSActivityIdleDisplaySleepDisabled", |
|
|
"NSActivityIdleSystemSleepDisabled", |
|
|
"NSActivitySuddenTerminationDisabled", |
|
|
"NSActivityAutomaticTerminationDisabled", |
|
|
"NSActivityUserInitiated", |
|
|
"NSActivityUserInitiatedAllowingIdleSystemSleep", |
|
|
"NSActivityBackground", |
|
|
"NSActivityLatencyCritical", |
|
|
"beginActivityWithOptions", |
|
|
"endActivity", |
|
|
"nope", |
|
|
"nap", |
|
|
"napping_allowed", |
|
|
"nope_scope", |
|
|
] |
|
|
|
