Checks lock screen status

lock-status.py

from Foundation import NSBundle
import objc

BIT_WIDTH = 32
INT_MAX  =  2**(BIT_WIDTH-1)-1

MobileKeyBag = NSBundle.bundleWithPath_('/System/Library/PrivateFrameworks/MobileKeyBag.framework')
objc.loadBundleFunctions(MobileKeyBag, globals(), [('MKBDeviceGetGracePeriod', '@@')])
grace = MKBDeviceGetGracePeriod({});

try:
    p = grace["GracePeriod"]
    if p == INT_MAX:
        print("Screen lock disabled")
    else:
        print "Screen lock {}".format("is not set" if p == -1 else "is set, grace period {}".format(p))
except:
    print("Something wrong!")