I would recommend the application Wifi Auto! If your battery in the device becomes exhausted quickly as possible, that the cause is the WiFi module. The device uses a battery to maintain the connection over WiFi even when the screen is off. This application is best suited to automatically turn on WiFi when the screen is unlocked. And turn off WiFi when the screen is locked.
