def screenshot():
    from sr_automation.platform.android.Android import Android
    from sr_automation.platform.sunriver.Chroot import Chroot
    from sr_automation.platform.sunriver.Sunriver import Sunriver
    device_id = Android.devices().keys()[0]
    android = Android(device_id)
    chroot  = Chroot(android)
    chroot.run("DISPLAY=:0 gnome-screenshot -f /tmp/screenshot.png").wait()
    android.cmd("pull /data/debian/tmp/screenshot.png /tmp/screenshot.png").wait()
    import os
    os.system("feh /tmp/screenshot.png; rm -f /tmp/screenshot.png")
 def __init__(self):
     self._device_id = Android.devices().keys()[0]
     self._android = Android(self._device_id)
     self._desktop = DesktopInYourPocket(self._android)
     self._linux = self.connect(Chroot(self._android), NetInterfaces(self._android))
     self._switch_to_android =  SwitchToAndroid(self._linux, self._desktop)
     self.start()
def install_dut():
    """Prepare DUT for use by automation framework"""
    from sr_automation.platform.android.Android import Android
    from sr_automation.platform.sunriver.Chroot import Chroot
    from sr_automation.platform.sunriver.Sunriver import Sunriver
    device_id = Android.devices().keys()[0]
    android = Android(device_id)
    chroot  = Chroot(android)
    Sunriver.install(chroot)
def devices():
    """Show connected devices"""
    from sr_automation.platform.android.Android import Android
    print Android.devices()
 def install():
     device_id = Android.devices().keys()[0]
     android = Android(device_id)
     chroot  = Chroot(android)
     Sunriver.install(chroot)
        self._android.ui(text="Start Desktop").click()

    def stop(self):
        #self.click_menu(pattern=DesktopInYourPocket.BUTTON_STOP)
        #self._android.cmd('shell /data/srctl stop')
        self.openApp()
        self._android.ui(text="Shutdown Desktop").click()
       
    def switch_to_desktop(self):
        if not self.is_desktop_running():
            #self.click_menu(pattern=DesktopInYourPocket.BUTTON_SWITCH_TO_DESKTOP)
            log.info('Switching to desktop')
            #self._android.cmd('shell /data/srctl switch desktop')
            self.openApp()
            self._android.ui(text="Switch to Desktop").click()
            sleep(8)


    def is_desktop_running(self):
        output = self._android.cmd("shell getprop sunriver.active").stdout.read().strip()
        if len(output) == 0:
            return False
        return bool(int(output))

if __name__ == "__main__":
    from sr_automation.platform.android.Android import Android
    device_id = Android.devices().keys()[0]
    android   = Android(device_id)
    desktop   = DesktopInYourPocket(android)
    desktop.stop()