Why not restrict internet access to the "Whatsapp" in your app settings when you need a break? Works fine for me.