Python package to do stuff on screen lock/unlock