Using AndroidViewClient (>= 4.7.0) (which uses python instead of monkeyrunner`) you can do
from com.dtmilano.android.viewclient import ViewClient
device, serialno = ViewClient.connectToDeviceOrExit()
if device.isLocked():
device.wake()
device.unlock()